Skip to content

Commit

Permalink
fix: linting
Browse files Browse the repository at this point in the history
  • Loading branch information
VisargD committed Jan 12, 2024
1 parent 0158a43 commit 509eb57
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions portkey_ai/api_resources/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -136,6 +138,7 @@ class DeltaToolCall(BaseModel):
function: Optional[DeltaToolCallFunction] = None
type: Optional[str] = None


class Message(TypedDict):
role: str
content: Optional[str]
Expand All @@ -147,6 +150,7 @@ class Function(BaseModel):
description: str
parameters: Dict[str, object]


class Tool(BaseModel):
function: Function
type: str
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 509eb57

Please sign in to comment.