azalea/MTfin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[F] Fix dict keys

Browse Source
This commit is contained in:
Azalea Gui
2026-03-09 00:43:24 -04:00
parent 2dd9fc344a
commit 645dfa8bc2
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils_ai.py
+8 -2
View File
@@ -38,10 +38,16 @@ def _call_openai_with_prompt(prompt_id: str, prompt_version: str, input_text: st
try: try:
# If response is a Pydantic model # If response is a Pydantic model
if hasattr(response, 'output') and response.output: if hasattr(response, 'output') and response.output:
return response.output[0].content[0].text for out_msg in response.output:
if getattr(out_msg, 'type', '') == 'message':
return out_msg.content[0].text
return response.output[-1].content[0].text
# If response is a dictionary # If response is a dictionary
elif isinstance(response, dict) and "output" in response: elif isinstance(response, dict) and "output" in response:
return response["output"][0]["content"][0]["text"] for out_msg in response["output"]:
if out_msg.get("type") == "message":
return out_msg["content"][0]["text"]
return response["output"][-1]["content"][0]["text"]
# Fallback for choices (if API changes slightly) # Fallback for choices (if API changes slightly)
elif hasattr(response, 'choices') and response.choices: elif hasattr(response, 'choices') and response.choices:
return response.choices[0].message.content return response.choices[0].message.content