Skip to content

Commit 0970e16

Browse files
Dttbdjameszyao
authored andcommitted
feat: add models object property
1 parent 07c0c2f commit 0970e16

12 files changed

+12
-1
lines changed

taskingai/client/models/entities/action.py

+1
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121

2222

2323
class Action(BaseModel):
24+
object: str = Field("Action")
2425
action_id: str = Field(..., min_length=20, max_length=30)
2526
name: str = Field(..., min_length=1, max_length=128)
2627
operation_id: str = Field(...)

taskingai/client/models/entities/assistant.py

+1
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222

2323

2424
class Assistant(BaseModel):
25+
object: str = Field("Assistant")
2526
assistant_id: str = Field(..., min_length=20, max_length=30)
2627
model_id: str = Field(..., min_length=8, max_length=8)
2728
name: str = Field("", min_length=0, max_length=256)

taskingai/client/models/entities/chat.py

+1
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919

2020

2121
class Chat(BaseModel):
22+
object: str = Field("Chat")
2223
chat_id: str = Field(..., min_length=20, max_length=30)
2324
assistant_id: str = Field(..., min_length=20, max_length=30)
2425
name: str = Field("", min_length=0, max_length=127)

taskingai/client/models/entities/chat_completion.py

+1
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020

2121

2222
class ChatCompletion(BaseModel):
23+
object: str = Field("ChatCompletion")
2324
finish_reason: ChatCompletionFinishReason = Field(...)
2425
message: ChatCompletionAssistantMessage = Field(...)
2526
created_timestamp: int = Field(...)

taskingai/client/models/entities/chat_completion_chunk.py

+1
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818

1919

2020
class ChatCompletionChunk(BaseModel):
21+
object: str = Field("ChatCompletionChunk")
2122
role: ChatCompletionRole = Field("assistant")
2223
index: int = Field(...)
2324
delta: str = Field(...)

taskingai/client/models/entities/chunk.py

+1
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919

2020

2121
class Chunk(BaseModel):
22+
object: str = Field("Chunk")
2223
chunk_id: str = Field(..., min_length=20, max_length=30)
2324
record_id: Optional[str] = Field(..., min_length=20, max_length=30)
2425
collection_id: str = Field(..., min_length=20, max_length=30)

taskingai/client/models/entities/collection.py

+1
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919

2020

2121
class Collection(BaseModel):
22+
object: str = Field("Collection")
2223
collection_id: str = Field(..., min_length=24, max_length=24)
2324
name: str = Field("", min_length=0, max_length=256)
2425
description: str = Field("", min_length=0, max_length=512)

taskingai/client/models/entities/message.py

+1
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919

2020

2121
class Message(BaseModel):
22+
object: str = Field("Message")
2223
message_id: str = Field(..., min_length=20, max_length=30)
2324
chat_id: str = Field(..., min_length=20, max_length=30)
2425
assistant_id: str = Field(None, min_length=20, max_length=30)

taskingai/client/models/entities/message_chunk.py

+1
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818

1919

2020
class MessageChunk(BaseModel):
21+
object: str = Field("MessageChunk")
2122
role: MessageRole = Field(...)
2223
index: int = Field(...)
2324
delta: str = Field(...)

taskingai/client/models/entities/message_generation_log.py

+1
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919

2020

2121
class MessageGenerationLog(BaseModel):
22+
object: str = Field("MessageGenerationLog")
2223
session_id: str = Field(..., min_length=20, max_length=30)
2324
event: str = Field(...)
2425
event_id: str = Field(..., min_length=20, max_length=30)

taskingai/client/models/entities/record.py

+1
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020

2121

2222
class Record(BaseModel):
23+
object: str = Field("Record")
2324
record_id: str = Field(..., min_length=20, max_length=30)
2425
collection_id: str = Field(..., min_length=20, max_length=30)
2526
title: str = Field("", min_length=0, max_length=255)

taskingai/client/models/schemas/chat_completion_request.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,8 @@
1515
from typing import Optional, List, Dict, Union
1616
from ..entities.chat_completion_function_message import ChatCompletionFunctionMessage
1717
from ..entities.chat_completion_system_message import ChatCompletionSystemMessage
18-
from ..entities.chat_completion_user_message import ChatCompletionUserMessage
1918
from ..entities.chat_completion_assistant_message import ChatCompletionAssistantMessage
19+
from ..entities.chat_completion_user_message import ChatCompletionUserMessage
2020
from ..entities.chat_completion_function import ChatCompletionFunction
2121

2222
__all__ = ["ChatCompletionRequest"]

0 commit comments

Comments
 (0)