From c755193f3bd68ca1c3cde8f3b0b3c331726a016d Mon Sep 17 00:00:00 2001
From: _1ring2rta_ <53417412+1ring2rta@users.noreply.github.com>
Date: Thu, 25 Apr 2024 09:20:58 +0800
Subject: [PATCH] Update openai.py

fix ``Error message is logged: RuntimeError: Event loop is closed``
---
 agentverse/llms/openai.py | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/agentverse/llms/openai.py b/agentverse/llms/openai.py
index 144e19d3..881ec3f7 100644
--- 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(
                     [
                         {