From 509eb57bb74f3e61f95cb6b800a7b6545c372d0e Mon Sep 17 00:00:00 2001 From: visargD Date: Fri, 12 Jan 2024 19:59:20 +0530 Subject: [PATCH] fix: linting --- portkey_ai/api_resources/utils.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/portkey_ai/api_resources/utils.py b/portkey_ai/api_resources/utils.py index 9138cd5d..a9e5cfce 100644 --- a/portkey_ai/api_resources/utils.py +++ b/portkey_ai/api_resources/utils.py @@ -120,11 +120,13 @@ class FunctionCall(BaseModel): name: str arguments: str + class ToolCall(BaseModel): id: str function: FunctionCall type: str + class DeltaToolCallFunction(BaseModel): arguments: Optional[str] = None name: Optional[str] = None @@ -136,6 +138,7 @@ class DeltaToolCall(BaseModel): function: Optional[DeltaToolCallFunction] = None type: Optional[str] = None + class Message(TypedDict): role: str content: Optional[str] @@ -147,6 +150,7 @@ class Function(BaseModel): description: str parameters: Dict[str, object] + class Tool(BaseModel): function: Function type: str @@ -332,6 +336,7 @@ def __getitem__(self, key): def get(self, key: str, default: Optional[Any] = None): return getattr(self, key, None) or default + class TopLogprob(BaseModel): token: str bytes: Optional[List[int]] = None