Skip to content

fix: add max_tokens to query_chunks api schema #25

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Mar 11, 2024
Merged

fix: add max_tokens to query_chunks api schema #25

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion taskingai/client/apis/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@

Author: James Yao
Organization: TaskingAI
Created: 03-Mar-2024
License: Apache 2.0
"""

Expand Down
11 changes: 8 additions & 3 deletions taskingai/client/apis/api_bulk_create_actions.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@

Author: James Yao
Organization: TaskingAI
Created: 03-Mar-2024
License: Apache 2.0
"""

Expand All @@ -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)

Expand Down Expand Up @@ -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)

Expand Down
11 changes: 8 additions & 3 deletions taskingai/client/apis/api_chat_completion.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@

Author: James Yao
Organization: TaskingAI
Created: 03-Mar-2024
License: Apache 2.0
"""

Expand All @@ -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)

Expand Down Expand Up @@ -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)

Expand Down
11 changes: 8 additions & 3 deletions taskingai/client/apis/api_create_assistant.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@

Author: James Yao
Organization: TaskingAI
Created: 03-Mar-2024
License: Apache 2.0
"""

Expand All @@ -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)

Expand Down Expand Up @@ -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)

Expand Down
13 changes: 10 additions & 3 deletions taskingai/client/apis/api_create_chat.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@

Author: James Yao
Organization: TaskingAI
Created: 03-Mar-2024
License: Apache 2.0
"""

Expand All @@ -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)

Expand Down Expand Up @@ -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)

Expand Down
13 changes: 10 additions & 3 deletions taskingai/client/apis/api_create_chunk.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@

Author: James Yao
Organization: TaskingAI
Created: 03-Mar-2024
License: Apache 2.0
"""

Expand All @@ -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)

Expand Down Expand Up @@ -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)

Expand Down
11 changes: 8 additions & 3 deletions taskingai/client/apis/api_create_collection.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@

Author: James Yao
Organization: TaskingAI
Created: 03-Mar-2024
License: Apache 2.0
"""

Expand All @@ -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)

Expand Down Expand Up @@ -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)

Expand Down
11 changes: 8 additions & 3 deletions taskingai/client/apis/api_create_message.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@

Author: James Yao
Organization: TaskingAI
Created: 03-Mar-2024
License: Apache 2.0
"""

Expand All @@ -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)
Expand Down Expand Up @@ -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)
Expand Down
13 changes: 10 additions & 3 deletions taskingai/client/apis/api_create_record.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@

Author: James Yao
Organization: TaskingAI
Created: 03-Mar-2024
License: Apache 2.0
"""

Expand All @@ -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)

Expand Down Expand Up @@ -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)

Expand Down
11 changes: 8 additions & 3 deletions taskingai/client/apis/api_delete_action.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@

Author: James Yao
Organization: TaskingAI
Created: 03-Mar-2024
License: Apache 2.0
"""

Expand All @@ -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)

Expand Down Expand Up @@ -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)

Expand Down
11 changes: 8 additions & 3 deletions taskingai/client/apis/api_delete_assistant.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@

Author: James Yao
Organization: TaskingAI
Created: 03-Mar-2024
License: Apache 2.0
"""

Expand All @@ -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)

Expand Down Expand Up @@ -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)

Expand Down
13 changes: 10 additions & 3 deletions taskingai/client/apis/api_delete_chat.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@

Author: James Yao
Organization: TaskingAI
Created: 03-Mar-2024
License: Apache 2.0
"""

Expand All @@ -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)

Expand Down Expand Up @@ -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)

Expand Down
Loading