Update main.py

main.py

@@ -160,11 +160,11 @@ async def chat(input_data: ChatInput):
                     tool_output_json = match.group(1).strip()
                     try:
                         tool_output = json.loads(tool_output_json)
-                        # if "artifacts" in tool_output:
-                        #     for artifact in tool_output["artifacts"]:
-                        #         artifact_content = requests.get(f"{API_URL}/artifact/{artifact['artifact_id']}").content
-                        #         print(artifact_content)
-                        #         tool_output["artifacts"][artifact["artifact_id"]] = artifact_content
+                        if "artifacts" in tool_output:
+                            for artifact in tool_output["artifacts"]:
+                                artifact_content = requests.get(f"{API_URL}/artifact/{artifact['artifact_id']}").content
+                                print(artifact_content)
+                                tool_output["artifacts"][artifact["artifact_id"]] = artifact_content
                     except Exception as e:
                         print(e)
                         print("Error parsing tool output as json: ", tool_output)