aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-28 19:34:30 +01:00
Code Issues Packages Projects Releases Wiki Activity

Handles missing assistant thoughts.

Browse Source
This commit is contained in:
Torantulino
2023-03-29 02:22:33 +01:00
parent 96e833bc8e
commit 4ca40b2f41
1 changed files with 23 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
AutonomousAI/main.py
View File

@@ -125,23 +125,32 @@ def print_assistant_thoughts(assistant_reply):
try:
# Parse and print Assistant response
assistant_reply_json = json.loads(assistant_reply)
assistant_thoughts = assistant_reply_json["thoughts"]
assistant_thoughts_text = assistant_thoughts["text"]
assistant_thoughts_reasoning = assistant_thoughts["reasoning"]
assistant_thoughts_plan = assistant_thoughts["plan"]
assistant_thoughts_criticism = assistant_thoughts["criticism"]
print("ASSISTANT THOUGHTS: " + assistant_thoughts_text, flush=True)
print("REASONING: " + assistant_thoughts_reasoning, flush=True)
print("PLAN: ", flush=True)
assistant_thoughts = assistant_reply_json.get("thoughts")
if assistant_thoughts:
assistant_thoughts_text = assistant_thoughts.get("text")
assistant_thoughts_reasoning = assistant_thoughts.get("reasoning")
assistant_thoughts_plan = assistant_thoughts.get("plan")
assistant_thoughts_criticism = assistant_thoughts.get("criticism")
else:
assistant_thoughts_text = None
assistant_thoughts_reasoning = None
assistant_thoughts_plan = None
assistant_thoughts_criticism = None
print(f"ASSISTANT THOUGHTS: {assistant_thoughts_text}", flush=True)
print(f"REASONING: {assistant_thoughts_reasoning}", flush=True)
if assistant_thoughts_plan:
# Split the input_string using the newline character and dash
lines = assistant_thoughts_plan.split('\n- ')
print("PLAN: ", flush=True)
if assistant_thoughts_plan:
# Split the input_string using the newline character and dash
lines = assistant_thoughts_plan.split('\n- ')
# Iterate through the lines and print each one with a bullet point
for line in lines:
print(f"- {line.strip()}", flush=True)
print("CRITICISM: " + assistant_thoughts_criticism, flush=True)
# Iterate through the lines and print each one with a bullet point
for line in lines:
print(f"- {line.strip()}", flush=True)
print(f"CRITICISM: " + assistant_thoughts_criticism, flush=True)
except json.decoder.JSONDecodeError:
print("Error: Invalid JSON", flush=True)
print(assistant_reply, flush=True)