From 3ae49b719157d83e0d8c80eb772f2b00778167bb Mon Sep 17 00:00:00 2001 From: Narendranath Gogineni Date: Tue, 27 Aug 2024 16:00:18 +0530 Subject: [PATCH] fix: Add huggi ngface_base_url parameter to support autocompletions for x-portkey-huggingface-base-url header --- portkey_ai/api_resources/base_client.py | 6 ++++++ portkey_ai/api_resources/client.py | 8 ++++++++ portkey_ai/api_resources/utils.py | 1 + 3 files changed, 15 insertions(+) diff --git a/portkey_ai/api_resources/base_client.py b/portkey_ai/api_resources/base_client.py index a628f68..a660f10 100644 --- a/portkey_ai/api_resources/base_client.py +++ b/portkey_ai/api_resources/base_client.py @@ -72,6 +72,7 @@ def __init__( azure_resource_name: Optional[str] = None, azure_deployment_id: Optional[str] = None, azure_api_version: Optional[str] = None, + huggingface_base_url: Optional[str] = None, http_client: Optional[httpx.Client] = None, request_timeout: Optional[int] = None, strict_open_ai_compliance: Optional[bool] = None, @@ -100,6 +101,7 @@ def __init__( self.azure_resource_name = azure_resource_name self.azure_deployment_id = azure_deployment_id self.azure_api_version = azure_api_version + self.huggingface_base_url = huggingface_base_url self.cache_namespace = cache_namespace self.request_timeout = request_timeout self.strict_open_ai_compliance = strict_open_ai_compliance @@ -128,6 +130,7 @@ def __init__( azure_resource_name=azure_resource_name, azure_deployment_id=azure_deployment_id, azure_api_version=azure_api_version, + huggingface_base_url=huggingface_base_url, cache_namespace=cache_namespace, request_timeout=request_timeout, strict_open_ai_compliance=strict_open_ai_compliance, @@ -559,6 +562,7 @@ def __init__( azure_resource_name: Optional[str] = None, azure_deployment_id: Optional[str] = None, azure_api_version: Optional[str] = None, + huggingface_base_url: Optional[str] = None, http_client: Optional[httpx.AsyncClient] = None, request_timeout: Optional[int] = None, strict_open_ai_compliance: Optional[bool] = None, @@ -587,6 +591,7 @@ def __init__( self.azure_resource_name = azure_resource_name self.azure_deployment_id = azure_deployment_id self.azure_api_version = azure_api_version + self.huggingface_base_url = huggingface_base_url self.cache_namespace = cache_namespace self.request_timeout = request_timeout self.strict_open_ai_compliance = strict_open_ai_compliance @@ -615,6 +620,7 @@ def __init__( azure_resource_name=azure_resource_name, azure_deployment_id=azure_deployment_id, azure_api_version=azure_api_version, + huggingface_base_url=huggingface_base_url, cache_namespace=cache_namespace, request_timeout=request_timeout, strict_open_ai_compliance=strict_open_ai_compliance, diff --git a/portkey_ai/api_resources/client.py b/portkey_ai/api_resources/client.py index 09621fa..9e4acf3 100644 --- a/portkey_ai/api_resources/client.py +++ b/portkey_ai/api_resources/client.py @@ -64,6 +64,7 @@ def __init__( azure_resource_name: Optional[str] = None, azure_deployment_id: Optional[str] = None, azure_api_version: Optional[str] = None, + huggingface_base_url: Optional[str] = None, http_client: Optional[httpx.Client] = None, request_timeout: Optional[int] = None, strict_open_ai_compliance: Optional[bool] = None, @@ -93,6 +94,7 @@ def __init__( azure_resource_name=azure_resource_name, azure_deployment_id=azure_deployment_id, azure_api_version=azure_api_version, + huggingface_base_url=huggingface_base_url, cache_namespace=cache_namespace, http_client=http_client, request_timeout=request_timeout, @@ -150,6 +152,7 @@ def copy( azure_resource_name: Optional[str] = None, azure_deployment_id: Optional[str] = None, azure_api_version: Optional[str] = None, + huggingface_base_url: Optional[str] = None, http_client: Optional[httpx.Client] = None, request_timeout: Optional[int] = None, strict_open_ai_compliance: Optional[bool] = None, @@ -179,6 +182,7 @@ def copy( azure_resource_name=azure_resource_name or self.azure_resource_name, azure_deployment_id=azure_deployment_id or self.azure_deployment_id, azure_api_version=azure_api_version or self.azure_api_version, + huggingface_base_url=huggingface_base_url or self.huggingface_base_url, cache_namespace=cache_namespace or self.cache_namespace, http_client=http_client or self._client, request_timeout=request_timeout or self.request_timeout, @@ -246,6 +250,7 @@ def __init__( azure_resource_name: Optional[str] = None, azure_deployment_id: Optional[str] = None, azure_api_version: Optional[str] = None, + huggingface_base_url: Optional[str] = None, http_client: Optional[httpx.AsyncClient] = None, request_timeout: Optional[int] = None, strict_open_ai_compliance: Optional[bool] = None, @@ -275,6 +280,7 @@ def __init__( azure_resource_name=azure_resource_name, azure_deployment_id=azure_deployment_id, azure_api_version=azure_api_version, + huggingface_base_url=huggingface_base_url, cache_namespace=cache_namespace, http_client=http_client, request_timeout=request_timeout, @@ -332,6 +338,7 @@ def copy( azure_resource_name: Optional[str] = None, azure_deployment_id: Optional[str] = None, azure_api_version: Optional[str] = None, + huggingface_base_url: Optional[str] = None, http_client: Optional[httpx.AsyncClient] = None, request_timeout: Optional[int] = None, strict_open_ai_compliance: Optional[bool] = None, @@ -361,6 +368,7 @@ def copy( azure_resource_name=azure_resource_name or self.azure_resource_name, azure_deployment_id=azure_deployment_id or self.azure_deployment_id, azure_api_version=azure_api_version or self.azure_api_version, + huggingface_base_url=huggingface_base_url or self.huggingface_base_url, cache_namespace=cache_namespace or self.cache_namespace, http_client=http_client or self._client, request_timeout=request_timeout or self.request_timeout, diff --git a/portkey_ai/api_resources/utils.py b/portkey_ai/api_resources/utils.py index 82780c7..951eec8 100644 --- a/portkey_ai/api_resources/utils.py +++ b/portkey_ai/api_resources/utils.py @@ -265,6 +265,7 @@ class Constructs(BaseModel): azure_resource_name: Optional[str] = None azure_deployment_id: Optional[str] = None azure_api_version: Optional[str] = None + huggingface_base_url: Optional[str] = None cache_namespace: Optional[str] = None request_timeout: Optional[int] = None