sunholo 0.91.0__tar.gz → 0.91.2__tar.gz

{sunholo-0.91.0 → sunholo-0.91.2}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.91.0
+Version: 0.91.2
 Summary: Large Language Model DevOps - a package to help deploy LLMs to the Cloud.
 Home-page: https://github.com/sunholo-data/sunholo-py
-Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.91.0.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.91.2.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0

{sunholo-0.91.0 → sunholo-0.91.2}/setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-version = '0.91.0'
+version = '0.91.2'
 
 setup(
     name='sunholo',

{sunholo-0.91.0 → sunholo-0.91.2}/sunholo/database/alloydb_client.py

@@ -63,7 +63,7 @@ class AlloyDBClient:
             if not alloydb_config:
                 raise ValueError("Must specify vac.alloydb_config")
             self.config = alloydb_config
-            self.vector_name = self.config.vector_name
+            self.vector_name = self.config.get("vac")
             project_id = alloydb_config["project_id"]
             region = alloydb_config["region"]
             cluster_name = alloydb_config["cluster"]
@@ -141,13 +141,16 @@ class AlloyDBClient:
 
         return engine
     
-    def _get_embedder(self, vector_name):
-        return get_embeddings(vector_name)
+    def _get_embedder(self):
+        return get_embeddings(self.vector_name)
     
-    def get_vectorstore(self):
+    def get_vectorstore(self, vector_name:str=None):
         if self.engine_type != "langchain":
             raise ValueError("Not available using pg8000 engine")
         
+        if vector_name:
+            self.vector_name = vector_name
+
         if self.vector_name is None:
             raise ValueError("No vectorname found - init with ConfigManager?")
         
@@ -158,14 +161,16 @@ class AlloyDBClient:
         self.vectorstore = AlloyDBVectorStore.create_sync(
                 engine=self.engine,
                 table_name=table_name,
-                embedding_service=self._get_embedder(self.vector_name),
+                embedding_service=self._get_embedder(),
                 metadata_columns=["source", "docstore_doc_id"]
                 #metadata_columns=["source", "eventTime"]
             )
         
         return self.vectorstore
     
-    def _similarity_search(self, query, source_filter:str="", free_filter:str=None):
+    def _similarity_search(self, query, source_filter:str="", free_filter:str=None, vector_name:str=None):
+
+        self.get_vectorstore(vector_name) 
 
         if free_filter is None:
             source_filter_cmd = f"source %LIKE% {source_filter}" if source_filter else None
@@ -176,14 +181,21 @@ class AlloyDBClient:
 
         return query, source_filter_cmd
 
-    def similarity_search(self, query, source_filter:str="", free_filter:str=None, k:int=5):
+    def similarity_search(self, query:str, source_filter:str="", free_filter:str=None, k:int=5, vector_name:str=None):
 
-        query, source_filter_cmd = self._similarity_search(query, source_filter, free_filter)     
+        query, source_filter_cmd = self._similarity_search(query, 
+                                                           source_filter=source_filter, 
+                                                           free_filter=free_filter, 
+                                                           vector_name=vector_name)    
 
         return self.vectorstore.similarity_search(query, filter=source_filter_cmd, k=k)
 
-    async def asimilarity_search(self, query, source_filter:str="", free_filter:str=None, k:int=5):
-        query, source_filter_cmd = self._similarity_search(query, source_filter, free_filter)     
+    async def asimilarity_search(self, query:str, source_filter:str="", free_filter:str=None, k:int=5, vector_name:str=None):
+
+        query, source_filter_cmd = self._similarity_search(query, 
+                                                           source_filter=source_filter, 
+                                                           free_filter=free_filter, 
+                                                           vector_name=vector_name)     
 
         return await self.vectorstore.asimilarity_search(query, filter=source_filter_cmd, k=k)
 

{sunholo-0.91.0 → sunholo-0.91.2}/sunholo.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.91.0
+Version: 0.91.2
 Summary: Large Language Model DevOps - a package to help deploy LLMs to the Cloud.
 Home-page: https://github.com/sunholo-data/sunholo-py
-Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.91.0.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.91.2.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0