sunholo 0.91.1__py3-none-any.whl → 0.91.3__py3-none-any.whl

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.dist-info → sunholo-0.91.3.dist-info}/METADATA

@@ -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.dist-info → sunholo-0.91.3.dist-info}/RECORD

@@ -59,7 +59,7 @@ sunholo/components/retriever.py,sha256=bKIVT7_18Ut3OJd0E0jyiISPnD9qkHWVjcQPT4i1_
 sunholo/components/vectorstore.py,sha256=xKk7micTRwZckaI7U6PxvFz_ZSjCH48xPTDYiDcv2tc,5913
 sunholo/database/__init__.py,sha256=bpB5Nk21kwqYj-qdVnvNgXjLsbflnH4g-San7OHMqR4,283
 sunholo/database/alloydb.py,sha256=YH8wNPS8gN-TDZEXQcVHxwd1NScHRfAxma3gK4R6KCk,11740
-sunholo/database/alloydb_client.py,sha256=urXmesN-41FrjmYIy1ybp7zUdw7Y4UnVUhWDacBELAU,17002
+sunholo/database/alloydb_client.py,sha256=QYEqMNuqZqDjWsUVk408eC35qmD8Tu87eI6lmYc1YbI,17547
 sunholo/database/database.py,sha256=VqhZdkXUNdvWn8sUcUV3YNby1JDVf7IykPVXWBtxo9U,7361
 sunholo/database/lancedb.py,sha256=DyfZntiFKBlVPaFooNN1Z6Pl-LAs4nxWKKuq8GBqN58,715
 sunholo/database/static_dbs.py,sha256=8cvcMwUK6c32AS2e_WguKXWMkFf5iN3g9WHzsh0C07Q,442
@@ -144,9 +144,9 @@ sunholo/vertex/init.py,sha256=1OQwcPBKZYBTDPdyU7IM4X4OmiXLdsNV30C-fee2scQ,2875
 sunholo/vertex/memory_tools.py,sha256=q_phxgGX2TG2j2MXNULF2xGzQnQPENwjPN9nZ_A9Gh0,7526
 sunholo/vertex/safety.py,sha256=S9PgQT1O_BQAkcqauWncRJaydiP8Q_Jzmu9gxYfy1VA,2482
 sunholo/vertex/type_dict_to_json.py,sha256=uTzL4o9tJRao4u-gJOFcACgWGkBOtqACmb6ihvCErL8,4694
-sunholo-0.91.1.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.91.1.dist-info/METADATA,sha256=VV2CVfBHDRqyFrsgGDzyd-xfFC8gOGn9LTZvuDwjO0Y,7806
-sunholo-0.91.1.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-sunholo-0.91.1.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.91.1.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.91.1.dist-info/RECORD,,
+sunholo-0.91.3.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.91.3.dist-info/METADATA,sha256=vte-1lgpB21AFvYd1RzsZNcKN0FG-vZhdQ0B7PGDXpQ,7806
+sunholo-0.91.3.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+sunholo-0.91.3.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.91.3.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.91.3.dist-info/RECORD,,