client v2.1.4

datastax · Oct 1, 2024 · 8b4ca67 · 8b4ca67
1 parent db36468
commit 8b4ca67
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/client/astra_assistants/astra_assistants_manager.py b/client/astra_assistants/astra_assistants_manager.py
@@ -1,9 +1,8 @@
 import logging
 import os
-from typing import List, Dict
+from typing import List
 
 from litellm import get_llm_provider
-from openai.types.beta.threads.run_submit_tool_outputs_params import ToolOutput
 
 from astra_assistants import patch, OpenAI
 from astra_assistants.astra_assistants_event_handler import AstraEventHandler
@@ -127,7 +126,7 @@ def stream_thread(self, content, tool_choice = None, thread_id: str = None, thre
             print(e)
             raise e
 
-    async def run_thread(self, content, tool = None, thread_id: str = None, thread = None, additional_instructions = None) -> ToolOutput:
+    async def run_thread(self, content, tool = None, thread_id: str = None, thread = None, additional_instructions = None):
         if thread_id is not None:
             thread = self.client.beta.threads.retrieve(thread_id)
         elif thread is None:

diff --git a/client/pyproject.toml b/client/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "astra-assistants"
-version = "2.1.3"
+version = "2.1.4"
 description = "Astra Assistants API - drop in replacement for OpenAI Assistants, powered by AstraDB"
 authors = ["phact <[email protected]>"]
 readme = "README.md"