Update nested_resource_class_methods.py (#612)

support api-base in _nested_resource_class_methods
openai · Sep 22, 2023 · 7fba4dc · 7fba4dc
1 parent 3f6ef06
commit 7fba4dc
Showing 1 changed file with 20 additions and 18 deletions.
diff --git a/openai/api_resources/abstract/nested_resource_class_methods.py b/openai/api_resources/abstract/nested_resource_class_methods.py
@@ -28,17 +28,18 @@ def nested_resource_url(cls, id, nested_id=None):
         setattr(cls, resource_url_method, classmethod(nested_resource_url))
 
         def nested_resource_request(
-            cls,
-            method,
-            url,
-            api_key=None,
-            request_id=None,
-            api_version=None,
-            organization=None,
-            **params,
+                cls,
+                method,
+                url,
+                api_base=None,
+                api_key=None,
+                request_id=None,
+                api_version=None,
+                organization=None,
+                **params,
         ):
             requestor = api_requestor.APIRequestor(
-                api_key, api_version=api_version, organization=organization
+                api_key, api_base=api_base, api_version=api_version, organization=organization
             )
             response, _, api_key = requestor.request(
                 method, url, params, request_id=request_id
@@ -48,17 +49,18 @@ def nested_resource_request(
             )
 
         async def anested_resource_request(
-            cls,
-            method,
-            url,
-            api_key=None,
-            request_id=None,
-            api_version=None,
-            organization=None,
-            **params,
+                cls,
+                method,
+                url,
+                api_key=None,
+                api_base=None,
+                request_id=None,
+                api_version=None,
+                organization=None,
+                **params,
         ):
             requestor = api_requestor.APIRequestor(
-                api_key, api_version=api_version, organization=organization
+                api_key, api_base=api_base, api_version=api_version, organization=organization
             )
             response, _, api_key = await requestor.arequest(
                 method, url, params, request_id=request_id