Fixing logic

ai-forever · Mar 6, 2024 · 44d8165 · 44d8165
1 parent bb0d87b
commit 44d8165
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/gigachat/models/chat_completion_chunk.py b/src/gigachat/models/chat_completion_chunk.py
@@ -1,4 +1,4 @@
-from typing import List, Optional
+from typing import List
 
 from gigachat.models.choices_chunk import ChoicesChunk
 from gigachat.pydantic_v1 import BaseModel, Field
@@ -13,5 +13,5 @@ class ChatCompletionChunk(BaseModel):
     """Дата и время создания ответа в формате Unix time"""
     model: str
     """Название модели, которая вернула ответ"""
-    object_: object = Optional[Field(default=None, alias="object")]
+    object_: str = Field(alias="object")
     """Название вызываемого метода"""
diff --git a/src/gigachat/models/messages_chunk.py b/src/gigachat/models/messages_chunk.py
@@ -10,7 +10,7 @@ class MessagesChunk(BaseModel):
 
     role: Optional[MessagesRole] = None
     """Роль автора сообщения"""
-    content: Optional[str] = None
+    content: str
     """Текст сообщения"""
     function_call: Optional[FunctionCall] = None
     """Вызов функции"""