sunholo 0.91.1__tar.gz → 0.91.3__tar.gz

{sunholo-0.91.1 → sunholo-0.91.3}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.91.1
+Version: 0.91.3
 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.1.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.91.3.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0

{sunholo-0.91.1 → sunholo-0.91.3}/setup.py

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

{sunholo-0.91.1 → sunholo-0.91.3}/sunholo/database/alloydb_client.py

@@ -89,11 +89,8 @@ class AlloyDBClient:
             log.info("Build with Langchain engine - will use default service account for auth")
             self.engine = self._create_engine()
             self.engine_type = "langchain"
-        
-        if self.engine_type == "langchain":
-            self.vectorstore = self.get_vectorstore()
-        else:        
-            self.vectorstore = None
+               
+        self.vectorstore = None
 
     def _build_instance_uri(self, project_id, region, cluster_name, instance_name):
         return f"projects/{project_id}/locations/{region}/clusters/{cluster_name}/instances/{instance_name}"
@@ -144,10 +141,13 @@ class AlloyDBClient:
     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?")
         
@@ -165,7 +165,9 @@ class AlloyDBClient:
         
         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 +178,21 @@ class AlloyDBClient:
 
         return query, source_filter_cmd
 
-    def similarity_search(self, query:str, 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:str, 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.1 → sunholo-0.91.3}/sunholo.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.91.1
+Version: 0.91.3
 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.1.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.91.3.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0