fix: chat schema

etalab-ia · Oct 25, 2024 · 2ebe5ea · 2ebe5ea
1 parent b6d2b81
commit 2ebe5ea
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/app/schemas/chat.py b/app/schemas/chat.py
@@ -4,8 +4,6 @@
     ChatCompletion,
     ChatCompletionChunk,
     ChatCompletionMessageParam,
-    ChatCompletionToolChoiceOptionParam,
-    ChatCompletionToolParam,
 )
 from pydantic import BaseModel, Field, model_validator
 
@@ -28,8 +26,6 @@ class ChatCompletionRequest(BaseModel):
     user: Optional[str] = None
     seed: Optional[int] = None
     stop: Union[Optional[str], List[str]] = Field(default_factory=list)
-    tool_choice: Optional[Union[Literal["none"], ChatCompletionToolChoiceOptionParam]] = "none"
-    tools: List[ChatCompletionToolParam] = None
     user: Optional[str] = None
     best_of: Optional[int] = None
     top_k: int = -1