diff --git a/src/llama_stack/apis/conversations/__init__.py b/src/llama_stack/apis/conversations/__init__.py index 2d214d27a..b6ddc5999 100644 --- a/src/llama_stack/apis/conversations/__init__.py +++ b/src/llama_stack/apis/conversations/__init__.py @@ -6,26 +6,22 @@ from .conversations import ( Conversation, - ConversationCreateRequest, ConversationDeletedResource, ConversationItem, ConversationItemCreateRequest, ConversationItemDeletedResource, ConversationItemList, Conversations, - ConversationUpdateRequest, Metadata, ) __all__ = [ "Conversation", - "ConversationCreateRequest", "ConversationDeletedResource", "ConversationItem", "ConversationItemCreateRequest", "ConversationItemDeletedResource", "ConversationItemList", "Conversations", - "ConversationUpdateRequest", "Metadata", ] diff --git a/src/llama_stack/apis/conversations/conversations.py b/src/llama_stack/apis/conversations/conversations.py index d75683efa..6ec7e67d6 100644 --- a/src/llama_stack/apis/conversations/conversations.py +++ b/src/llama_stack/apis/conversations/conversations.py @@ -102,32 +102,6 @@ register_schema(ConversationItem, name="ConversationItem") # ] -@json_schema_type -class ConversationCreateRequest(BaseModel): - """Request body for creating a conversation.""" - - items: list[ConversationItem] | None = Field( - default=[], - description="Initial items to include in the conversation context. You may add up to 20 items at a time.", - max_length=20, - ) - metadata: Metadata | None = Field( - default={}, - description="Set of 16 key-value pairs that can be attached to an object. Useful for storing additional information", - max_length=16, - ) - - -@json_schema_type -class ConversationUpdateRequest(BaseModel): - """Request body for updating a conversation.""" - - metadata: Metadata = Field( - ..., - description="Set of 16 key-value pairs that can be attached to an object. This can be useful for storing additional information about the object in a structured format, and querying for objects via API or the dashboard. Keys are strings with a maximum length of 64 characters. Values are strings with a maximum length of 512 characters.", - ) - - @json_schema_type class ConversationDeletedResource(BaseModel): """Response for deleted conversation.""" diff --git a/tests/unit/conversations/test_api_models.py b/tests/unit/conversations/test_api_models.py index 0e52778b8..8416cba0b 100644 --- a/tests/unit/conversations/test_api_models.py +++ b/tests/unit/conversations/test_api_models.py @@ -7,18 +7,11 @@ from llama_stack.apis.conversations.conversations import ( Conversation, - ConversationCreateRequest, ConversationItem, ConversationItemList, ) -def test_conversation_create_request_defaults(): - request = ConversationCreateRequest() - assert request.items == [] - assert request.metadata == {} - - def test_conversation_model_defaults(): conversation = Conversation( id="conv_123456789",