added check for tool calls

openai_utils.py

@@ -88,38 +88,43 @@ def wait_for_response(thread, run):
             print(f'{run_status.status.capitalize()}... Please wait.')
             time.sleep(1.5)  # Wait before checking again
         elif run_status.status == 'requires_action':
-            outputs = {}
-            for tool_call in run_status.required_action.submit_tool_outputs.tool_calls:
-                if tool_call.function.name =="getListOfCompanies":
-                    try:
-                        args = json.loads(tool_call.function.arguments)
-                        print(f"Processing tool_call {tool_call.id}. Calling 'getListOfCompanies with args: {args}" )                        
-                        res = getListOfCompanies(args['query'])
-                        # matches = res['matches']
-                        # sorted_matches = sorted(matches, key=lambda x: x['score'], reverse=True)
-                        # output = ""
-                        # for m in sorted_matches:
-                        #     if 'summary' in m.metadata:
-                        #         try:                
-                        #             data = json.loads(m["metadata"]["summary"])
-                        #             output = output + f"company name: {m.metadata['company_name']}, description: {data['Summary']}\n"
-                        #         except Exception as e:
-                        #             output = output + f"company name: {m.metadata['company_name']}, description: {m.metadata['summary']}\n"
-                        if len(res)>0:
-                            outputs[tool_call.id] = res
-                    except Exception as e:
-                        print(f"Error calling tools, {str(e)}")
-                        traceback.print_exc()
-
-            tool_outputs=[{"tool_call_id": k, "output": v} for (k,v) in outputs.items()]
-            print(f"Finished tools calling: {tool_outputs}")
-            run = st.session_state.openai_client.beta.threads.runs.submit_tool_outputs(
-                thread_id=thread.id,
-                run_id=run.id,
-                tool_outputs=tool_outputs,
-            )
-            print(run_status.required_action)
-            #return run_status
+            required_action = run_status.required_action
+            if required_action.type == 'submit_tool_outputs':
+                outputs = {}
+                for tool_call in required_action.submit_tool_outputs.tool_calls:
+                    if tool_call.function.name =="getListOfCompanies":
+                        try:
+                            args = json.loads(tool_call.function.arguments)
+                            print(f"Processing tool_call {tool_call.id}. Calling 'getListOfCompanies with args: {args}" )
+                            res = getListOfCompanies(args['query'])
+                            # matches = res['matches']
+                            # sorted_matches = sorted(matches, key=lambda x: x['score'], reverse=True)
+                            # output = ""
+                            # for m in sorted_matches:
+                            #     if 'summary' in m.metadata:
+                            #         try:
+                            #             data = json.loads(m["metadata"]["summary"])
+                            #             output = output + f"company name: {m.metadata['company_name']}, description: {data['Summary']}\n"
+                            #         except Exception as e:
+                            #             output = output + f"company name: {m.metadata['company_name']}, description: {m.metadata['summary']}\n"
+                            if len(res)>0:
+                                outputs[tool_call.id] = res
+                        except Exception as e:
+                            print(f"Error calling tools, {str(e)}")
+                            traceback.print_exc()
+
+                tool_outputs=[{"tool_call_id": k, "output": v} for (k,v) in outputs.items()]
+                print(f"Finished tools calling: {tool_outputs}")
+                run = st.session_state.openai_client.beta.threads.runs.submit_tool_outputs(
+                    thread_id=thread.id,
+                    run_id=run.id,
+                    tool_outputs=tool_outputs,
+                )
+                print(run_status.required_action)
+                #return run_status
+            else:
+                print(f"Unknown required action: {required_action.type}")
+                return run_status
         else:
             print(f"Run status: {run_status.status}")
             return run_status