sunholo 0.83.4__tar.gz → 0.83.5__tar.gz

{sunholo-0.83.4 → sunholo-0.83.5}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.83.4
+Version: 0.83.5
 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.83.4.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.83.5.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0
@@ -21,7 +21,7 @@ License-File: LICENSE.txt
 Requires-Dist: google-auth
 Requires-Dist: ruamel.yaml
 Requires-Dist: langchain>=0.2.12
-Requires-Dist: langchain_experimental>=0.0.61
+Requires-Dist: langchain_experimental==0.0.63
 Requires-Dist: langchain-community>=0.2.11
 Provides-Extra: all
 Requires-Dist: asyncpg; extra == "all"

{sunholo-0.83.4 → sunholo-0.83.5}/setup.py

@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 
 # Define your base version
-version = '0.83.4'
+version = '0.83.5'
 
 setup(
     name='sunholo',
@@ -32,7 +32,7 @@ setup(
         #"tenacity==8.3.0", #TODO: remove soon
         "ruamel.yaml",
         "langchain>=0.2.12",
-        "langchain_experimental>=0.0.61",
+        "langchain_experimental==0.0.63",
         "langchain-community>=0.2.11",
     ],
     extras_require={

{sunholo-0.83.4 → sunholo-0.83.5}/sunholo/database/alloydb_client.py

@@ -9,6 +9,7 @@ except ImportError:
     pass
 
 from .database import get_vector_size
+from .uuid import generate_uuid_from_object_id
 from ..custom_logging import log
 from ..utils import ConfigManager
 
@@ -122,10 +123,10 @@ class AlloyDBClient:
             region=self.config["region"],
             cluster=self.config["cluster"],
             instance=self.config["instance"],
-            password=self.password,
             database=self.database,
             ip_type=self.config.get("ip_type") or IPTypes.PRIVATE
         )
+        self._loop = engine._loop
 
         log.info(f"Created AlloyDB engine for {engine}")
 
@@ -190,6 +191,34 @@ class AlloyDBClient:
                 await conn.close()
 
         return result
+    
+    def get_document_from_docstore(self, source:str, vector_name):
+        query = self._get_document_from_docstore(source, vector_name)
+
+        return self.execute_sql(query)
+
+    async def get_document_from_docstore_async(self, source:str, vector_name:str):
+        query = self._get_document_from_docstore(source, vector_name)
+
+        document = await self.execute_sql_async(query)
+
+        return document
+    
+    def _get_document_from_docstore(self, source:str, vector_name:str):
+        if not isinstance(source, str):
+            raise ValueError("The 'source' parameter must be a single string, not a list of strings or other iterable.")
+
+        table_name = f"{vector_name}_docstore"
+        doc_id = generate_uuid_from_object_id(source)
+
+        query = f"""
+            SELECT * 
+            FROM {table_name}
+            WHERE doc_id = {doc_id}
+            LIMIT 1;
+        """
+
+        return query
 
     async def get_sources_from_docstore_async(self, sources, vector_name, search_type="OR", just_source_name=False):
         """Fetches sources from the docstore asynchronously."""
@@ -233,7 +262,7 @@ class AlloyDBClient:
             FROM {table_name}
             WHERE {conditions}
             ORDER BY langchain_metadata->>'objectId' ASC
-            LIMIT 500;
+            LIMIT 50;
         """
 
         return query

{sunholo-0.83.4 → sunholo-0.83.5}/sunholo/utils/config_class.py

@@ -89,7 +89,7 @@ class ConfigManager:
                 continue
             if filename.endswith(('.yaml', '.yml', '.json')):
                 config_file = os.path.join(folder, filename)
-                log.debug(f"Checking config file: {config_file}")
+               
                 if filename in self.config_cache:
                     cached_config, cache_time = self.config_cache[filename]
                     time_to_recache = (current_time - cache_time)

{sunholo-0.83.4 → sunholo-0.83.5}/sunholo.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.83.4
+Version: 0.83.5
 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.83.4.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.83.5.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0
@@ -21,7 +21,7 @@ License-File: LICENSE.txt
 Requires-Dist: google-auth
 Requires-Dist: ruamel.yaml
 Requires-Dist: langchain>=0.2.12
-Requires-Dist: langchain_experimental>=0.0.61
+Requires-Dist: langchain_experimental==0.0.63
 Requires-Dist: langchain-community>=0.2.11
 Provides-Extra: all
 Requires-Dist: asyncpg; extra == "all"

{sunholo-0.83.4 → sunholo-0.83.5}/sunholo.egg-info/requires.txt

@@ -1,7 +1,7 @@
 google-auth
 ruamel.yaml
 langchain>=0.2.12
-langchain_experimental>=0.0.61
+langchain_experimental==0.0.63
 langchain-community>=0.2.11
 
 [all]