gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-31 01:17:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix bug for hypo_select_with_llm when not support response_schema (#1208)

Browse Source
This commit is contained in:
amstrongzyf
2025-08-28 14:36:23 +08:00
committed by GitHub
parent 70fdfc89b0
commit d759ca95e7
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/scenarios/data_science/proposal/exp_gen/proposal.py
+6 -4
View File
@@ -1140,7 +1140,11 @@ class DSProposalV2ExpGen(ExpGen):
merge_hours=DS_RD_SETTING.merge_hours,
# extra_exp_feedback_list_desc=extra_exp_feedback_list_str,
selected_extra_hypo_l=selected_extra_hypo_l,
hypothesis_output_format=T(".prompts_v2:output_format.hypothesis_select_format").r(),
hypothesis_output_format=(
T(".prompts_v2:output_format.hypothesis_select_format").r()
if not self.supports_response_schema
else None
),
sota_flag=sota_flag,
current_sota_score=current_sota_score,
ratio_merge_or_ensemble=ratio_merge_or_ensemble,
@@ -1157,9 +1161,7 @@ class DSProposalV2ExpGen(ExpGen):
user_prompt=user_prompt,
system_prompt=sys_prompt,
response_format=HypothesisSimple if self.supports_response_schema else {"type": "json_object"},
json_target_type=(
Dict[str, Dict[str, str | Dict[str, str | int]]] if not self.supports_response_schema else None
),
json_target_type=(Dict[str, str] if not self.supports_response_schema else None),
)
response_dict = json.loads(response)