From b2491f30f2eda415728d8e961cad67a4b1fdc934 Mon Sep 17 00:00:00 2001 From: jameszyao Date: Mon, 11 Mar 2024 12:31:05 +0800 Subject: [PATCH] fix: add max_tokens to query_chunks api schema --- taskingai/client/apis/__init__.py | 1 - .../client/apis/api_bulk_create_actions.py | 11 ++++++--- taskingai/client/apis/api_chat_completion.py | 11 ++++++--- taskingai/client/apis/api_create_assistant.py | 11 ++++++--- taskingai/client/apis/api_create_chat.py | 13 ++++++++--- taskingai/client/apis/api_create_chunk.py | 13 ++++++++--- .../client/apis/api_create_collection.py | 11 ++++++--- taskingai/client/apis/api_create_message.py | 11 ++++++--- taskingai/client/apis/api_create_record.py | 13 ++++++++--- taskingai/client/apis/api_delete_action.py | 11 ++++++--- taskingai/client/apis/api_delete_assistant.py | 11 ++++++--- taskingai/client/apis/api_delete_chat.py | 13 ++++++++--- taskingai/client/apis/api_delete_chunk.py | 13 ++++++++--- .../client/apis/api_delete_collection.py | 11 ++++++--- taskingai/client/apis/api_delete_message.py | 15 +++++++++--- taskingai/client/apis/api_delete_record.py | 13 ++++++++--- taskingai/client/apis/api_generate_message.py | 11 ++++++--- taskingai/client/apis/api_get_action.py | 11 ++++++--- taskingai/client/apis/api_get_assistant.py | 11 ++++++--- taskingai/client/apis/api_get_chat.py | 13 ++++++++--- taskingai/client/apis/api_get_chunk.py | 13 ++++++++--- taskingai/client/apis/api_get_collection.py | 11 ++++++--- taskingai/client/apis/api_get_message.py | 15 +++++++++--- taskingai/client/apis/api_get_record.py | 13 ++++++++--- taskingai/client/apis/api_list_actions.py | 11 ++++++--- taskingai/client/apis/api_list_assistants.py | 11 ++++++--- taskingai/client/apis/api_list_chats.py | 13 ++++++++--- taskingai/client/apis/api_list_chunks.py | 13 ++++++++--- taskingai/client/apis/api_list_collections.py | 11 ++++++--- taskingai/client/apis/api_list_messages.py | 13 ++++++++--- taskingai/client/apis/api_list_records.py | 13 ++++++++--- .../apis/api_query_collection_chunks.py | 11 ++++++--- taskingai/client/apis/api_run_action.py | 13 ++++++++--- taskingai/client/apis/api_text_embedding.py | 11 ++++++--- taskingai/client/apis/api_update_action.py | 13 ++++++++--- taskingai/client/apis/api_update_assistant.py | 11 ++++++--- taskingai/client/apis/api_update_chat.py | 13 ++++++++--- taskingai/client/apis/api_update_chunk.py | 13 ++++++++--- .../client/apis/api_update_collection.py | 11 ++++++--- taskingai/client/apis/api_update_message.py | 13 ++++++++--- taskingai/client/apis/api_update_record.py | 11 ++++++--- taskingai/client/models/__init__.py | 1 - taskingai/client/models/entities/__init__.py | 1 - taskingai/client/models/entities/action.py | 1 - .../models/entities/action_authentication.py | 1 - .../entities/action_authentication_type.py | 1 - .../models/entities/action_body_type.py | 1 - .../client/models/entities/action_method.py | 1 - .../client/models/entities/action_param.py | 1 - taskingai/client/models/entities/assistant.py | 3 +-- .../models/entities/assistant_memory.py | 1 - .../models/entities/assistant_memory_type.py | 1 - taskingai/client/models/entities/chat.py | 1 - .../client/models/entities/chat_completion.py | 1 - .../chat_completion_assistant_message.py | 1 - .../models/entities/chat_completion_chunk.py | 1 - .../entities/chat_completion_finish_reason.py | 1 - .../entities/chat_completion_function.py | 1 - .../entities/chat_completion_function_call.py | 1 - .../chat_completion_function_message.py | 1 - .../chat_completion_function_parameters.py | 3 +-- ...completion_function_parameters_property.py | 1 - .../entities/chat_completion_message.py | 1 - .../models/entities/chat_completion_role.py | 1 - .../chat_completion_system_message.py | 1 - .../entities/chat_completion_user_message.py | 1 - .../client/models/entities/chat_memory.py | 1 - .../models/entities/chat_memory_message.py | 1 - taskingai/client/models/entities/chunk.py | 1 - .../client/models/entities/collection.py | 1 - taskingai/client/models/entities/message.py | 1 - .../client/models/entities/message_chunk.py | 1 - .../client/models/entities/message_content.py | 1 - .../models/entities/message_generation_log.py | 1 - .../client/models/entities/message_role.py | 1 - taskingai/client/models/entities/record.py | 1 - .../client/models/entities/record_type.py | 1 - .../models/entities/retrieval_config.py | 2 -- .../models/entities/retrieval_method.py | 1 - .../client/models/entities/retrieval_ref.py | 1 - .../client/models/entities/retrieval_type.py | 1 - .../client/models/entities/sort_order_enum.py | 1 - taskingai/client/models/entities/status.py | 1 - .../entities/text_embedding_input_type.py | 1 - .../models/entities/text_embedding_output.py | 1 - .../client/models/entities/text_splitter.py | 1 - .../models/entities/text_splitter_type.py | 1 - taskingai/client/models/entities/tool_ref.py | 1 - taskingai/client/models/entities/tool_type.py | 1 - taskingai/client/models/schemas/__init__.py | 2 +- .../schemas/action_bulk_create_request.py | 1 - .../schemas/action_bulk_create_response.py | 1 - .../models/schemas/action_get_response.py | 1 - .../models/schemas/action_list_request.py | 1 - .../models/schemas/action_list_response.py | 1 - .../models/schemas/action_run_request.py | 1 - .../models/schemas/action_run_response.py | 1 - .../models/schemas/action_update_request.py | 1 - .../models/schemas/action_update_response.py | 1 - .../schemas/assistant_create_request.py | 3 +-- .../schemas/assistant_create_response.py | 1 - .../models/schemas/assistant_get_response.py | 1 - .../models/schemas/assistant_list_request.py | 1 - .../models/schemas/assistant_list_response.py | 1 - .../schemas/assistant_update_request.py | 3 +-- .../schemas/assistant_update_response.py | 1 - .../models/schemas/base_data_response.py | 23 +++++++++++++++++++ .../models/schemas/base_empty_response.py | 1 - .../models/schemas/chat_completion_request.py | 11 ++++----- .../schemas/chat_completion_response.py | 1 - .../models/schemas/chat_create_request.py | 1 - .../models/schemas/chat_create_response.py | 1 - .../models/schemas/chat_get_response.py | 1 - .../models/schemas/chat_list_request.py | 1 - .../models/schemas/chat_list_response.py | 1 - .../models/schemas/chat_update_request.py | 1 - .../models/schemas/chat_update_response.py | 1 - .../models/schemas/chunk_create_request.py | 1 - .../models/schemas/chunk_create_response.py | 1 - .../models/schemas/chunk_get_response.py | 1 - .../models/schemas/chunk_list_request.py | 1 - .../models/schemas/chunk_list_response.py | 1 - .../models/schemas/chunk_query_request.py | 3 ++- .../models/schemas/chunk_query_response.py | 1 - .../models/schemas/chunk_update_request.py | 1 - .../models/schemas/chunk_update_response.py | 1 - .../schemas/collection_create_request.py | 1 - .../schemas/collection_create_response.py | 1 - .../models/schemas/collection_get_response.py | 1 - .../models/schemas/collection_list_request.py | 1 - .../schemas/collection_list_response.py | 1 - .../schemas/collection_update_request.py | 1 - .../schemas/collection_update_response.py | 1 - .../models/schemas/message_create_request.py | 1 - .../models/schemas/message_create_response.py | 1 - .../schemas/message_generate_request.py | 1 - .../schemas/message_generate_response.py | 1 - .../models/schemas/message_get_response.py | 1 - .../models/schemas/message_list_request.py | 1 - .../models/schemas/message_list_response.py | 1 - .../models/schemas/message_update_request.py | 1 - .../models/schemas/message_update_response.py | 1 - .../models/schemas/record_create_request.py | 1 - .../models/schemas/record_create_response.py | 1 - .../models/schemas/record_get_response.py | 1 - .../models/schemas/record_list_request.py | 1 - .../models/schemas/record_list_response.py | 1 - .../models/schemas/record_update_request.py | 1 - .../models/schemas/record_update_response.py | 1 - .../models/schemas/text_embedding_request.py | 1 - .../models/schemas/text_embedding_response.py | 1 - taskingai/retrieval/chunk.py | 6 +++++ 152 files changed, 405 insertions(+), 240 deletions(-) create mode 100644 taskingai/client/models/schemas/base_data_response.py diff --git a/taskingai/client/apis/__init__.py b/taskingai/client/apis/__init__.py index caaa634..cf6503e 100644 --- a/taskingai/client/apis/__init__.py +++ b/taskingai/client/apis/__init__.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/apis/api_bulk_create_actions.py b/taskingai/client/apis/api_bulk_create_actions.py index fbd4bf9..88e77b6 100644 --- a/taskingai/client/apis/api_bulk_create_actions.py +++ b/taskingai/client/apis/api_bulk_create_actions.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,10 @@ __all__ = ["api_bulk_create_actions", "async_api_bulk_create_actions"] -def api_bulk_create_actions(payload: ActionBulkCreateRequest, **kwargs) -> ActionBulkCreateResponse: +def api_bulk_create_actions( + payload: ActionBulkCreateRequest, + **kwargs, +) -> ActionBulkCreateResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -48,7 +50,10 @@ def api_bulk_create_actions(payload: ActionBulkCreateRequest, **kwargs) -> Actio ) -async def async_api_bulk_create_actions(payload: ActionBulkCreateRequest, **kwargs) -> ActionBulkCreateResponse: +async def async_api_bulk_create_actions( + payload: ActionBulkCreateRequest, + **kwargs, +) -> ActionBulkCreateResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_chat_completion.py b/taskingai/client/apis/api_chat_completion.py index 22e533e..226a9c1 100644 --- a/taskingai/client/apis/api_chat_completion.py +++ b/taskingai/client/apis/api_chat_completion.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,10 @@ __all__ = ["api_chat_completion", "async_api_chat_completion"] -def api_chat_completion(payload: ChatCompletionRequest, **kwargs) -> ChatCompletionResponse: +def api_chat_completion( + payload: ChatCompletionRequest, + **kwargs, +) -> ChatCompletionResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -48,7 +50,10 @@ def api_chat_completion(payload: ChatCompletionRequest, **kwargs) -> ChatComplet ) -async def async_api_chat_completion(payload: ChatCompletionRequest, **kwargs) -> ChatCompletionResponse: +async def async_api_chat_completion( + payload: ChatCompletionRequest, + **kwargs, +) -> ChatCompletionResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_create_assistant.py b/taskingai/client/apis/api_create_assistant.py index 0fa2e2b..121a86c 100644 --- a/taskingai/client/apis/api_create_assistant.py +++ b/taskingai/client/apis/api_create_assistant.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,10 @@ __all__ = ["api_create_assistant", "async_api_create_assistant"] -def api_create_assistant(payload: AssistantCreateRequest, **kwargs) -> AssistantCreateResponse: +def api_create_assistant( + payload: AssistantCreateRequest, + **kwargs, +) -> AssistantCreateResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -48,7 +50,10 @@ def api_create_assistant(payload: AssistantCreateRequest, **kwargs) -> Assistant ) -async def async_api_create_assistant(payload: AssistantCreateRequest, **kwargs) -> AssistantCreateResponse: +async def async_api_create_assistant( + payload: AssistantCreateRequest, + **kwargs, +) -> AssistantCreateResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_create_chat.py b/taskingai/client/apis/api_create_chat.py index bbbf141..a4d31bd 100644 --- a/taskingai/client/apis/api_create_chat.py +++ b/taskingai/client/apis/api_create_chat.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_create_chat", "async_api_create_chat"] -def api_create_chat(assistant_id: str, payload: ChatCreateRequest, **kwargs) -> ChatCreateResponse: +def api_create_chat( + assistant_id: str, + payload: ChatCreateRequest, + **kwargs, +) -> ChatCreateResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -50,7 +53,11 @@ def api_create_chat(assistant_id: str, payload: ChatCreateRequest, **kwargs) -> ) -async def async_api_create_chat(assistant_id: str, payload: ChatCreateRequest, **kwargs) -> ChatCreateResponse: +async def async_api_create_chat( + assistant_id: str, + payload: ChatCreateRequest, + **kwargs, +) -> ChatCreateResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_create_chunk.py b/taskingai/client/apis/api_create_chunk.py index e16518e..606e768 100644 --- a/taskingai/client/apis/api_create_chunk.py +++ b/taskingai/client/apis/api_create_chunk.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_create_chunk", "async_api_create_chunk"] -def api_create_chunk(collection_id: str, payload: ChunkCreateRequest, **kwargs) -> ChunkCreateResponse: +def api_create_chunk( + collection_id: str, + payload: ChunkCreateRequest, + **kwargs, +) -> ChunkCreateResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -50,7 +53,11 @@ def api_create_chunk(collection_id: str, payload: ChunkCreateRequest, **kwargs) ) -async def async_api_create_chunk(collection_id: str, payload: ChunkCreateRequest, **kwargs) -> ChunkCreateResponse: +async def async_api_create_chunk( + collection_id: str, + payload: ChunkCreateRequest, + **kwargs, +) -> ChunkCreateResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_create_collection.py b/taskingai/client/apis/api_create_collection.py index 1318f34..2b5b6ab 100644 --- a/taskingai/client/apis/api_create_collection.py +++ b/taskingai/client/apis/api_create_collection.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,10 @@ __all__ = ["api_create_collection", "async_api_create_collection"] -def api_create_collection(payload: CollectionCreateRequest, **kwargs) -> CollectionCreateResponse: +def api_create_collection( + payload: CollectionCreateRequest, + **kwargs, +) -> CollectionCreateResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -48,7 +50,10 @@ def api_create_collection(payload: CollectionCreateRequest, **kwargs) -> Collect ) -async def async_api_create_collection(payload: CollectionCreateRequest, **kwargs) -> CollectionCreateResponse: +async def async_api_create_collection( + payload: CollectionCreateRequest, + **kwargs, +) -> CollectionCreateResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_create_message.py b/taskingai/client/apis/api_create_message.py index 9818361..eacd956 100644 --- a/taskingai/client/apis/api_create_message.py +++ b/taskingai/client/apis/api_create_message.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -19,7 +18,10 @@ def api_create_message( - assistant_id: str, chat_id: str, payload: MessageCreateRequest, **kwargs + assistant_id: str, + chat_id: str, + payload: MessageCreateRequest, + **kwargs, ) -> MessageCreateResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -54,7 +56,10 @@ def api_create_message( async def async_api_create_message( - assistant_id: str, chat_id: str, payload: MessageCreateRequest, **kwargs + assistant_id: str, + chat_id: str, + payload: MessageCreateRequest, + **kwargs, ) -> MessageCreateResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_create_record.py b/taskingai/client/apis/api_create_record.py index 596f43d..26dd154 100644 --- a/taskingai/client/apis/api_create_record.py +++ b/taskingai/client/apis/api_create_record.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_create_record", "async_api_create_record"] -def api_create_record(collection_id: str, payload: RecordCreateRequest, **kwargs) -> RecordCreateResponse: +def api_create_record( + collection_id: str, + payload: RecordCreateRequest, + **kwargs, +) -> RecordCreateResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -50,7 +53,11 @@ def api_create_record(collection_id: str, payload: RecordCreateRequest, **kwargs ) -async def async_api_create_record(collection_id: str, payload: RecordCreateRequest, **kwargs) -> RecordCreateResponse: +async def async_api_create_record( + collection_id: str, + payload: RecordCreateRequest, + **kwargs, +) -> RecordCreateResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_delete_action.py b/taskingai/client/apis/api_delete_action.py index 5fcb04d..9a7b676 100644 --- a/taskingai/client/apis/api_delete_action.py +++ b/taskingai/client/apis/api_delete_action.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,10 @@ __all__ = ["api_delete_action", "async_api_delete_action"] -def api_delete_action(action_id: str, **kwargs) -> BaseEmptyResponse: +def api_delete_action( + action_id: str, + **kwargs, +) -> BaseEmptyResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -50,7 +52,10 @@ def api_delete_action(action_id: str, **kwargs) -> BaseEmptyResponse: ) -async def async_api_delete_action(action_id: str, **kwargs) -> BaseEmptyResponse: +async def async_api_delete_action( + action_id: str, + **kwargs, +) -> BaseEmptyResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_delete_assistant.py b/taskingai/client/apis/api_delete_assistant.py index 6f4dbec..c0463d6 100644 --- a/taskingai/client/apis/api_delete_assistant.py +++ b/taskingai/client/apis/api_delete_assistant.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,10 @@ __all__ = ["api_delete_assistant", "async_api_delete_assistant"] -def api_delete_assistant(assistant_id: str, **kwargs) -> BaseEmptyResponse: +def api_delete_assistant( + assistant_id: str, + **kwargs, +) -> BaseEmptyResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -50,7 +52,10 @@ def api_delete_assistant(assistant_id: str, **kwargs) -> BaseEmptyResponse: ) -async def async_api_delete_assistant(assistant_id: str, **kwargs) -> BaseEmptyResponse: +async def async_api_delete_assistant( + assistant_id: str, + **kwargs, +) -> BaseEmptyResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_delete_chat.py b/taskingai/client/apis/api_delete_chat.py index d635216..1ce8f83 100644 --- a/taskingai/client/apis/api_delete_chat.py +++ b/taskingai/client/apis/api_delete_chat.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_delete_chat", "async_api_delete_chat"] -def api_delete_chat(assistant_id: str, chat_id: str, **kwargs) -> BaseEmptyResponse: +def api_delete_chat( + assistant_id: str, + chat_id: str, + **kwargs, +) -> BaseEmptyResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -51,7 +54,11 @@ def api_delete_chat(assistant_id: str, chat_id: str, **kwargs) -> BaseEmptyRespo ) -async def async_api_delete_chat(assistant_id: str, chat_id: str, **kwargs) -> BaseEmptyResponse: +async def async_api_delete_chat( + assistant_id: str, + chat_id: str, + **kwargs, +) -> BaseEmptyResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_delete_chunk.py b/taskingai/client/apis/api_delete_chunk.py index 426b231..914b03f 100644 --- a/taskingai/client/apis/api_delete_chunk.py +++ b/taskingai/client/apis/api_delete_chunk.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_delete_chunk", "async_api_delete_chunk"] -def api_delete_chunk(collection_id: str, chunk_id: str, **kwargs) -> BaseEmptyResponse: +def api_delete_chunk( + collection_id: str, + chunk_id: str, + **kwargs, +) -> BaseEmptyResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -51,7 +54,11 @@ def api_delete_chunk(collection_id: str, chunk_id: str, **kwargs) -> BaseEmptyRe ) -async def async_api_delete_chunk(collection_id: str, chunk_id: str, **kwargs) -> BaseEmptyResponse: +async def async_api_delete_chunk( + collection_id: str, + chunk_id: str, + **kwargs, +) -> BaseEmptyResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_delete_collection.py b/taskingai/client/apis/api_delete_collection.py index cd873f2..3bf03b0 100644 --- a/taskingai/client/apis/api_delete_collection.py +++ b/taskingai/client/apis/api_delete_collection.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,10 @@ __all__ = ["api_delete_collection", "async_api_delete_collection"] -def api_delete_collection(collection_id: str, **kwargs) -> BaseEmptyResponse: +def api_delete_collection( + collection_id: str, + **kwargs, +) -> BaseEmptyResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -50,7 +52,10 @@ def api_delete_collection(collection_id: str, **kwargs) -> BaseEmptyResponse: ) -async def async_api_delete_collection(collection_id: str, **kwargs) -> BaseEmptyResponse: +async def async_api_delete_collection( + collection_id: str, + **kwargs, +) -> BaseEmptyResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_delete_message.py b/taskingai/client/apis/api_delete_message.py index ecd4be3..45692ab 100644 --- a/taskingai/client/apis/api_delete_message.py +++ b/taskingai/client/apis/api_delete_message.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,12 @@ __all__ = ["api_delete_message", "async_api_delete_message"] -def api_delete_message(assistant_id: str, chat_id: str, message_id: str, **kwargs) -> BaseEmptyResponse: +def api_delete_message( + assistant_id: str, + chat_id: str, + message_id: str, + **kwargs, +) -> BaseEmptyResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -52,7 +56,12 @@ def api_delete_message(assistant_id: str, chat_id: str, message_id: str, **kwarg ) -async def async_api_delete_message(assistant_id: str, chat_id: str, message_id: str, **kwargs) -> BaseEmptyResponse: +async def async_api_delete_message( + assistant_id: str, + chat_id: str, + message_id: str, + **kwargs, +) -> BaseEmptyResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_delete_record.py b/taskingai/client/apis/api_delete_record.py index ac34397..5f857c2 100644 --- a/taskingai/client/apis/api_delete_record.py +++ b/taskingai/client/apis/api_delete_record.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_delete_record", "async_api_delete_record"] -def api_delete_record(collection_id: str, record_id: str, **kwargs) -> BaseEmptyResponse: +def api_delete_record( + collection_id: str, + record_id: str, + **kwargs, +) -> BaseEmptyResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -51,7 +54,11 @@ def api_delete_record(collection_id: str, record_id: str, **kwargs) -> BaseEmpty ) -async def async_api_delete_record(collection_id: str, record_id: str, **kwargs) -> BaseEmptyResponse: +async def async_api_delete_record( + collection_id: str, + record_id: str, + **kwargs, +) -> BaseEmptyResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_generate_message.py b/taskingai/client/apis/api_generate_message.py index 7bcc98e..e3b395f 100644 --- a/taskingai/client/apis/api_generate_message.py +++ b/taskingai/client/apis/api_generate_message.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -19,7 +18,10 @@ def api_generate_message( - assistant_id: str, chat_id: str, payload: MessageGenerateRequest, **kwargs + assistant_id: str, + chat_id: str, + payload: MessageGenerateRequest, + **kwargs, ) -> MessageGenerateResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -54,7 +56,10 @@ def api_generate_message( async def async_api_generate_message( - assistant_id: str, chat_id: str, payload: MessageGenerateRequest, **kwargs + assistant_id: str, + chat_id: str, + payload: MessageGenerateRequest, + **kwargs, ) -> MessageGenerateResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_get_action.py b/taskingai/client/apis/api_get_action.py index 6f8f255..69566a6 100644 --- a/taskingai/client/apis/api_get_action.py +++ b/taskingai/client/apis/api_get_action.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,10 @@ __all__ = ["api_get_action", "async_api_get_action"] -def api_get_action(action_id: str, **kwargs) -> ActionGetResponse: +def api_get_action( + action_id: str, + **kwargs, +) -> ActionGetResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -50,7 +52,10 @@ def api_get_action(action_id: str, **kwargs) -> ActionGetResponse: ) -async def async_api_get_action(action_id: str, **kwargs) -> ActionGetResponse: +async def async_api_get_action( + action_id: str, + **kwargs, +) -> ActionGetResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_get_assistant.py b/taskingai/client/apis/api_get_assistant.py index 6b397e7..08179fd 100644 --- a/taskingai/client/apis/api_get_assistant.py +++ b/taskingai/client/apis/api_get_assistant.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,10 @@ __all__ = ["api_get_assistant", "async_api_get_assistant"] -def api_get_assistant(assistant_id: str, **kwargs) -> AssistantGetResponse: +def api_get_assistant( + assistant_id: str, + **kwargs, +) -> AssistantGetResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -50,7 +52,10 @@ def api_get_assistant(assistant_id: str, **kwargs) -> AssistantGetResponse: ) -async def async_api_get_assistant(assistant_id: str, **kwargs) -> AssistantGetResponse: +async def async_api_get_assistant( + assistant_id: str, + **kwargs, +) -> AssistantGetResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_get_chat.py b/taskingai/client/apis/api_get_chat.py index f108dc1..3a4089c 100644 --- a/taskingai/client/apis/api_get_chat.py +++ b/taskingai/client/apis/api_get_chat.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_get_chat", "async_api_get_chat"] -def api_get_chat(assistant_id: str, chat_id: str, **kwargs) -> ChatGetResponse: +def api_get_chat( + assistant_id: str, + chat_id: str, + **kwargs, +) -> ChatGetResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -51,7 +54,11 @@ def api_get_chat(assistant_id: str, chat_id: str, **kwargs) -> ChatGetResponse: ) -async def async_api_get_chat(assistant_id: str, chat_id: str, **kwargs) -> ChatGetResponse: +async def async_api_get_chat( + assistant_id: str, + chat_id: str, + **kwargs, +) -> ChatGetResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_get_chunk.py b/taskingai/client/apis/api_get_chunk.py index 2153ad7..69277a3 100644 --- a/taskingai/client/apis/api_get_chunk.py +++ b/taskingai/client/apis/api_get_chunk.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_get_chunk", "async_api_get_chunk"] -def api_get_chunk(collection_id: str, chunk_id: str, **kwargs) -> ChunkGetResponse: +def api_get_chunk( + collection_id: str, + chunk_id: str, + **kwargs, +) -> ChunkGetResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -51,7 +54,11 @@ def api_get_chunk(collection_id: str, chunk_id: str, **kwargs) -> ChunkGetRespon ) -async def async_api_get_chunk(collection_id: str, chunk_id: str, **kwargs) -> ChunkGetResponse: +async def async_api_get_chunk( + collection_id: str, + chunk_id: str, + **kwargs, +) -> ChunkGetResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_get_collection.py b/taskingai/client/apis/api_get_collection.py index fcd6f00..4d3fa5b 100644 --- a/taskingai/client/apis/api_get_collection.py +++ b/taskingai/client/apis/api_get_collection.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,10 @@ __all__ = ["api_get_collection", "async_api_get_collection"] -def api_get_collection(collection_id: str, **kwargs) -> CollectionGetResponse: +def api_get_collection( + collection_id: str, + **kwargs, +) -> CollectionGetResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -50,7 +52,10 @@ def api_get_collection(collection_id: str, **kwargs) -> CollectionGetResponse: ) -async def async_api_get_collection(collection_id: str, **kwargs) -> CollectionGetResponse: +async def async_api_get_collection( + collection_id: str, + **kwargs, +) -> CollectionGetResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_get_message.py b/taskingai/client/apis/api_get_message.py index ed1db76..277f716 100644 --- a/taskingai/client/apis/api_get_message.py +++ b/taskingai/client/apis/api_get_message.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,12 @@ __all__ = ["api_get_message", "async_api_get_message"] -def api_get_message(assistant_id: str, chat_id: str, message_id: str, **kwargs) -> MessageGetResponse: +def api_get_message( + assistant_id: str, + chat_id: str, + message_id: str, + **kwargs, +) -> MessageGetResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -52,7 +56,12 @@ def api_get_message(assistant_id: str, chat_id: str, message_id: str, **kwargs) ) -async def async_api_get_message(assistant_id: str, chat_id: str, message_id: str, **kwargs) -> MessageGetResponse: +async def async_api_get_message( + assistant_id: str, + chat_id: str, + message_id: str, + **kwargs, +) -> MessageGetResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_get_record.py b/taskingai/client/apis/api_get_record.py index 96097a8..dccb312 100644 --- a/taskingai/client/apis/api_get_record.py +++ b/taskingai/client/apis/api_get_record.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_get_record", "async_api_get_record"] -def api_get_record(collection_id: str, record_id: str, **kwargs) -> RecordGetResponse: +def api_get_record( + collection_id: str, + record_id: str, + **kwargs, +) -> RecordGetResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -51,7 +54,11 @@ def api_get_record(collection_id: str, record_id: str, **kwargs) -> RecordGetRes ) -async def async_api_get_record(collection_id: str, record_id: str, **kwargs) -> RecordGetResponse: +async def async_api_get_record( + collection_id: str, + record_id: str, + **kwargs, +) -> RecordGetResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_list_actions.py b/taskingai/client/apis/api_list_actions.py index aac945b..96a10e8 100644 --- a/taskingai/client/apis/api_list_actions.py +++ b/taskingai/client/apis/api_list_actions.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,10 @@ __all__ = ["api_list_actions", "async_api_list_actions"] -def api_list_actions(payload: ActionListRequest, **kwargs) -> ActionListResponse: +def api_list_actions( + payload: ActionListRequest, + **kwargs, +) -> ActionListResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -48,7 +50,10 @@ def api_list_actions(payload: ActionListRequest, **kwargs) -> ActionListResponse ) -async def async_api_list_actions(payload: ActionListRequest, **kwargs) -> ActionListResponse: +async def async_api_list_actions( + payload: ActionListRequest, + **kwargs, +) -> ActionListResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_list_assistants.py b/taskingai/client/apis/api_list_assistants.py index df32338..9ac1451 100644 --- a/taskingai/client/apis/api_list_assistants.py +++ b/taskingai/client/apis/api_list_assistants.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,10 @@ __all__ = ["api_list_assistants", "async_api_list_assistants"] -def api_list_assistants(payload: AssistantListRequest, **kwargs) -> AssistantListResponse: +def api_list_assistants( + payload: AssistantListRequest, + **kwargs, +) -> AssistantListResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -48,7 +50,10 @@ def api_list_assistants(payload: AssistantListRequest, **kwargs) -> AssistantLis ) -async def async_api_list_assistants(payload: AssistantListRequest, **kwargs) -> AssistantListResponse: +async def async_api_list_assistants( + payload: AssistantListRequest, + **kwargs, +) -> AssistantListResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_list_chats.py b/taskingai/client/apis/api_list_chats.py index 09eca78..bc203af 100644 --- a/taskingai/client/apis/api_list_chats.py +++ b/taskingai/client/apis/api_list_chats.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_list_chats", "async_api_list_chats"] -def api_list_chats(assistant_id: str, payload: ChatListRequest, **kwargs) -> ChatListResponse: +def api_list_chats( + assistant_id: str, + payload: ChatListRequest, + **kwargs, +) -> ChatListResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -50,7 +53,11 @@ def api_list_chats(assistant_id: str, payload: ChatListRequest, **kwargs) -> Cha ) -async def async_api_list_chats(assistant_id: str, payload: ChatListRequest, **kwargs) -> ChatListResponse: +async def async_api_list_chats( + assistant_id: str, + payload: ChatListRequest, + **kwargs, +) -> ChatListResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_list_chunks.py b/taskingai/client/apis/api_list_chunks.py index 84cf5eb..508136d 100644 --- a/taskingai/client/apis/api_list_chunks.py +++ b/taskingai/client/apis/api_list_chunks.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_list_chunks", "async_api_list_chunks"] -def api_list_chunks(collection_id: str, payload: ChunkListRequest, **kwargs) -> ChunkListResponse: +def api_list_chunks( + collection_id: str, + payload: ChunkListRequest, + **kwargs, +) -> ChunkListResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -50,7 +53,11 @@ def api_list_chunks(collection_id: str, payload: ChunkListRequest, **kwargs) -> ) -async def async_api_list_chunks(collection_id: str, payload: ChunkListRequest, **kwargs) -> ChunkListResponse: +async def async_api_list_chunks( + collection_id: str, + payload: ChunkListRequest, + **kwargs, +) -> ChunkListResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_list_collections.py b/taskingai/client/apis/api_list_collections.py index 514be1b..7a0b381 100644 --- a/taskingai/client/apis/api_list_collections.py +++ b/taskingai/client/apis/api_list_collections.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,10 @@ __all__ = ["api_list_collections", "async_api_list_collections"] -def api_list_collections(payload: CollectionListRequest, **kwargs) -> CollectionListResponse: +def api_list_collections( + payload: CollectionListRequest, + **kwargs, +) -> CollectionListResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -48,7 +50,10 @@ def api_list_collections(payload: CollectionListRequest, **kwargs) -> Collection ) -async def async_api_list_collections(payload: CollectionListRequest, **kwargs) -> CollectionListResponse: +async def async_api_list_collections( + payload: CollectionListRequest, + **kwargs, +) -> CollectionListResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_list_messages.py b/taskingai/client/apis/api_list_messages.py index 2e4d2f3..202476c 100644 --- a/taskingai/client/apis/api_list_messages.py +++ b/taskingai/client/apis/api_list_messages.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,12 @@ __all__ = ["api_list_messages", "async_api_list_messages"] -def api_list_messages(assistant_id: str, chat_id: str, payload: MessageListRequest, **kwargs) -> MessageListResponse: +def api_list_messages( + assistant_id: str, + chat_id: str, + payload: MessageListRequest, + **kwargs, +) -> MessageListResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -52,7 +56,10 @@ def api_list_messages(assistant_id: str, chat_id: str, payload: MessageListReque async def async_api_list_messages( - assistant_id: str, chat_id: str, payload: MessageListRequest, **kwargs + assistant_id: str, + chat_id: str, + payload: MessageListRequest, + **kwargs, ) -> MessageListResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_list_records.py b/taskingai/client/apis/api_list_records.py index 9785f82..9968f84 100644 --- a/taskingai/client/apis/api_list_records.py +++ b/taskingai/client/apis/api_list_records.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_list_records", "async_api_list_records"] -def api_list_records(collection_id: str, payload: RecordListRequest, **kwargs) -> RecordListResponse: +def api_list_records( + collection_id: str, + payload: RecordListRequest, + **kwargs, +) -> RecordListResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -50,7 +53,11 @@ def api_list_records(collection_id: str, payload: RecordListRequest, **kwargs) - ) -async def async_api_list_records(collection_id: str, payload: RecordListRequest, **kwargs) -> RecordListResponse: +async def async_api_list_records( + collection_id: str, + payload: RecordListRequest, + **kwargs, +) -> RecordListResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_query_collection_chunks.py b/taskingai/client/apis/api_query_collection_chunks.py index e838640..a927f6c 100644 --- a/taskingai/client/apis/api_query_collection_chunks.py +++ b/taskingai/client/apis/api_query_collection_chunks.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_query_collection_chunks", "async_api_query_collection_chunks"] -def api_query_collection_chunks(collection_id: str, payload: ChunkQueryRequest, **kwargs) -> ChunkQueryResponse: +def api_query_collection_chunks( + collection_id: str, + payload: ChunkQueryRequest, + **kwargs, +) -> ChunkQueryResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -51,7 +54,9 @@ def api_query_collection_chunks(collection_id: str, payload: ChunkQueryRequest, async def async_api_query_collection_chunks( - collection_id: str, payload: ChunkQueryRequest, **kwargs + collection_id: str, + payload: ChunkQueryRequest, + **kwargs, ) -> ChunkQueryResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_run_action.py b/taskingai/client/apis/api_run_action.py index c77d8da..7c4542e 100644 --- a/taskingai/client/apis/api_run_action.py +++ b/taskingai/client/apis/api_run_action.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_run_action", "async_api_run_action"] -def api_run_action(action_id: str, payload: ActionRunRequest, **kwargs) -> ActionRunResponse: +def api_run_action( + action_id: str, + payload: ActionRunRequest, + **kwargs, +) -> ActionRunResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -50,7 +53,11 @@ def api_run_action(action_id: str, payload: ActionRunRequest, **kwargs) -> Actio ) -async def async_api_run_action(action_id: str, payload: ActionRunRequest, **kwargs) -> ActionRunResponse: +async def async_api_run_action( + action_id: str, + payload: ActionRunRequest, + **kwargs, +) -> ActionRunResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_text_embedding.py b/taskingai/client/apis/api_text_embedding.py index 87560fa..e803f90 100644 --- a/taskingai/client/apis/api_text_embedding.py +++ b/taskingai/client/apis/api_text_embedding.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,10 @@ __all__ = ["api_text_embedding", "async_api_text_embedding"] -def api_text_embedding(payload: TextEmbeddingRequest, **kwargs) -> TextEmbeddingResponse: +def api_text_embedding( + payload: TextEmbeddingRequest, + **kwargs, +) -> TextEmbeddingResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -48,7 +50,10 @@ def api_text_embedding(payload: TextEmbeddingRequest, **kwargs) -> TextEmbedding ) -async def async_api_text_embedding(payload: TextEmbeddingRequest, **kwargs) -> TextEmbeddingResponse: +async def async_api_text_embedding( + payload: TextEmbeddingRequest, + **kwargs, +) -> TextEmbeddingResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_update_action.py b/taskingai/client/apis/api_update_action.py index 7ce12bc..7206239 100644 --- a/taskingai/client/apis/api_update_action.py +++ b/taskingai/client/apis/api_update_action.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_update_action", "async_api_update_action"] -def api_update_action(action_id: str, payload: ActionUpdateRequest, **kwargs) -> ActionUpdateResponse: +def api_update_action( + action_id: str, + payload: ActionUpdateRequest, + **kwargs, +) -> ActionUpdateResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -50,7 +53,11 @@ def api_update_action(action_id: str, payload: ActionUpdateRequest, **kwargs) -> ) -async def async_api_update_action(action_id: str, payload: ActionUpdateRequest, **kwargs) -> ActionUpdateResponse: +async def async_api_update_action( + action_id: str, + payload: ActionUpdateRequest, + **kwargs, +) -> ActionUpdateResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_update_assistant.py b/taskingai/client/apis/api_update_assistant.py index fe9d064..fdb0694 100644 --- a/taskingai/client/apis/api_update_assistant.py +++ b/taskingai/client/apis/api_update_assistant.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_update_assistant", "async_api_update_assistant"] -def api_update_assistant(assistant_id: str, payload: AssistantUpdateRequest, **kwargs) -> AssistantUpdateResponse: +def api_update_assistant( + assistant_id: str, + payload: AssistantUpdateRequest, + **kwargs, +) -> AssistantUpdateResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -51,7 +54,9 @@ def api_update_assistant(assistant_id: str, payload: AssistantUpdateRequest, **k async def async_api_update_assistant( - assistant_id: str, payload: AssistantUpdateRequest, **kwargs + assistant_id: str, + payload: AssistantUpdateRequest, + **kwargs, ) -> AssistantUpdateResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_update_chat.py b/taskingai/client/apis/api_update_chat.py index 454e7bc..95abbb1 100644 --- a/taskingai/client/apis/api_update_chat.py +++ b/taskingai/client/apis/api_update_chat.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,12 @@ __all__ = ["api_update_chat", "async_api_update_chat"] -def api_update_chat(assistant_id: str, chat_id: str, payload: ChatUpdateRequest, **kwargs) -> ChatUpdateResponse: +def api_update_chat( + assistant_id: str, + chat_id: str, + payload: ChatUpdateRequest, + **kwargs, +) -> ChatUpdateResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -52,7 +56,10 @@ def api_update_chat(assistant_id: str, chat_id: str, payload: ChatUpdateRequest, async def async_api_update_chat( - assistant_id: str, chat_id: str, payload: ChatUpdateRequest, **kwargs + assistant_id: str, + chat_id: str, + payload: ChatUpdateRequest, + **kwargs, ) -> ChatUpdateResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_update_chunk.py b/taskingai/client/apis/api_update_chunk.py index b6dda49..a6e70b9 100644 --- a/taskingai/client/apis/api_update_chunk.py +++ b/taskingai/client/apis/api_update_chunk.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,12 @@ __all__ = ["api_update_chunk", "async_api_update_chunk"] -def api_update_chunk(collection_id: str, chunk_id: str, payload: ChunkUpdateRequest, **kwargs) -> ChunkUpdateResponse: +def api_update_chunk( + collection_id: str, + chunk_id: str, + payload: ChunkUpdateRequest, + **kwargs, +) -> ChunkUpdateResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -52,7 +56,10 @@ def api_update_chunk(collection_id: str, chunk_id: str, payload: ChunkUpdateRequ async def async_api_update_chunk( - collection_id: str, chunk_id: str, payload: ChunkUpdateRequest, **kwargs + collection_id: str, + chunk_id: str, + payload: ChunkUpdateRequest, + **kwargs, ) -> ChunkUpdateResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_update_collection.py b/taskingai/client/apis/api_update_collection.py index a6aa2d1..6142c2c 100644 --- a/taskingai/client/apis/api_update_collection.py +++ b/taskingai/client/apis/api_update_collection.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -18,7 +17,11 @@ __all__ = ["api_update_collection", "async_api_update_collection"] -def api_update_collection(collection_id: str, payload: CollectionUpdateRequest, **kwargs) -> CollectionUpdateResponse: +def api_update_collection( + collection_id: str, + payload: CollectionUpdateRequest, + **kwargs, +) -> CollectionUpdateResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -51,7 +54,9 @@ def api_update_collection(collection_id: str, payload: CollectionUpdateRequest, async def async_api_update_collection( - collection_id: str, payload: CollectionUpdateRequest, **kwargs + collection_id: str, + payload: CollectionUpdateRequest, + **kwargs, ) -> CollectionUpdateResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_update_message.py b/taskingai/client/apis/api_update_message.py index 2bf7247..beb1229 100644 --- a/taskingai/client/apis/api_update_message.py +++ b/taskingai/client/apis/api_update_message.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -19,7 +18,11 @@ def api_update_message( - assistant_id: str, chat_id: str, message_id: str, payload: MessageUpdateRequest, **kwargs + assistant_id: str, + chat_id: str, + message_id: str, + payload: MessageUpdateRequest, + **kwargs, ) -> MessageUpdateResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -55,7 +58,11 @@ def api_update_message( async def async_api_update_message( - assistant_id: str, chat_id: str, message_id: str, payload: MessageUpdateRequest, **kwargs + assistant_id: str, + chat_id: str, + message_id: str, + payload: MessageUpdateRequest, + **kwargs, ) -> MessageUpdateResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/apis/api_update_record.py b/taskingai/client/apis/api_update_record.py index 1351153..1e40d15 100644 --- a/taskingai/client/apis/api_update_record.py +++ b/taskingai/client/apis/api_update_record.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -19,7 +18,10 @@ def api_update_record( - collection_id: str, record_id: str, payload: RecordUpdateRequest, **kwargs + collection_id: str, + record_id: str, + payload: RecordUpdateRequest, + **kwargs, ) -> RecordUpdateResponse: # get api client sync_api_client = get_api_client(async_client=False) @@ -54,7 +56,10 @@ def api_update_record( async def async_api_update_record( - collection_id: str, record_id: str, payload: RecordUpdateRequest, **kwargs + collection_id: str, + record_id: str, + payload: RecordUpdateRequest, + **kwargs, ) -> RecordUpdateResponse: # get api client async_api_client = get_api_client(async_client=True) diff --git a/taskingai/client/models/__init__.py b/taskingai/client/models/__init__.py index 4653939..ecfbff1 100644 --- a/taskingai/client/models/__init__.py +++ b/taskingai/client/models/__init__.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/__init__.py b/taskingai/client/models/entities/__init__.py index 6f4d966..8d3644f 100644 --- a/taskingai/client/models/entities/__init__.py +++ b/taskingai/client/models/entities/__init__.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/action.py b/taskingai/client/models/entities/action.py index 11a1db9..c69158e 100644 --- a/taskingai/client/models/entities/action.py +++ b/taskingai/client/models/entities/action.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/action_authentication.py b/taskingai/client/models/entities/action_authentication.py index 3913800..f5b0daf 100644 --- a/taskingai/client/models/entities/action_authentication.py +++ b/taskingai/client/models/entities/action_authentication.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/action_authentication_type.py b/taskingai/client/models/entities/action_authentication_type.py index ee3eef1..2af9d3f 100644 --- a/taskingai/client/models/entities/action_authentication_type.py +++ b/taskingai/client/models/entities/action_authentication_type.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/action_body_type.py b/taskingai/client/models/entities/action_body_type.py index 28a4c30..0f46243 100644 --- a/taskingai/client/models/entities/action_body_type.py +++ b/taskingai/client/models/entities/action_body_type.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/action_method.py b/taskingai/client/models/entities/action_method.py index 2c01286..c6038b0 100644 --- a/taskingai/client/models/entities/action_method.py +++ b/taskingai/client/models/entities/action_method.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/action_param.py b/taskingai/client/models/entities/action_param.py index 7704a2a..7d47e92 100644 --- a/taskingai/client/models/entities/action_param.py +++ b/taskingai/client/models/entities/action_param.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/assistant.py b/taskingai/client/models/entities/assistant.py index e74f6c1..3d39e7a 100644 --- a/taskingai/client/models/entities/assistant.py +++ b/taskingai/client/models/entities/assistant.py @@ -8,12 +8,11 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ from pydantic import BaseModel, Field -from typing import Dict, List +from typing import List, Dict from .assistant_memory import AssistantMemory from .tool_ref import ToolRef from .retrieval_ref import RetrievalRef diff --git a/taskingai/client/models/entities/assistant_memory.py b/taskingai/client/models/entities/assistant_memory.py index 6d2456d..e71f71f 100644 --- a/taskingai/client/models/entities/assistant_memory.py +++ b/taskingai/client/models/entities/assistant_memory.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/assistant_memory_type.py b/taskingai/client/models/entities/assistant_memory_type.py index c00527d..8ab833a 100644 --- a/taskingai/client/models/entities/assistant_memory_type.py +++ b/taskingai/client/models/entities/assistant_memory_type.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chat.py b/taskingai/client/models/entities/chat.py index bf53609..b624ae6 100644 --- a/taskingai/client/models/entities/chat.py +++ b/taskingai/client/models/entities/chat.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chat_completion.py b/taskingai/client/models/entities/chat_completion.py index b4cfed7..2229f6e 100644 --- a/taskingai/client/models/entities/chat_completion.py +++ b/taskingai/client/models/entities/chat_completion.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chat_completion_assistant_message.py b/taskingai/client/models/entities/chat_completion_assistant_message.py index 449c96e..6dcfeb8 100644 --- a/taskingai/client/models/entities/chat_completion_assistant_message.py +++ b/taskingai/client/models/entities/chat_completion_assistant_message.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chat_completion_chunk.py b/taskingai/client/models/entities/chat_completion_chunk.py index b43336f..add19ed 100644 --- a/taskingai/client/models/entities/chat_completion_chunk.py +++ b/taskingai/client/models/entities/chat_completion_chunk.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chat_completion_finish_reason.py b/taskingai/client/models/entities/chat_completion_finish_reason.py index 61a58bf..5f9987b 100644 --- a/taskingai/client/models/entities/chat_completion_finish_reason.py +++ b/taskingai/client/models/entities/chat_completion_finish_reason.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chat_completion_function.py b/taskingai/client/models/entities/chat_completion_function.py index f03a496..97893d7 100644 --- a/taskingai/client/models/entities/chat_completion_function.py +++ b/taskingai/client/models/entities/chat_completion_function.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chat_completion_function_call.py b/taskingai/client/models/entities/chat_completion_function_call.py index 7ede894..c498e1e 100644 --- a/taskingai/client/models/entities/chat_completion_function_call.py +++ b/taskingai/client/models/entities/chat_completion_function_call.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chat_completion_function_message.py b/taskingai/client/models/entities/chat_completion_function_message.py index 27a18a6..98166f6 100644 --- a/taskingai/client/models/entities/chat_completion_function_message.py +++ b/taskingai/client/models/entities/chat_completion_function_message.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chat_completion_function_parameters.py b/taskingai/client/models/entities/chat_completion_function_parameters.py index 4d42beb..85a4746 100644 --- a/taskingai/client/models/entities/chat_completion_function_parameters.py +++ b/taskingai/client/models/entities/chat_completion_function_parameters.py @@ -8,12 +8,11 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ from pydantic import BaseModel, Field -from typing import Dict, List +from typing import List, Dict from .chat_completion_function_parameters_property import ChatCompletionFunctionParametersProperty __all__ = ["ChatCompletionFunctionParameters"] diff --git a/taskingai/client/models/entities/chat_completion_function_parameters_property.py b/taskingai/client/models/entities/chat_completion_function_parameters_property.py index d909c59..056009d 100644 --- a/taskingai/client/models/entities/chat_completion_function_parameters_property.py +++ b/taskingai/client/models/entities/chat_completion_function_parameters_property.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chat_completion_message.py b/taskingai/client/models/entities/chat_completion_message.py index ce82cd6..04a90fb 100644 --- a/taskingai/client/models/entities/chat_completion_message.py +++ b/taskingai/client/models/entities/chat_completion_message.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chat_completion_role.py b/taskingai/client/models/entities/chat_completion_role.py index 475a9bc..2873057 100644 --- a/taskingai/client/models/entities/chat_completion_role.py +++ b/taskingai/client/models/entities/chat_completion_role.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chat_completion_system_message.py b/taskingai/client/models/entities/chat_completion_system_message.py index 35076ff..00cada1 100644 --- a/taskingai/client/models/entities/chat_completion_system_message.py +++ b/taskingai/client/models/entities/chat_completion_system_message.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chat_completion_user_message.py b/taskingai/client/models/entities/chat_completion_user_message.py index c20fb0c..f94c195 100644 --- a/taskingai/client/models/entities/chat_completion_user_message.py +++ b/taskingai/client/models/entities/chat_completion_user_message.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chat_memory.py b/taskingai/client/models/entities/chat_memory.py index 4a30379..53c5248 100644 --- a/taskingai/client/models/entities/chat_memory.py +++ b/taskingai/client/models/entities/chat_memory.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chat_memory_message.py b/taskingai/client/models/entities/chat_memory_message.py index e2cbf71..1ad9dfc 100644 --- a/taskingai/client/models/entities/chat_memory_message.py +++ b/taskingai/client/models/entities/chat_memory_message.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/chunk.py b/taskingai/client/models/entities/chunk.py index ab22914..227f042 100644 --- a/taskingai/client/models/entities/chunk.py +++ b/taskingai/client/models/entities/chunk.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/collection.py b/taskingai/client/models/entities/collection.py index eb97c0a..58163be 100644 --- a/taskingai/client/models/entities/collection.py +++ b/taskingai/client/models/entities/collection.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/message.py b/taskingai/client/models/entities/message.py index 9109d76..ef0890a 100644 --- a/taskingai/client/models/entities/message.py +++ b/taskingai/client/models/entities/message.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/message_chunk.py b/taskingai/client/models/entities/message_chunk.py index d1eb6cf..3b21815 100644 --- a/taskingai/client/models/entities/message_chunk.py +++ b/taskingai/client/models/entities/message_chunk.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/message_content.py b/taskingai/client/models/entities/message_content.py index 8ea926c..99b8162 100644 --- a/taskingai/client/models/entities/message_content.py +++ b/taskingai/client/models/entities/message_content.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/message_generation_log.py b/taskingai/client/models/entities/message_generation_log.py index de0bbea..cc14047 100644 --- a/taskingai/client/models/entities/message_generation_log.py +++ b/taskingai/client/models/entities/message_generation_log.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/message_role.py b/taskingai/client/models/entities/message_role.py index 8e8cbe0..edb25da 100644 --- a/taskingai/client/models/entities/message_role.py +++ b/taskingai/client/models/entities/message_role.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/record.py b/taskingai/client/models/entities/record.py index a1afee2..ee584b1 100644 --- a/taskingai/client/models/entities/record.py +++ b/taskingai/client/models/entities/record.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/record_type.py b/taskingai/client/models/entities/record_type.py index 711c13f..e2a890a 100644 --- a/taskingai/client/models/entities/record_type.py +++ b/taskingai/client/models/entities/record_type.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/retrieval_config.py b/taskingai/client/models/entities/retrieval_config.py index 1eef5c1..2c88262 100644 --- a/taskingai/client/models/entities/retrieval_config.py +++ b/taskingai/client/models/entities/retrieval_config.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -23,4 +22,3 @@ class RetrievalConfig(BaseModel): top_k: int = Field(3, ge=1, le=20) max_tokens: Optional[int] = Field(None, ge=1, le=8192) method: RetrievalMethod = Field(...) - function_description: Optional[str] = Field(None) diff --git a/taskingai/client/models/entities/retrieval_method.py b/taskingai/client/models/entities/retrieval_method.py index 72bb888..2ffc12f 100644 --- a/taskingai/client/models/entities/retrieval_method.py +++ b/taskingai/client/models/entities/retrieval_method.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/retrieval_ref.py b/taskingai/client/models/entities/retrieval_ref.py index 24dfde3..3f57e95 100644 --- a/taskingai/client/models/entities/retrieval_ref.py +++ b/taskingai/client/models/entities/retrieval_ref.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/retrieval_type.py b/taskingai/client/models/entities/retrieval_type.py index 2bc2add..43f6220 100644 --- a/taskingai/client/models/entities/retrieval_type.py +++ b/taskingai/client/models/entities/retrieval_type.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/sort_order_enum.py b/taskingai/client/models/entities/sort_order_enum.py index 6f0b0b9..9d77915 100644 --- a/taskingai/client/models/entities/sort_order_enum.py +++ b/taskingai/client/models/entities/sort_order_enum.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/status.py b/taskingai/client/models/entities/status.py index bf1753e..e8cd336 100644 --- a/taskingai/client/models/entities/status.py +++ b/taskingai/client/models/entities/status.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/text_embedding_input_type.py b/taskingai/client/models/entities/text_embedding_input_type.py index 3a5b985..29c6af8 100644 --- a/taskingai/client/models/entities/text_embedding_input_type.py +++ b/taskingai/client/models/entities/text_embedding_input_type.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/text_embedding_output.py b/taskingai/client/models/entities/text_embedding_output.py index 72e3aec..4dea4e6 100644 --- a/taskingai/client/models/entities/text_embedding_output.py +++ b/taskingai/client/models/entities/text_embedding_output.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/text_splitter.py b/taskingai/client/models/entities/text_splitter.py index 9e6a3d7..c1d7ae0 100644 --- a/taskingai/client/models/entities/text_splitter.py +++ b/taskingai/client/models/entities/text_splitter.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/text_splitter_type.py b/taskingai/client/models/entities/text_splitter_type.py index a0037b0..5e7d2f8 100644 --- a/taskingai/client/models/entities/text_splitter_type.py +++ b/taskingai/client/models/entities/text_splitter_type.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/tool_ref.py b/taskingai/client/models/entities/tool_ref.py index e864ee0..bc10bd8 100644 --- a/taskingai/client/models/entities/tool_ref.py +++ b/taskingai/client/models/entities/tool_ref.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/entities/tool_type.py b/taskingai/client/models/entities/tool_type.py index 592d112..fffc450 100644 --- a/taskingai/client/models/entities/tool_type.py +++ b/taskingai/client/models/entities/tool_type.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/__init__.py b/taskingai/client/models/schemas/__init__.py index e561f9d..2385513 100644 --- a/taskingai/client/models/schemas/__init__.py +++ b/taskingai/client/models/schemas/__init__.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ @@ -28,6 +27,7 @@ from .assistant_list_response import * from .assistant_update_request import * from .assistant_update_response import * +from .base_data_response import * from .base_empty_response import * from .chat_completion_request import * from .chat_completion_response import * diff --git a/taskingai/client/models/schemas/action_bulk_create_request.py b/taskingai/client/models/schemas/action_bulk_create_request.py index dc7b778..0c3c971 100644 --- a/taskingai/client/models/schemas/action_bulk_create_request.py +++ b/taskingai/client/models/schemas/action_bulk_create_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/action_bulk_create_response.py b/taskingai/client/models/schemas/action_bulk_create_response.py index c3a577c..36b3681 100644 --- a/taskingai/client/models/schemas/action_bulk_create_response.py +++ b/taskingai/client/models/schemas/action_bulk_create_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/action_get_response.py b/taskingai/client/models/schemas/action_get_response.py index 67c49d6..e9c6159 100644 --- a/taskingai/client/models/schemas/action_get_response.py +++ b/taskingai/client/models/schemas/action_get_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/action_list_request.py b/taskingai/client/models/schemas/action_list_request.py index 1b859fe..7aecf68 100644 --- a/taskingai/client/models/schemas/action_list_request.py +++ b/taskingai/client/models/schemas/action_list_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/action_list_response.py b/taskingai/client/models/schemas/action_list_response.py index f3cf544..1123acd 100644 --- a/taskingai/client/models/schemas/action_list_response.py +++ b/taskingai/client/models/schemas/action_list_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/action_run_request.py b/taskingai/client/models/schemas/action_run_request.py index 8261fda..f1466c9 100644 --- a/taskingai/client/models/schemas/action_run_request.py +++ b/taskingai/client/models/schemas/action_run_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/action_run_response.py b/taskingai/client/models/schemas/action_run_response.py index e89c0e0..605dc47 100644 --- a/taskingai/client/models/schemas/action_run_response.py +++ b/taskingai/client/models/schemas/action_run_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/action_update_request.py b/taskingai/client/models/schemas/action_update_request.py index 2f78f5f..0774bf8 100644 --- a/taskingai/client/models/schemas/action_update_request.py +++ b/taskingai/client/models/schemas/action_update_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/action_update_response.py b/taskingai/client/models/schemas/action_update_response.py index e72cf40..54a56b7 100644 --- a/taskingai/client/models/schemas/action_update_response.py +++ b/taskingai/client/models/schemas/action_update_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/assistant_create_request.py b/taskingai/client/models/schemas/assistant_create_request.py index 775911d..3ececa7 100644 --- a/taskingai/client/models/schemas/assistant_create_request.py +++ b/taskingai/client/models/schemas/assistant_create_request.py @@ -8,12 +8,11 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ from pydantic import BaseModel, Field -from typing import Dict, List +from typing import List, Dict from ..entities.assistant_memory import AssistantMemory from ..entities.tool_ref import ToolRef from ..entities.retrieval_ref import RetrievalRef diff --git a/taskingai/client/models/schemas/assistant_create_response.py b/taskingai/client/models/schemas/assistant_create_response.py index 7d3c008..6bb0a14 100644 --- a/taskingai/client/models/schemas/assistant_create_response.py +++ b/taskingai/client/models/schemas/assistant_create_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/assistant_get_response.py b/taskingai/client/models/schemas/assistant_get_response.py index 32e2067..76c9393 100644 --- a/taskingai/client/models/schemas/assistant_get_response.py +++ b/taskingai/client/models/schemas/assistant_get_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/assistant_list_request.py b/taskingai/client/models/schemas/assistant_list_request.py index ddf159e..123445f 100644 --- a/taskingai/client/models/schemas/assistant_list_request.py +++ b/taskingai/client/models/schemas/assistant_list_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/assistant_list_response.py b/taskingai/client/models/schemas/assistant_list_response.py index 71b40fd..1634f8f 100644 --- a/taskingai/client/models/schemas/assistant_list_response.py +++ b/taskingai/client/models/schemas/assistant_list_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/assistant_update_request.py b/taskingai/client/models/schemas/assistant_update_request.py index f49f3f8..fe54f45 100644 --- a/taskingai/client/models/schemas/assistant_update_request.py +++ b/taskingai/client/models/schemas/assistant_update_request.py @@ -8,12 +8,11 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ from pydantic import BaseModel, Field -from typing import Optional, Dict, List +from typing import Optional, List, Dict from ..entities.assistant_memory import AssistantMemory from ..entities.tool_ref import ToolRef from ..entities.retrieval_ref import RetrievalRef diff --git a/taskingai/client/models/schemas/assistant_update_response.py b/taskingai/client/models/schemas/assistant_update_response.py index 5ce2142..a66d5d9 100644 --- a/taskingai/client/models/schemas/assistant_update_response.py +++ b/taskingai/client/models/schemas/assistant_update_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/base_data_response.py b/taskingai/client/models/schemas/base_data_response.py new file mode 100644 index 0000000..9761238 --- /dev/null +++ b/taskingai/client/models/schemas/base_data_response.py @@ -0,0 +1,23 @@ +# -*- coding: utf-8 -*- + +# base_data_response.py + +""" +This script is automatically generated for TaskingAI python client +Do not modify the file manually + +Author: James Yao +Organization: TaskingAI +License: Apache 2.0 +""" + +from pydantic import BaseModel, Field +from typing import Any + + +__all__ = ["BaseDataResponse"] + + +class BaseDataResponse(BaseModel): + status: str = Field("success") + data: Any = Field(...) diff --git a/taskingai/client/models/schemas/base_empty_response.py b/taskingai/client/models/schemas/base_empty_response.py index 0b87e67..93b0de1 100644 --- a/taskingai/client/models/schemas/base_empty_response.py +++ b/taskingai/client/models/schemas/base_empty_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chat_completion_request.py b/taskingai/client/models/schemas/chat_completion_request.py index 54862de..dfed9e0 100644 --- a/taskingai/client/models/schemas/chat_completion_request.py +++ b/taskingai/client/models/schemas/chat_completion_request.py @@ -8,16 +8,15 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ from pydantic import BaseModel, Field -from typing import Optional, Dict, List, Union -from ..entities.chat_completion_user_message import ChatCompletionUserMessage +from typing import Optional, List, Dict, Union +from ..entities.chat_completion_assistant_message import ChatCompletionAssistantMessage from ..entities.chat_completion_system_message import ChatCompletionSystemMessage +from ..entities.chat_completion_user_message import ChatCompletionUserMessage from ..entities.chat_completion_function_message import ChatCompletionFunctionMessage -from ..entities.chat_completion_assistant_message import ChatCompletionAssistantMessage from ..entities.chat_completion_function import ChatCompletionFunction __all__ = ["ChatCompletionRequest"] @@ -29,9 +28,9 @@ class ChatCompletionRequest(BaseModel): stream: bool = Field(False) messages: List[ Union[ - ChatCompletionUserMessage, - ChatCompletionAssistantMessage, ChatCompletionFunctionMessage, + ChatCompletionAssistantMessage, + ChatCompletionUserMessage, ChatCompletionSystemMessage, ] ] = Field(...) diff --git a/taskingai/client/models/schemas/chat_completion_response.py b/taskingai/client/models/schemas/chat_completion_response.py index 60cc99e..3b0bda5 100644 --- a/taskingai/client/models/schemas/chat_completion_response.py +++ b/taskingai/client/models/schemas/chat_completion_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chat_create_request.py b/taskingai/client/models/schemas/chat_create_request.py index 841dc9c..e62a659 100644 --- a/taskingai/client/models/schemas/chat_create_request.py +++ b/taskingai/client/models/schemas/chat_create_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chat_create_response.py b/taskingai/client/models/schemas/chat_create_response.py index c42b58c..f879c94 100644 --- a/taskingai/client/models/schemas/chat_create_response.py +++ b/taskingai/client/models/schemas/chat_create_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chat_get_response.py b/taskingai/client/models/schemas/chat_get_response.py index d82ef4f..6d54665 100644 --- a/taskingai/client/models/schemas/chat_get_response.py +++ b/taskingai/client/models/schemas/chat_get_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chat_list_request.py b/taskingai/client/models/schemas/chat_list_request.py index 68f2bc0..3bdeb47 100644 --- a/taskingai/client/models/schemas/chat_list_request.py +++ b/taskingai/client/models/schemas/chat_list_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chat_list_response.py b/taskingai/client/models/schemas/chat_list_response.py index 84c8b34..acdebba 100644 --- a/taskingai/client/models/schemas/chat_list_response.py +++ b/taskingai/client/models/schemas/chat_list_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chat_update_request.py b/taskingai/client/models/schemas/chat_update_request.py index 5dec445..45de54a 100644 --- a/taskingai/client/models/schemas/chat_update_request.py +++ b/taskingai/client/models/schemas/chat_update_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chat_update_response.py b/taskingai/client/models/schemas/chat_update_response.py index 781cece..a019e31 100644 --- a/taskingai/client/models/schemas/chat_update_response.py +++ b/taskingai/client/models/schemas/chat_update_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chunk_create_request.py b/taskingai/client/models/schemas/chunk_create_request.py index f9047ca..e599c62 100644 --- a/taskingai/client/models/schemas/chunk_create_request.py +++ b/taskingai/client/models/schemas/chunk_create_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chunk_create_response.py b/taskingai/client/models/schemas/chunk_create_response.py index 4cd8b34..cba6ac8 100644 --- a/taskingai/client/models/schemas/chunk_create_response.py +++ b/taskingai/client/models/schemas/chunk_create_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chunk_get_response.py b/taskingai/client/models/schemas/chunk_get_response.py index 26f5bb3..3d1afeb 100644 --- a/taskingai/client/models/schemas/chunk_get_response.py +++ b/taskingai/client/models/schemas/chunk_get_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chunk_list_request.py b/taskingai/client/models/schemas/chunk_list_request.py index 30782f3..aa2348f 100644 --- a/taskingai/client/models/schemas/chunk_list_request.py +++ b/taskingai/client/models/schemas/chunk_list_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chunk_list_response.py b/taskingai/client/models/schemas/chunk_list_response.py index 0bd00f0..c0fd794 100644 --- a/taskingai/client/models/schemas/chunk_list_response.py +++ b/taskingai/client/models/schemas/chunk_list_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chunk_query_request.py b/taskingai/client/models/schemas/chunk_query_request.py index 36c8417..3214262 100644 --- a/taskingai/client/models/schemas/chunk_query_request.py +++ b/taskingai/client/models/schemas/chunk_query_request.py @@ -8,11 +8,11 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ from pydantic import BaseModel, Field +from typing import Optional __all__ = ["ChunkQueryRequest"] @@ -20,4 +20,5 @@ class ChunkQueryRequest(BaseModel): top_k: int = Field(..., ge=1, le=20) + max_tokens: Optional[int] = Field(None, ge=1) query_text: str = Field(..., min_length=1, max_length=32768) diff --git a/taskingai/client/models/schemas/chunk_query_response.py b/taskingai/client/models/schemas/chunk_query_response.py index f7a3ca9..2eda666 100644 --- a/taskingai/client/models/schemas/chunk_query_response.py +++ b/taskingai/client/models/schemas/chunk_query_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chunk_update_request.py b/taskingai/client/models/schemas/chunk_update_request.py index 8600cd4..44bfa29 100644 --- a/taskingai/client/models/schemas/chunk_update_request.py +++ b/taskingai/client/models/schemas/chunk_update_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/chunk_update_response.py b/taskingai/client/models/schemas/chunk_update_response.py index 934147d..5cc6db8 100644 --- a/taskingai/client/models/schemas/chunk_update_response.py +++ b/taskingai/client/models/schemas/chunk_update_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/collection_create_request.py b/taskingai/client/models/schemas/collection_create_request.py index 1eb5129..285756a 100644 --- a/taskingai/client/models/schemas/collection_create_request.py +++ b/taskingai/client/models/schemas/collection_create_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/collection_create_response.py b/taskingai/client/models/schemas/collection_create_response.py index 0859a28..2701226 100644 --- a/taskingai/client/models/schemas/collection_create_response.py +++ b/taskingai/client/models/schemas/collection_create_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/collection_get_response.py b/taskingai/client/models/schemas/collection_get_response.py index 85f33ca..8ec536c 100644 --- a/taskingai/client/models/schemas/collection_get_response.py +++ b/taskingai/client/models/schemas/collection_get_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/collection_list_request.py b/taskingai/client/models/schemas/collection_list_request.py index 2506894..8fe0db2 100644 --- a/taskingai/client/models/schemas/collection_list_request.py +++ b/taskingai/client/models/schemas/collection_list_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/collection_list_response.py b/taskingai/client/models/schemas/collection_list_response.py index 181ac71..cb88f59 100644 --- a/taskingai/client/models/schemas/collection_list_response.py +++ b/taskingai/client/models/schemas/collection_list_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/collection_update_request.py b/taskingai/client/models/schemas/collection_update_request.py index 374ae26..9dbfe54 100644 --- a/taskingai/client/models/schemas/collection_update_request.py +++ b/taskingai/client/models/schemas/collection_update_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/collection_update_response.py b/taskingai/client/models/schemas/collection_update_response.py index 90a7569..dfdb080 100644 --- a/taskingai/client/models/schemas/collection_update_response.py +++ b/taskingai/client/models/schemas/collection_update_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/message_create_request.py b/taskingai/client/models/schemas/message_create_request.py index 55cb093..736b43d 100644 --- a/taskingai/client/models/schemas/message_create_request.py +++ b/taskingai/client/models/schemas/message_create_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/message_create_response.py b/taskingai/client/models/schemas/message_create_response.py index 705188a..efec178 100644 --- a/taskingai/client/models/schemas/message_create_response.py +++ b/taskingai/client/models/schemas/message_create_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/message_generate_request.py b/taskingai/client/models/schemas/message_generate_request.py index fa3bfca..0497742 100644 --- a/taskingai/client/models/schemas/message_generate_request.py +++ b/taskingai/client/models/schemas/message_generate_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/message_generate_response.py b/taskingai/client/models/schemas/message_generate_response.py index 247c044..ec16712 100644 --- a/taskingai/client/models/schemas/message_generate_response.py +++ b/taskingai/client/models/schemas/message_generate_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/message_get_response.py b/taskingai/client/models/schemas/message_get_response.py index 44a5497..6aebcc9 100644 --- a/taskingai/client/models/schemas/message_get_response.py +++ b/taskingai/client/models/schemas/message_get_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/message_list_request.py b/taskingai/client/models/schemas/message_list_request.py index 8ab0de4..b3d5beb 100644 --- a/taskingai/client/models/schemas/message_list_request.py +++ b/taskingai/client/models/schemas/message_list_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/message_list_response.py b/taskingai/client/models/schemas/message_list_response.py index 8797e78..2b75b46 100644 --- a/taskingai/client/models/schemas/message_list_response.py +++ b/taskingai/client/models/schemas/message_list_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/message_update_request.py b/taskingai/client/models/schemas/message_update_request.py index 6f83588..07905c7 100644 --- a/taskingai/client/models/schemas/message_update_request.py +++ b/taskingai/client/models/schemas/message_update_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/message_update_response.py b/taskingai/client/models/schemas/message_update_response.py index 63cbfb3..db674bf 100644 --- a/taskingai/client/models/schemas/message_update_response.py +++ b/taskingai/client/models/schemas/message_update_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/record_create_request.py b/taskingai/client/models/schemas/record_create_request.py index af0f3ff..4963daa 100644 --- a/taskingai/client/models/schemas/record_create_request.py +++ b/taskingai/client/models/schemas/record_create_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/record_create_response.py b/taskingai/client/models/schemas/record_create_response.py index af80e2c..dcfcd71 100644 --- a/taskingai/client/models/schemas/record_create_response.py +++ b/taskingai/client/models/schemas/record_create_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/record_get_response.py b/taskingai/client/models/schemas/record_get_response.py index 2606644..ad431d4 100644 --- a/taskingai/client/models/schemas/record_get_response.py +++ b/taskingai/client/models/schemas/record_get_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/record_list_request.py b/taskingai/client/models/schemas/record_list_request.py index 9c9ff7b..8f44a33 100644 --- a/taskingai/client/models/schemas/record_list_request.py +++ b/taskingai/client/models/schemas/record_list_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/record_list_response.py b/taskingai/client/models/schemas/record_list_response.py index 5647be2..34de646 100644 --- a/taskingai/client/models/schemas/record_list_response.py +++ b/taskingai/client/models/schemas/record_list_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/record_update_request.py b/taskingai/client/models/schemas/record_update_request.py index 09f19a7..bf28a0f 100644 --- a/taskingai/client/models/schemas/record_update_request.py +++ b/taskingai/client/models/schemas/record_update_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/record_update_response.py b/taskingai/client/models/schemas/record_update_response.py index fd82ee4..af1e8df 100644 --- a/taskingai/client/models/schemas/record_update_response.py +++ b/taskingai/client/models/schemas/record_update_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/text_embedding_request.py b/taskingai/client/models/schemas/text_embedding_request.py index 8242bc2..b5b0892 100644 --- a/taskingai/client/models/schemas/text_embedding_request.py +++ b/taskingai/client/models/schemas/text_embedding_request.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/client/models/schemas/text_embedding_response.py b/taskingai/client/models/schemas/text_embedding_response.py index e1c221b..9c43730 100644 --- a/taskingai/client/models/schemas/text_embedding_response.py +++ b/taskingai/client/models/schemas/text_embedding_response.py @@ -8,7 +8,6 @@ Author: James Yao Organization: TaskingAI -Created: 03-Mar-2024 License: Apache 2.0 """ diff --git a/taskingai/retrieval/chunk.py b/taskingai/retrieval/chunk.py index b277678..580913e 100644 --- a/taskingai/retrieval/chunk.py +++ b/taskingai/retrieval/chunk.py @@ -241,18 +241,21 @@ def query_chunks( collection_id: str, query_text: str, top_k: int = 3, + max_tokens: Optional[int] = None, ) -> List[Chunk]: """ Query chunks in a collection. :param collection_id: The ID of the collection. :param query_text: The query text. :param top_k: The number of most relevant chunks to return. + :param max_tokens: The maximum number of tokens to return. """ # only add non-None parameters body = ChunkQueryRequest( top_k=top_k, query_text=query_text, + max_tokens=max_tokens, ) response: ChunkQueryResponse = api_query_collection_chunks( collection_id=collection_id, @@ -265,18 +268,21 @@ async def a_query_chunks( collection_id: str, query_text: str, top_k: int = 3, + max_tokens: Optional[int] = None, ) -> List[Chunk]: """ Query chunks in a collection. :param collection_id: The ID of the collection. :param query_text: The query text. :param top_k: The number of most relevant chunks to return. + :param max_tokens: The maximum number of tokens to return. """ # only add non-None parameters body = ChunkQueryRequest( top_k=top_k, query_text=query_text, + max_tokens=max_tokens, ) response: ChunkQueryResponse = await async_api_query_collection_chunks( collection_id=collection_id,