diff --git a/portkey_ai/api_resources/base_client.py b/portkey_ai/api_resources/base_client.py
index 7c76372..0b0dca1 100644
--- 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
index e08a93b..0a62dcf 100644
--- 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__(