fix: remove offset from list api

Author: jameszyao

taskingai/assistant/assistant.py

@@ -28,7 +28,6 @@
 def list_assistants(
     order: str = "desc",
     limit: int = 20,
-    offset: Optional[int] = None,
     after: Optional[str] = None,
     before: Optional[str] = None,
 ) -> List[Assistant]:
@@ -38,22 +37,19 @@ def list_assistants(
 
     :param order: The order of the assistants. It can be "asc" or "desc".
     :param limit: The maximum number of assistants to return.
-    :param offset: The offset of the assistants.
     :param after: The cursor to get the next page of assistants.
     :param before: The cursor to get the previous page of assistants.
     :return: The list of assistants.
     """
-    # verify only one of offset, after and before is not None
-    offset_params = [offset, after, before]
-    if sum([1 if param is not None else 0 for param in offset_params]) > 1:
-        raise ValueError("Only one of offset, after and before can be specified.")
+
+    if after and before:
+        raise ValueError("Only one of after and before can be specified.")
 
     api_instance = get_api_instance(ModuleType.assistant)
     # only add non-None parameters
     params = {
         "order": order,
         "limit": limit,
-        "offset": offset,
         "after": after,
         "before": before,
     }
@@ -62,36 +58,31 @@ def list_assistants(
     assistants: List[Assistant] = [Assistant(**item) for item in response.data]
     return assistants
 
+
 async def a_list_assistants(
-    order: str = "desc",
-    limit: int = 20,
-    offset: Optional[int] = None,
-    after: Optional[str] = None,
-    before: Optional[str] = None,
+        order: str = "desc",
+        limit: int = 20,
+        after: Optional[str] = None,
+        before: Optional[str] = None,
 ) -> List[Assistant]:
-
     """
-    List assistants in async mode.
+    List assistants.
 
     :param order: The order of the assistants. It can be "asc" or "desc".
     :param limit: The maximum number of assistants to return.
-    :param offset: The offset of the assistants.
     :param after: The cursor to get the next page of assistants.
     :param before: The cursor to get the previous page of assistants.
     :return: The list of assistants.
     """
 
-    # verify only one of offset, after and before is not None
-    offset_params = [offset, after, before]
-    if sum([1 if param is not None else 0 for param in offset_params]) > 1:
-        raise ValueError("Only one of offset, after and before can be specified.")
+    if after and before:
+        raise ValueError("Only one of after and before can be specified.")
 
     api_instance = get_api_instance(ModuleType.assistant, async_client=True)
     # only add non-None parameters
     params = {
         "order": order,
         "limit": limit,
-        "offset": offset,
         "after": after,
         "before": before,
     }

taskingai/assistant/chat.py

@@ -37,6 +37,8 @@ def list_chats(
     :param before: The cursor to get the previous page of chats.
     :return: The list of chats.
     """
+    if after and before:
+        raise ValueError("Only one of after and before can be specified.")
 
     api_instance = get_api_instance(ModuleType.assistant)
     # only add non-None parameters
@@ -71,6 +73,8 @@ async def a_list_chats(
     :param before: The cursor to get the previous page of chats.
     :return: The list of chats.
     """
+    if after and before:
+        raise ValueError("Only one of after and before can be specified.")
 
     api_instance = get_api_instance(ModuleType.assistant, async_client=True)
     # only add non-None parameters

taskingai/assistant/message.py

@@ -55,8 +55,7 @@ def list_messages(
     :param before: The cursor to get the previous page of messages.
     :return: The list of messages.
     """
-    offset_params = [after, before]
-    if sum([1 if param is not None else 0 for param in offset_params]) > 1:
+    if after and before:
         raise ValueError("Only one of after and before can be specified.")
 
     api_instance = get_api_instance(ModuleType.assistant)
@@ -96,8 +95,7 @@ async def a_list_messages(
     :param before: The cursor to get the previous page of messages.
     :return: The list of messages.
     """
-    offset_params = [after, before]
-    if sum([1 if param is not None else 0 for param in offset_params]) > 1:
+    if after and before:
         raise ValueError("Only one of after and before can be specified.")
 
     api_instance = get_api_instance(ModuleType.assistant, async_client=True)

taskingai/client/api/assistant_api.py

@@ -968,7 +968,6 @@ def list_assistants(self, **kwargs):  # noqa: E501
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: AssistantListResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -994,13 +993,12 @@ def list_assistants_with_http_info(self, **kwargs):  # noqa: E501
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: AssistantListResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['limit', 'order', 'after', 'before', 'offset']  # noqa: E501
+        all_params = ['limit', 'order', 'after', 'before']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1029,8 +1027,6 @@ def list_assistants_with_http_info(self, **kwargs):  # noqa: E501
             query_params.append(('after', params['after']))  # noqa: E501
         if 'before' in params:
             query_params.append(('before', params['before']))  # noqa: E501
-        if 'offset' in params:
-            query_params.append(('offset', params['offset']))  # noqa: E501
 
         header_params = {}
 

taskingai/client/api/assistant_async_api.py

@@ -970,7 +970,6 @@ async def list_assistants(self, **kwargs):  # noqa: E501
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: AssistantListResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -996,13 +995,12 @@ async def list_assistants_with_http_info(self, **kwargs):  # noqa: E501
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: AssistantListResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['limit', 'order', 'after', 'before', 'offset']  # noqa: E501
+        all_params = ['limit', 'order', 'after', 'before']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1031,8 +1029,6 @@ async def list_assistants_with_http_info(self, **kwargs):  # noqa: E501
             query_params.append(('after', params['after']))  # noqa: E501
         if 'before' in params:
             query_params.append(('before', params['before']))  # noqa: E501
-        if 'offset' in params:
-            query_params.append(('offset', params['offset']))  # noqa: E501
 
         header_params = {}
 

taskingai/client/api/retrieval_api.py

@@ -633,7 +633,6 @@ def list_collections(self, **kwargs):  # noqa: E501
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: CollectionListResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -659,13 +658,12 @@ def list_collections_with_http_info(self, **kwargs):  # noqa: E501
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: CollectionListResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['limit', 'order', 'after', 'before', 'offset']  # noqa: E501
+        all_params = ['limit', 'order', 'after', 'before']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -694,8 +692,6 @@ def list_collections_with_http_info(self, **kwargs):  # noqa: E501
             query_params.append(('after', params['after']))  # noqa: E501
         if 'before' in params:
             query_params.append(('before', params['before']))  # noqa: E501
-        if 'offset' in params:
-            query_params.append(('offset', params['offset']))  # noqa: E501
 
         header_params = {}
 
@@ -740,7 +736,6 @@ def list_records(self, collection_id, **kwargs):  # noqa: E501
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: RecordListResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -767,13 +762,12 @@ def list_records_with_http_info(self, collection_id, **kwargs):  # noqa: E501
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: RecordListResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['collection_id', 'limit', 'order', 'after', 'before', 'offset']  # noqa: E501
+        all_params = ['collection_id', 'limit', 'order', 'after', 'before']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -808,8 +802,6 @@ def list_records_with_http_info(self, collection_id, **kwargs):  # noqa: E501
             query_params.append(('after', params['after']))  # noqa: E501
         if 'before' in params:
             query_params.append(('before', params['before']))  # noqa: E501
-        if 'offset' in params:
-            query_params.append(('offset', params['offset']))  # noqa: E501
 
         header_params = {}
 

taskingai/client/api/retrieval_async_api.py

@@ -633,7 +633,6 @@ async def list_collections(self, **kwargs):  # noqa: E501
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: CollectionListResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -659,13 +658,12 @@ async def list_collections_with_http_info(self, **kwargs):  # noqa: E501
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: CollectionListResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['limit', 'order', 'after', 'before', 'offset']  # noqa: E501
+        all_params = ['limit', 'order', 'after', 'before']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -694,8 +692,6 @@ async def list_collections_with_http_info(self, **kwargs):  # noqa: E501
             query_params.append(('after', params['after']))  # noqa: E501
         if 'before' in params:
             query_params.append(('before', params['before']))  # noqa: E501
-        if 'offset' in params:
-            query_params.append(('offset', params['offset']))  # noqa: E501
 
         header_params = {}
 
@@ -740,7 +736,6 @@ async def list_records(self, collection_id, **kwargs):  # noqa: E501
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: RecordListResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -767,13 +762,12 @@ async def list_records_with_http_info(self, collection_id, **kwargs):  # noqa: E
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: RecordListResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['collection_id', 'limit', 'order', 'after', 'before', 'offset']  # noqa: E501
+        all_params = ['collection_id', 'limit', 'order', 'after', 'before']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -808,8 +802,6 @@ async def list_records_with_http_info(self, collection_id, **kwargs):  # noqa: E
             query_params.append(('after', params['after']))  # noqa: E501
         if 'before' in params:
             query_params.append(('before', params['before']))  # noqa: E501
-        if 'offset' in params:
-            query_params.append(('offset', params['offset']))  # noqa: E501
 
         header_params = {}
 

taskingai/client/api/tool_api.py

@@ -609,7 +609,6 @@ def list_actions(self, **kwargs):  # noqa: E501
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: ActionListResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -635,13 +634,12 @@ def list_actions_with_http_info(self, **kwargs):  # noqa: E501
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: ActionListResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['limit', 'order', 'after', 'before', 'offset']  # noqa: E501
+        all_params = ['limit', 'order', 'after', 'before']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -670,8 +668,6 @@ def list_actions_with_http_info(self, **kwargs):  # noqa: E501
             query_params.append(('after', params['after']))  # noqa: E501
         if 'before' in params:
             query_params.append(('before', params['before']))  # noqa: E501
-        if 'offset' in params:
-            query_params.append(('offset', params['offset']))  # noqa: E501
 
         header_params = {}
 
@@ -715,7 +711,6 @@ def list_functions(self, **kwargs):  # noqa: E501
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: FunctionListResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -741,13 +736,12 @@ def list_functions_with_http_info(self, **kwargs):  # noqa: E501
         :param object order:
         :param object after:
         :param object before:
-        :param object offset:
         :return: FunctionListResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['limit', 'order', 'after', 'before', 'offset']  # noqa: E501
+        all_params = ['limit', 'order', 'after', 'before']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -776,8 +770,6 @@ def list_functions_with_http_info(self, **kwargs):  # noqa: E501
             query_params.append(('after', params['after']))  # noqa: E501
         if 'before' in params:
             query_params.append(('before', params['before']))  # noqa: E501
-        if 'offset' in params:
-            query_params.append(('offset', params['offset']))  # noqa: E501
 
         header_params = {}