fix: Add huggi

ngface_base_url parameter to support autocompletions for x-portkey-huggingface-base-url header
Portkey-AI · Aug 27, 2024 · 3ae49b7 · 3ae49b7
1 parent 62cf6b2
commit 3ae49b7
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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