From 1cc9885f15d911bc249a74b100e20b97d4564433 Mon Sep 17 00:00:00 2001 From: csgulati09 Date: Tue, 14 May 2024 06:59:32 -0400 Subject: [PATCH 1/2] fix: type of metadata --- portkey_ai/api_resources/client.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/portkey_ai/api_resources/client.py b/portkey_ai/api_resources/client.py index d927906..5c75eff 100644 --- a/portkey_ai/api_resources/client.py +++ b/portkey_ai/api_resources/client.py @@ -37,7 +37,7 @@ def __init__( config: Optional[Union[Mapping, str]] = None, provider: Optional[str] = None, trace_id: Optional[str] = None, - metadata: Optional[str] = None, + metadata: Optional[dict[str, str]] = None, **kwargs, ) -> None: super().__init__( @@ -77,7 +77,7 @@ def copy( config: Optional[Union[Mapping, str]] = None, provider: Optional[str] = None, trace_id: Optional[str] = None, - metadata: Optional[str] = None, + metadata: Optional[dict[str, str]] = None, **kwargs, ) -> Portkey: return self.__class__( @@ -125,7 +125,7 @@ def __init__( config: Optional[Union[Mapping, str]] = None, provider: Optional[str] = None, trace_id: Optional[str] = None, - metadata: Optional[str] = None, + metadata: Optional[dict[str, str]] = None, **kwargs, ) -> None: super().__init__( @@ -165,7 +165,7 @@ def copy( config: Optional[Union[Mapping, str]] = None, provider: Optional[str] = None, trace_id: Optional[str] = None, - metadata: Optional[str] = None, + metadata: Optional[dict[str, str]] = None, **kwargs, ) -> AsyncPortkey: return self.__class__( From 95473390b7949cf789c9d426690c7afabf635407 Mon Sep 17 00:00:00 2001 From: csgulati09 Date: Tue, 14 May 2024 07:05:50 -0400 Subject: [PATCH 2/2] fix: metadata type in base_client --- portkey_ai/api_resources/base_client.py | 4 ++-- portkey_ai/api_resources/client.py | 8 ++++---- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/portkey_ai/api_resources/base_client.py b/portkey_ai/api_resources/base_client.py index b40b4c9..0f6ef6f 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() @@ -402,7 +402,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 5c75eff..9c0bba1 100644 --- a/portkey_ai/api_resources/client.py +++ b/portkey_ai/api_resources/client.py @@ -37,7 +37,7 @@ def __init__( config: Optional[Union[Mapping, str]] = None, provider: Optional[str] = None, trace_id: Optional[str] = None, - metadata: Optional[dict[str, str]] = None, + metadata: Union[Optional[dict[str, str]], str] = None, **kwargs, ) -> None: super().__init__( @@ -77,7 +77,7 @@ def copy( config: Optional[Union[Mapping, str]] = None, provider: Optional[str] = None, trace_id: Optional[str] = None, - metadata: Optional[dict[str, str]] = None, + metadata: Union[Optional[dict[str, str]], str] = None, **kwargs, ) -> Portkey: return self.__class__( @@ -125,7 +125,7 @@ def __init__( config: Optional[Union[Mapping, str]] = None, provider: Optional[str] = None, trace_id: Optional[str] = None, - metadata: Optional[dict[str, str]] = None, + metadata: Union[Optional[dict[str, str]], str] = None, **kwargs, ) -> None: super().__init__( @@ -165,7 +165,7 @@ def copy( config: Optional[Union[Mapping, str]] = None, provider: Optional[str] = None, trace_id: Optional[str] = None, - metadata: Optional[dict[str, str]] = None, + metadata: Union[Optional[dict[str, str]], str] = None, **kwargs, ) -> AsyncPortkey: return self.__class__(