feat: update retrieval and assistant

Author: Dttbd

README.md

@@ -138,38 +138,6 @@ taskingai.retrieval.delete_collection(collection_id=coll.collection_id)
 print("Collection deleted.")
 ```
 
-### Tools
-
-The Tools module in TaskingAI is an essential suite designed to augment the capabilities of TaskingAI agents. Here is an example of how to create, run, and delete a tool action:
-
-```python
-import taskingai
-
-# Define a schema for the tool action
-OPENAPI_SCHEMA = {
-    # Schema definition goes here
-}
-
-# Create a tool action based on the defined schema
-actions = taskingai.tool.bulk_create_actions(
-    openapi_schema=OPENAPI_SCHEMA,
-    authentication={"type": "none"},
-)
-action = actions[0]
-print(f"Action created: {action.action_id}")
-
-# Run the action for a test purpose
-result = taskingai.tool.run_action(
-    action_id=action.action_id,
-    parameters={"number": 42}
-)
-print(f"Action result: {result}")
-
-# Delete the action when done
-taskingai.tool.delete_action(action_id=action.action_id)
-print("Action deleted.")
-```
-
 ## Contributing
 
 We welcome contributions of all kinds. Please read our [Contributing Guidelines](./CONTRIBUTING.md) for more information on how to get started.

examples/assistant/chat_with_assistant.ipynb

@@ -1,6 +1,5 @@
 from .config import *
 from . import assistant
-from . import tool
 from . import retrieval
 from . import inference
 from . import file

examples/crud/assistant_crud.ipynb

@@ -15,6 +15,8 @@
     "a_create_chat",
     "a_update_chat",
     "a_delete_chat",
+    "clean_chat_context",
+    "a_clean_chat_context",
 ]
 
 
@@ -238,3 +240,27 @@ async def a_delete_chat(
     """
 
     await async_api_delete_chat(assistant_id=assistant_id, chat_id=chat_id)
+
+
+def clean_chat_context(assistant_id: str, chat_id: str) -> Message:
+    """
+    Clean chat context.
+
+    :param assistant_id: The ID of the assistant.
+    :param chat_id: The ID of the chat.
+    """
+
+    response = api_clean_chat_context(assistant_id=assistant_id, chat_id=chat_id)
+    return response.data
+
+
+async def a_clean_chat_context(assistant_id: str, chat_id: str) -> Message:
+    """
+    Clean chat context in async mode.
+
+    :param assistant_id: The ID of the assistant.
+    :param chat_id: The ID of the chat.
+    """
+
+    response = await async_api_clean_chat_context(assistant_id=assistant_id, chat_id=chat_id)
+    return response.data

examples/crud/retrieval_crud.ipynb

@@ -6,22 +6,10 @@
 __all__ = [
     "AssistantMemory",
     "AssistantMemoryType",
-    "AssistantNaiveMemory",
-    "AssistantZeroMemory",
     "AssistantMessageWindowMemory",
 ]
 
 
-class AssistantNaiveMemory(AssistantMemory):
-    def __init__(self):
-        super().__init__(type=AssistantMemoryType.NAIVE)
-
-
-class AssistantZeroMemory(AssistantMemory):
-    def __init__(self):
-        super().__init__(type=AssistantMemoryType.ZERO)
-
-
 class AssistantMessageWindowMemory(AssistantMemory):
-    def __init__(self, max_messages: int, max_tokens: int):
-        super().__init__(type=AssistantMemoryType.MESSAGE_WINDOW, max_messages=max_messages, max_tokens=max_tokens)
+    def __init__(self, max_tokens: int):
+        super().__init__(type=AssistantMemoryType.MESSAGE_WINDOW, max_tokens=max_tokens)

examples/crud/tool_crud.ipynb

@@ -11,43 +11,33 @@
 License: Apache 2.0
 """
 
-from .api_bulk_create_actions import *
 from .api_chat_completion import *
+from .api_clean_chat_context import *
 from .api_create_assistant import *
 from .api_create_chat import *
-from .api_create_chunk import *
 from .api_create_collection import *
 from .api_create_message import *
 from .api_create_record import *
-from .api_delete_action import *
 from .api_delete_assistant import *
 from .api_delete_chat import *
-from .api_delete_chunk import *
 from .api_delete_collection import *
 from .api_delete_message import *
 from .api_delete_record import *
 from .api_generate_message import *
-from .api_get_action import *
 from .api_get_assistant import *
 from .api_get_chat import *
-from .api_get_chunk import *
 from .api_get_collection import *
 from .api_get_message import *
 from .api_get_record import *
-from .api_list_actions import *
 from .api_list_assistants import *
 from .api_list_chats import *
-from .api_list_chunks import *
 from .api_list_collections import *
 from .api_list_messages import *
 from .api_list_records import *
 from .api_query_collection_chunks import *
-from .api_run_action import *
 from .api_text_embedding import *
-from .api_update_action import *
 from .api_update_assistant import *
 from .api_update_chat import *
-from .api_update_chunk import *
 from .api_update_collection import *
 from .api_update_message import *
 from .api_update_record import *