2.5.1

Author: phact

client/astra_assistants/patch.py

@@ -24,7 +24,11 @@
 from openai.types.beta.threads import message_create_params, Message
 
 from dotenv import load_dotenv
-from openai.types.beta.vector_stores import VectorStoreFile
+
+try:
+    from openai.types.beta.vector_stores import VectorStoreObject
+except ImportError:
+    from openai.types.vector_stores import VectorStoreObject
 
 from astra_assistants import OpenAIWithDefaultKey, AsyncOpenAIWithDefaultKey
 
@@ -776,4 +780,4 @@ def patch(client: Union[OpenAI, AsyncOpenAI]):
     client.copy = MethodType(enhance_copy_method(client.copy), client.copy)
     client.with_options = client.copy
 
-    return client
+    return client

client/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "astra-assistants"
-version = "2.5.0"
+version = "2.5.1"
 description = "Astra Assistants API - drop in replacement for OpenAI Assistants, powered by AstraDB"
 authors = ["phact <estevezsebastian@gmail.com>"]
 readme = "README.md"