fix for new openai 1.20.0 version

Author: alexeichhorn

typegpt/openai/_async/client.py

@@ -1,3 +1,4 @@
+import inspect
 from typing import Mapping
 
 import httpx
@@ -25,6 +26,7 @@ def __init__(
         *,
         api_key: str | None = None,
         organization: str | None = None,
+        project: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
         max_retries: int = DEFAULT_MAX_RETRIES,
@@ -35,18 +37,31 @@ def __init__(
         # only needed to have same subclass capabilities (i.e. for Azure)
         _strict_response_validation: bool = False,
     ) -> None:
-        super().__init__(
-            api_key=api_key,
-            organization=organization,
-            base_url=base_url,
-            timeout=timeout,
-            max_retries=max_retries,
-            default_headers=default_headers,
-            default_query=default_query,
-            http_client=http_client,
-        )
+        init_signature = inspect.signature(super.__init__)
+        if "project" in init_signature.parameters:  # openai version >= 1.20.0
+            super().__init__(
+                api_key=api_key,
+                organization=organization,
+                project=project,
+                base_url=base_url,
+                timeout=timeout,
+                max_retries=max_retries,
+                default_headers=default_headers,
+                default_query=default_query,
+                http_client=http_client,
+            )
+        else:
+            super().__init__(
+                api_key=api_key,
+                organization=organization,
+                base_url=base_url,
+                timeout=timeout,
+                max_retries=max_retries,
+                default_headers=default_headers,
+                default_query=default_query,
+                http_client=http_client,
+            )
         self.chat = AsyncTypeChat(self)
 
 
-class AsyncTypeAzureOpenAI(AsyncAzureOpenAI, AsyncTypeOpenAI):
-    ...
+class AsyncTypeAzureOpenAI(AsyncAzureOpenAI, AsyncTypeOpenAI): ...

typegpt/openai/_sync/client.py

@@ -1,3 +1,4 @@
+import inspect
 from typing import Mapping
 
 import httpx
@@ -25,6 +26,7 @@ def __init__(
         *,
         api_key: str | None = None,
         organization: str | None = None,
+        project: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
         max_retries: int = DEFAULT_MAX_RETRIES,
@@ -35,18 +37,31 @@ def __init__(
         # only needed to have same subclass capabilities (i.e. for Azure)
         _strict_response_validation: bool = False,
     ) -> None:
-        super().__init__(
-            api_key=api_key,
-            organization=organization,
-            base_url=base_url,
-            timeout=timeout,
-            max_retries=max_retries,
-            default_headers=default_headers,
-            default_query=default_query,
-            http_client=http_client,
-        )
+        init_signature = inspect.signature(super.__init__)
+        if "project" in init_signature.parameters:  # openai version >= 1.20.0
+            super().__init__(
+                api_key=api_key,
+                organization=organization,
+                project=project,
+                base_url=base_url,
+                timeout=timeout,
+                max_retries=max_retries,
+                default_headers=default_headers,
+                default_query=default_query,
+                http_client=http_client,
+            )
+        else:
+            super().__init__(
+                api_key=api_key,
+                organization=organization,
+                base_url=base_url,
+                timeout=timeout,
+                max_retries=max_retries,
+                default_headers=default_headers,
+                default_query=default_query,
+                http_client=http_client,
+            )
         self.chat = TypeChat(self)
 
 
-class TypeAzureOpenAI(AzureOpenAI, TypeOpenAI):
-    ...
+class TypeAzureOpenAI(AzureOpenAI, TypeOpenAI): ...