fix: linting issues

Portkey-AI · Jul 31, 2024 · 4bf1949 · 4bf1949
1 parent 68b0e65
commit 4bf1949
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/portkey_ai/api_resources/client.py b/portkey_ai/api_resources/client.py
@@ -182,7 +182,8 @@ def copy(
             cache_namespace=cache_namespace or self.cache_namespace,
             http_client=http_client or self._client,
             request_timeout=request_timeout or self.request_timeout,
-            strict_open_ai_compliance=strict_open_ai_compliance or self.strict_open_ai_compliance,
+            strict_open_ai_compliance=strict_open_ai_compliance
+            or self.strict_open_ai_compliance,
             **self.kwargs,
             **kwargs,
         )
@@ -363,7 +364,8 @@ def copy(
             cache_namespace=cache_namespace or self.cache_namespace,
             http_client=http_client or self._client,
             request_timeout=request_timeout or self.request_timeout,
-            strict_open_ai_compliance=strict_open_ai_compliance or self.strict_open_ai_compliance,
+            strict_open_ai_compliance=strict_open_ai_compliance
+            or self.strict_open_ai_compliance,
             **self.kwargs,
             **kwargs,
         )

diff --git a/portkey_ai/api_resources/types/chat_complete_type.py b/portkey_ai/api_resources/types/chat_complete_type.py
@@ -96,14 +96,14 @@ class ChoiceLogprobs(BaseModel, extra="allow"):
     content: Optional[List[ChatCompletionTokenLogprob]] = None
 
 
-class Choice(BaseModel, extra = "allow"):
+class Choice(BaseModel, extra="allow"):
     finish_reason: Optional[str]
     index: Optional[int]
     logprobs: Optional[ChoiceLogprobs] = None
     message: Optional[ChatCompletionMessage]
 
 
-class ChatCompletions(BaseModel, extra = "allow"):
+class ChatCompletions(BaseModel, extra="allow"):
     id: Optional[str]
     choices: Optional[List[Choice]]
     created: Optional[int]