fix: openai async client bug #130

Update openai.py
OpenBMB · Apr 25, 2024 · 4dd772d · 4dd772d
2 parents 6e6dff6 + c755193
commit 4dd772d
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/agentverse/llms/openai.py b/agentverse/llms/openai.py
@@ -232,11 +232,12 @@ def generate_response(
         try:
             # Execute function call
             if functions != []:
-                response = openai_client.chat.completions.create(
-                    messages=messages,
-                    functions=functions,
-                    **self.args.dict(),
-                )
+                async with async_openai_client:   
+                    response = openai_client.chat.completions.create(
+                        messages=messages,
+                        functions=functions,
+                        **self.args.dict(),
+                    )
 
                 logger.log_prompt(
                     [
@@ -275,10 +276,11 @@ def generate_response(
                     )
 
             else:
-                response = openai_client.chat.completions.create(
-                    messages=messages,
-                    **self.args.dict(),
-                )
+                async with async_openai_client:   
+                    response = openai_client.chat.completions.create(
+                        messages=messages,
+                        **self.args.dict(),
+                    )
                 logger.log_prompt(
                     [
                         {