Fix: organization error

caszhang · Nov 25, 2023 · 312c0b2 · 312c0b2
1 parent 4d60a06
commit 312c0b2
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/XAgent/ai_functions/request/openai.py b/XAgent/ai_functions/request/openai.py
@@ -134,19 +134,28 @@ def chatcompletion_request(**kwargs):
             azure_endpoint = chatcompletion_kwargs.pop("azure_endpoint", None)
             api_version = chatcompletion_kwargs.pop("api_version", None)
             api_key = chatcompletion_kwargs.pop("api_key", None)
+            organization = chatcompletion_kwargs.pop("organization", None)
             chatcompletion_kwargs.update(kwargs)
             client = openai.AzureOpenAI(
-                api_key=api_key, azure_endpoint=azure_endpoint, api_version=api_version, timeout=request_timeout
+                api_key=api_key, 
+                organization=organization,
+                azure_endpoint=azure_endpoint, 
+                api_version=api_version, 
+                timeout=request_timeout
             )
         else:
             if "base_url" in chatcompletion_kwargs:
                 base_url = chatcompletion_kwargs.pop("base_url", None)
             else:
                 base_url = chatcompletion_kwargs.pop("api_base", None)
             api_key = chatcompletion_kwargs.pop("api_key", None)
+            organization = chatcompletion_kwargs.pop("organization", None)
             chatcompletion_kwargs.update(kwargs)
             client = openai.OpenAI(
-                api_key=api_key, base_url=base_url, timeout=request_timeout
+                api_key=api_key, 
+                organization=organization,
+                base_url=base_url, 
+                timeout=request_timeout
             )
         try:
             completions = client.chat.completions.create(**chatcompletion_kwargs)