Fix tests for GPT assistant (microsoft#1505)

* fixed tests for gpt assistant * fixed tests for gpt assistant * api_type openai * tests updated --------- Co-authored-by: Chi Wang <[email protected]>
meirm · Feb 2, 2024 · b0f42a1 · b0f42a1
1 parent 015070b
commit b0f42a1
Show file tree

Hide file tree

Showing 4 changed files with 239 additions and 189 deletions.
diff --git a/autogen/oai/client.py b/autogen/oai/client.py
@@ -90,13 +90,12 @@ def __init__(self, *, config_list: Optional[List[Dict[str, Any]]] = None, **base
             {
                 "model": "gpt-3.5-turbo",
                 "api_key": os.environ.get("OPENAI_API_KEY"),
-                "api_type": "open_ai",
+                "api_type": "openai",
                 "base_url": "https://api.openai.com/v1",
             },
             {
                 "model": "llama-7B",
                 "base_url": "http://127.0.0.1:8080",
-                "api_type": "open_ai",
             }
         ]
         ```

diff --git a/autogen/oai/completion.py b/autogen/oai/completion.py
@@ -744,13 +744,13 @@ def create(
                 {
                     "model": "gpt-3.5-turbo",
                     "api_key": os.environ.get("OPENAI_API_KEY"),
-                    "api_type": "open_ai",
+                    "api_type": "openai",
                     "base_url": "https://api.openai.com/v1",
                 },
                 {
                     "model": "llama-7B",
                     "base_url": "http://127.0.0.1:8080",
-                    "api_type": "open_ai",
+                    "api_type": "openai",
                 }
             ],
             prompt="Hi",