feat: debug, cache_force_refresh, custom_host, forward_headers header…

…s added
Portkey-AI · May 31, 2024 · f706e7f · f706e7f
1 parent a5224ed
commit f706e7f
Show file tree

Hide file tree

Showing 3 changed files with 59 additions and 0 deletions.
diff --git a/portkey_ai/api_resources/base_client.py b/portkey_ai/api_resources/base_client.py
@@ -54,6 +54,10 @@ def __init__(
         provider: Optional[str] = None,
         trace_id: Optional[str] = None,
         metadata: Optional[str] = None,
+        debug: Optional[bool] = None,
+        cache_force_refresh: Optional[bool] = None,
+        custom_host: Optional[str] = None,
+        forward_headers: Optional[str] = None,
         openai_project: Optional[str] = None,
         openai_organization: Optional[str] = None,
         aws_secret_access_key: Optional[str] = None,
@@ -75,6 +79,10 @@ def __init__(
         self.provider = provider
         self.trace_id = trace_id
         self.metadata = metadata
+        self.debug = debug
+        self.cache_force_refresh = cache_force_refresh
+        self.custom_host = custom_host
+        self.forward_headers = forward_headers
         self.openai_project = openai_project
         self.openai_organization = openai_organization
         self.aws_secret_access_key=aws_secret_access_key
@@ -96,6 +104,10 @@ def __init__(
             provider=provider,
             trace_id=trace_id,
             metadata=metadata,
+            debug=debug,
+            cache_force_refresh=cache_force_refresh,
+            custom_host=custom_host,
+            forward_headers=forward_headers,
             openai_project=openai_project,
             openai_organization=openai_organization,
             aws_secret_access_key=aws_secret_access_key,
@@ -439,6 +451,10 @@ def __init__(
         provider: Optional[str] = None,
         trace_id: Optional[str] = None,
         metadata: Optional[str] = None,
+        debug: Optional[bool] = None,
+        cache_force_refresh: Optional[bool] = None,
+        custom_host: Optional[str] = None,
+        forward_headers: Optional[str] = None,
         openai_project: Optional[str] = None,
         openai_organization: Optional[str] = None,
         aws_secret_access_key: Optional[str] = None,
@@ -460,6 +476,10 @@ def __init__(
         self.provider = provider
         self.trace_id = trace_id
         self.metadata = metadata
+        self.debug = debug
+        self.cache_force_refresh = cache_force_refresh
+        self.custom_host = custom_host
+        self.forward_headers = forward_headers
         self.openai_project = openai_project
         self.openai_organization = openai_organization
         self.aws_secret_access_key=aws_secret_access_key
@@ -481,6 +501,10 @@ def __init__(
             provider=provider,
             trace_id=trace_id,
             metadata=metadata,
+            debug=debug,
+            cache_force_refresh=cache_force_refresh,
+            custom_host=custom_host,
+            forward_headers=forward_headers,
             openai_project=openai_project,
             openai_organization=openai_organization,
             aws_secret_access_key=aws_secret_access_key,

diff --git a/portkey_ai/api_resources/client.py b/portkey_ai/api_resources/client.py
@@ -38,6 +38,10 @@ def __init__(
         provider: Optional[str] = None,
         trace_id: Optional[str] = None,
         metadata: Optional[str] = None,
+        debug: Optional[bool] = None,
+        cache_force_refresh: Optional[bool] = None,
+        custom_host: Optional[str] = None,
+        forward_headers: Optional[str] = None,
         openai_project: Optional[str] = None,
         openai_organization: Optional[str] = None,
         aws_secret_access_key: Optional[str] = None,
@@ -60,6 +64,10 @@ def __init__(
             provider=provider,
             trace_id=trace_id,
             metadata=metadata,
+            debug=debug,
+            cache_force_refresh=cache_force_refresh,
+            custom_host=custom_host,
+            forward_headers=forward_headers,
             openai_project=openai_project,
             openai_organization=openai_organization,
             aws_secret_access_key=aws_secret_access_key,
@@ -102,6 +110,10 @@ def copy(
         provider: Optional[str] = None,
         trace_id: Optional[str] = None,
         metadata: Optional[str] = None,
+        debug: Optional[bool] = None,
+        cache_force_refresh: Optional[bool] = None,
+        custom_host: Optional[str] = None,
+        forward_headers: Optional[str] = None,
         openai_project: Optional[str] = None,
         openai_organization: Optional[str] = None,
         aws_secret_access_key: Optional[str] = None,
@@ -124,6 +136,10 @@ def copy(
             provider=provider or self.provider,
             trace_id=trace_id or self.trace_id,
             metadata=metadata or self.metadata,
+            debug=debug or self.debug,
+            cache_force_refresh=cache_force_refresh or self.cache_force_refresh,
+            custom_host=custom_host or self.custom_host,
+            forward_headers=forward_headers or self.forward_headers,
             openai_project=openai_project or self.openai_project,
             openai_organization=openai_organization or self.openai_organization,
             aws_secret_access_key=aws_secret_access_key or self.aws_secret_access_key,
@@ -174,6 +190,10 @@ def __init__(
         provider: Optional[str] = None,
         trace_id: Optional[str] = None,
         metadata: Optional[str] = None,
+        debug: Optional[bool] = None,
+        cache_force_refresh: Optional[bool] = None,
+        custom_host: Optional[str] = None,
+        forward_headers: Optional[str] = None,
         openai_project: Optional[str] = None,
         openai_organization: Optional[str] = None,
         aws_secret_access_key: Optional[str] = None,
@@ -196,6 +216,10 @@ def __init__(
             provider=provider,
             trace_id=trace_id,
             metadata=metadata,
+            debug=debug,
+            cache_force_refresh=cache_force_refresh,
+            custom_host=custom_host,
+            forward_headers=forward_headers,
             openai_project=openai_project,
             openai_organization=openai_organization,
             aws_secret_access_key=aws_secret_access_key,
@@ -238,6 +262,10 @@ def copy(
         provider: Optional[str] = None,
         trace_id: Optional[str] = None,
         metadata: Optional[str] = None,
+        debug: Optional[bool] = None,
+        cache_force_refresh: Optional[bool] = None,
+        custom_host: Optional[str] = None,
+        forward_headers: Optional[str] = None,
         openai_project: Optional[str] = None,
         openai_organization: Optional[str] = None,
         aws_secret_access_key: Optional[str] = None,
@@ -260,6 +288,10 @@ def copy(
             provider=provider or self.provider,
             trace_id=trace_id or self.trace_id,
             metadata=metadata or self.metadata,
+            debug=debug or self.debug,
+            cache_force_refresh=cache_force_refresh or self.cache_force_refresh,
+            custom_host=custom_host or self.custom_host,
+            forward_headers=forward_headers or self.forward_headers,
             openai_project=openai_project or self.openai_project,
             openai_organization=openai_organization or self.openai_organization,
             aws_secret_access_key=aws_secret_access_key or self.aws_secret_access_key,

diff --git a/portkey_ai/api_resources/utils.py b/portkey_ai/api_resources/utils.py
@@ -243,6 +243,9 @@ class Constructs(BaseModel):
     cache_force_refresh: Optional[bool] = None
     trace_id: Optional[str] = None
     metadata: Optional[Dict[str, Any]] = None
+    debug: Optional[bool] = None
+    custom_host: Optional[str] = None
+    forward_headers:Optional[str] = None
     weight: Optional[float] = None
     retry: Optional[RetrySettings] = None
     deployment_id: Optional[str] = None