fix: add max_tokens to query_chunks api schema

Author: jameszyao

taskingai/client/apis/__init__.py

@@ -8,7 +8,6 @@
 
 Author: James Yao
 Organization: TaskingAI
-Created: 03-Mar-2024
 License: Apache 2.0
 """
 

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)
 

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)
 

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)
 

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)
 

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)
 

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)
 

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)

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)
 

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)
 

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)
 

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)