github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-08 03:00:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat: Update LlmCallBlock to handle different response formats

Browse Source
This commit is contained in:
Toran Bruce Richards
2024-07-10 12:00:56 +01:00
parent 6cd5007857
commit cac41edafc
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
rnd/autogpt_server/autogpt_server/blocks/ai.py
View File

@@ -102,9 +102,13 @@ class LlmCallBlock(Block):
)
logger.warning(f"LLM attempt-{retry_count} response: {response_text}")
parsed_dict, parsed_error = parse_response(response_text)
if not parsed_error:
yield "response", {k: str(v) for k, v in parsed_dict.items()}
if input_data.expected_format:
parsed_dict, parsed_error = parse_response(response_text)
if not parsed_error:
yield "response", {k: str(v) for k, v in parsed_dict.items()}
return
else:
yield "response", {"response": response_text}
return
retry_prompt = f"""