sunholo 0.83.3__tar.gz → 0.83.4__tar.gz

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

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

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

@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 
 # Define your base version
-version = '0.83.3'
+version = '0.83.4'
 
 setup(
     name='sunholo',

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

@@ -79,9 +79,11 @@ class AlloyDBClient:
             log.info("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()
+            self.engine_type = "pg8000"
         else:
             log.info("Build with Langchain engine - will use default service account for auth")
             self.engine = self._create_engine()
+            self.engine_type = "langchain"
 
     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}"
@@ -128,8 +130,17 @@ class AlloyDBClient:
         log.info(f"Created AlloyDB engine for {engine}")
 
         return engine
-
+    
     def execute_sql(self, sql_statement):
+        if self.engine_type == "pg8000":
+            return self._execute_sql_pg8000(sql_statement)
+        elif self.engine_type == "langchain":
+            return self._execute_sql_langchain(sql_statement)
+    
+    def _execute_sql_langchain(self, sql_statement):
+        return self.engine._fetch(query = sql_statement)
+
+    def _execute_sql_pg8000(self, sql_statement):
         """Executes a given SQL statement with error handling.
 
          - sql_statement (str): The SQL statement to execute.
@@ -152,6 +163,17 @@ class AlloyDBClient:
         return result
     
     async def execute_sql_async(self, sql_statement):
+        if self.engine_type == "pg8000":
+            result = await self._execute_sql_async_pg8000(sql_statement)
+        elif self.engine_type == "langchain":
+            result = await self._execute_sql_async_langchain(sql_statement)
+        
+        return result
+
+    async def _execute_sql_async_langchain(self, sql_statement):
+        return await self.engine._afetch(query = sql_statement)
+        
+    async def _execute_sql_async_pg8000(self, sql_statement):
         """Executes a given SQL statement asynchronously with error handling."""
         sql_ = sqlalchemy.text(sql_statement)
         result = None

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

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