Update main.py

main.py

@@ -149,7 +149,7 @@ async def chat(input_data: ChatInput):
             
             elif kind == "on_tool_end":
                 tool_output = event['data'].get('output', '').content
-                #print(type(tool_output))
+                print(type(tool_output))
                 #print(dir(tool_output))
                 #print the keys
                 pattern = r'data: (.*?)\ndata:'
@@ -163,8 +163,8 @@ async def chat(input_data: ChatInput):
                         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
+                                #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)