fix: fix clean chat context api response

Author: Dttbd

taskingai/assistant/assistant.py

@@ -348,23 +348,25 @@ async def a_delete_assistant(assistant_id: str) -> None:
     await async_api_delete_assistant(assistant_id=assistant_id)
 
 
-def clean_chat_context(assistant_id: str, chat_id: str) -> ChatCleanContextResponse:
+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.
     """
 
-    api_clean_chat_context(assistant_id=assistant_id, chat_id=chat_id)
+    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) -> ChatCleanContextResponse:
+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.
     """
 
-    await async_api_clean_chat_context(assistant_id=assistant_id, chat_id=chat_id)
+    response = await async_api_clean_chat_context(assistant_id=assistant_id, chat_id=chat_id)
+    return response.data