Reset client on setting api key

VRSEN · Jan 29, 2024 · 6bbd3c7 · 6bbd3c7
1 parent 80eaadf
commit 6bbd3c7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/agency_swarm/util/oai.py b/agency_swarm/util/oai.py
@@ -34,3 +34,6 @@ def set_openai_key(key):
     if not key:
         raise ValueError("Invalid API key. The API key cannot be empty.")
     openai.api_key = key
+    global client
+    with client_lock:
+        client = None