feat: embedding dimension from async

vanna-ai · Jul 8, 2024 · 66b9861 · 66b9861
1 parent 6b6a4e5
commit 66b9861
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/vanna/qdrant/qdrant.py b/src/vanna/qdrant/qdrant.py
@@ -380,12 +380,8 @@ async def aremove_collection(self, collection_name: str) -> bool:
 
     @cached_property
     def embeddings_dimension(self):
-        if self._client:
-            return len(self.generate_embedding("ABCDEF"))
-        elif self._async_client:
-            import asyncio
-            return len(asyncio.run(self.agenerate_embedding("ABCDEF")))
-        return 0
+        import asyncio
+        return len(asyncio.run(self.agenerate_embedding("ABCDEF")))
 
     def get_similar_question_sql(self, question: str, **kwargs) -> list:
         results = self._client.search(