sunholo 0.99.1__py3-none-any.whl → 0.99.2__py3-none-any.whl

sunholo/database/alloydb_client.py

@@ -87,7 +87,7 @@ class AlloyDBClient:
         self.password = password
         self.inst_url = ""
         if user:
-            log.info("User specified {user} - using pg8000 engine")
+            log.info(f"User specified {user} - using pg8000 engine")
             self.inst_url = self._build_instance_uri(project_id, region, cluster_name, instance_name)
             self.engine = self._create_engine_from_pg8000(user, password, self.database)
             self.engine_type = "pg8000"
@@ -204,23 +204,18 @@ class AlloyDBClient:
 
         return await self.vectorstore.asimilarity_search(query, filter=source_filter_cmd, k=k)
     
-    def create_index(self, vectorstore=None, vectorsize:int=None):
+    def create_index(self, vectorstore=None):
         from langchain_google_alloydb_pg.indexes import IVFFlatIndex
 
-        vector_size = vectorsize or get_vector_size(self.vector_name)
-        table_name = f"{self.vector_name}_vectorstore_{vector_size}"
-
-        vsize = f'''ALTER TABLE
-        "public"."{table_name}"
-        ALTER COLUMN "embedding" TYPE vector({vector_size});
-        '''
-        self.execute_sql(vsize)
-
         index = IVFFlatIndex()
         vs = vectorstore or self.vectorstore
 
         return vs.apply_vector_index(index)
+    
+    def refresh_index(self, vectorstore=None):
+        vs = vectorstore or self.vectorstore
 
+        return vs.reindex()
 
     def execute_sql(self, sql_statement):
         log.info(f"Executing sync SQL statement: {sql_statement}")
@@ -482,7 +477,7 @@ class AlloyDBClient:
         CREATE TABLE IF NOT EXISTS "{vectorstore_id}" (
         langchain_id UUID NOT NULL,
         content TEXT NOT NULL,
-        embedding vector NOT NULL,
+        embedding vector({vector_size}) NOT NULL,
         source TEXT,
         langchain_metadata JSONB,
         docstore_doc_id UUID,

{sunholo-0.99.1.dist-info → sunholo-0.99.2.dist-info}/METADATA

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

{sunholo-0.99.1.dist-info → sunholo-0.99.2.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=2lzD7oV-jq1xI5157xZ5uByTKi0-Kk7us0fH3WSY7Ck,19176
+sunholo/database/alloydb_client.py,sha256=UQpmILNeInuqMrMgf-AeIB7_I44AgJuR_WSrBafDC_k,18986
 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=tBZxqVZ4InTmdBvLlOYwoSEWu4-kGquc-gxDwZCC4FA,7667
 sunholo/vertex/safety.py,sha256=S9PgQT1O_BQAkcqauWncRJaydiP8Q_Jzmu9gxYfy1VA,2482
 sunholo/vertex/type_dict_to_json.py,sha256=uTzL4o9tJRao4u-gJOFcACgWGkBOtqACmb6ihvCErL8,4694
-sunholo-0.99.1.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.99.1.dist-info/METADATA,sha256=APy3Yi0OiehX4Nix3bfmhrQE7HC9O9JeBaoPC7yxZkM,8310
-sunholo-0.99.1.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-sunholo-0.99.1.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.99.1.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.99.1.dist-info/RECORD,,
+sunholo-0.99.2.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.99.2.dist-info/METADATA,sha256=fws38XF7U74myvRCU-tsXdM0wf3Lc6SCcM2tcVA7yPc,8310
+sunholo-0.99.2.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+sunholo-0.99.2.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.99.2.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.99.2.dist-info/RECORD,,