Merge pull request #130 from Portkey-AI/hotfix/metadata

Hotfix/metadata
Portkey-AI · May 31, 2024 · 4f84fe4 · 4f84fe4
2 parents a3e7f3f + 5aa28d6
commit 4f84fe4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/portkey_ai/api_resources/base_client.py b/portkey_ai/api_resources/base_client.py
@@ -53,7 +53,7 @@ def __init__(
         config: Optional[Union[Mapping, str]] = None,
         provider: Optional[str] = None,
         trace_id: Optional[str] = None,
-        metadata: Optional[str] = None,
+        metadata: Union[Optional[dict[str, str]], str] = None,
         **kwargs,
     ) -> None:
         self.api_key = api_key or default_api_key()
@@ -472,7 +472,7 @@ def __init__(
         config: Optional[Union[Mapping, str]] = None,
         provider: Optional[str] = None,
         trace_id: Optional[str] = None,
-        metadata: Optional[str] = None,
+        metadata: Union[Optional[dict[str, str]], str] = None,
         **kwargs,
     ) -> None:
         self.api_key = api_key or default_api_key()

diff --git a/portkey_ai/api_resources/client.py b/portkey_ai/api_resources/client.py
@@ -44,7 +44,7 @@ def __init__(
         config: Optional[Union[Mapping, str]] = None,
         provider: Optional[str] = None,
         trace_id: Optional[str] = None,
-        metadata: Optional[str] = None,
+        metadata: Union[Optional[dict[str, str]], str] = None,
         **kwargs,
     ) -> None:
         super().__init__(
@@ -88,7 +88,7 @@ def copy(
         config: Optional[Union[Mapping, str]] = None,
         provider: Optional[str] = None,
         trace_id: Optional[str] = None,
-        metadata: Optional[str] = None,
+        metadata: Union[Optional[dict[str, str]], str] = None,
         **kwargs,
     ) -> Portkey:
         return self.__class__(
@@ -143,7 +143,7 @@ def __init__(
         config: Optional[Union[Mapping, str]] = None,
         provider: Optional[str] = None,
         trace_id: Optional[str] = None,
-        metadata: Optional[str] = None,
+        metadata: Union[Optional[dict[str, str]], str] = None,
         **kwargs,
     ) -> None:
         super().__init__(
@@ -187,7 +187,7 @@ def copy(
         config: Optional[Union[Mapping, str]] = None,
         provider: Optional[str] = None,
         trace_id: Optional[str] = None,
-        metadata: Optional[str] = None,
+        metadata: Union[Optional[dict[str, str]], str] = None,
         **kwargs,
     ) -> AsyncPortkey:
         return self.__class__(