sunholo 0.123.2__py3-none-any.whl → 0.123.3__py3-none-any.whl

sunholo/database/alloydb_client.py

@@ -262,21 +262,38 @@ class AlloyDBClient:
     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."""
+    async def _execute_sql_async_pg8000(self, sql_statement, values=None):
+        """Executes a given SQL statement asynchronously with error handling.
+        
+        Args:
+            sql_statement (str): The SQL statement to execute
+            values (list, optional): Values for parameterized query
+            
+        Returns:
+            Result of SQL execution
+        """
         sql_ = sqlalchemy.text(sql_statement)
         result = None
-        async with self.engine.connect() as conn:
-            try:
-                log.info(f"Executing SQL statement asynchronously: {sql_}")
+        
+        # Get connection directly, avoiding async with
+        conn = await self.engine.connect()
+        try:
+            log.info(f"Executing SQL statement asynchronously: {sql_}")
+            if values:
+                result = await conn.execute(sql_, values)
+            else:
                 result = await conn.execute(sql_)
-            except DatabaseError as e:
-                if "already exists" in str(e):
-                    log.warning(f"Error ignored: {str(e)}. Assuming object already exists.")
-                else:
-                    raise
-            finally:
-                await conn.close()
+            
+            # Explicitly commit transaction
+            await conn.commit()
+        except DatabaseError as e:
+            if "already exists" in str(e):
+                log.warning(f"Error ignored: {str(e)}. Assuming object already exists.")
+            else:
+                raise
+        finally:
+            # Close connection only here, not inside the context manager
+            await conn.close()
 
         return result
     
@@ -540,7 +557,7 @@ class AlloyDBClient:
         """
         try:
             # Simple query to check connection
-            result = await self.execute_sql_async("SELECT 1")
+            _ = await self.execute_sql_async("SELECT 1")
             return True
         except Exception as e:
             log.warning(f"Database connection check failed: {e}")
@@ -565,7 +582,7 @@ class AlloyDBClient:
                 # Re-create the engine
                 self.engine = self._create_engine()
                 
-            log.info(f"Successfully reconnected to AlloyDB")
+            log.info("Successfully reconnected to AlloyDB")
             return True
         except Exception as e:
             log.error(f"Failed to reconnect to AlloyDB: {e}")
@@ -640,7 +657,8 @@ class AlloyDBClient:
         # Grant permissions if users are provided
         if users:
             for user in users:
-                await self.execute_sql_async(f'GRANT SELECT, INSERT, UPDATE, DELETE ON TABLE "{table_name}" TO "{user}";')
+                grant_sql = f'GRANT SELECT, INSERT, UPDATE, DELETE ON TABLE "{table_name}" TO "{user}";'
+                await self.execute_sql_async(grant_sql)
         
         return result
 
@@ -689,24 +707,4 @@ class AlloyDBClient:
         result = await self.execute_sql_async(sql, values)
         log.info(f"Inserted data into table {table_name}")
         
-        return result
-
-    async def execute_sql_async(self, sql_statement, values=None):
-        """
-        Executes a given SQL statement asynchronously with optional parameter values.
-        
-        Args:
-            sql_statement (str): The SQL statement to execute
-            values (list, optional): Values for parameterized query
-            
-        Returns:
-            Result of SQL execution
-        """
-        log.info(f"Executing async SQL statement: {sql_statement}")
-        if self.engine_type == "pg8000":
-            result = await self._execute_sql_async_pg8000(sql_statement, values)
-        elif self.engine_type == "langchain":
-            result = await self._execute_sql_async_langchain(sql_statement, values)
-        
-        return result
-
+        return result

{sunholo-0.123.2.dist-info → sunholo-0.123.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sunholo
-Version: 0.123.2
+Version: 0.123.3
 Summary: Large Language Model DevOps - a package to help deploy LLMs to the Cloud.
 Author-email: Holosun ApS <multivac@sunholo.com>
 License: Apache License, Version 2.0

{sunholo-0.123.2.dist-info → sunholo-0.123.3.dist-info}/RECORD

@@ -60,7 +60,7 @@ sunholo/components/retriever.py,sha256=Wmchv3huAM4w7DIS-a5Lp9Hi7M8pE6vZdxgseiT9S
 sunholo/components/vectorstore.py,sha256=k7GS1Y5c6ZGXSDAJvyCes6dTjhDAi0fjGbVLqpyfzBc,5918
 sunholo/database/__init__.py,sha256=bpB5Nk21kwqYj-qdVnvNgXjLsbflnH4g-San7OHMqR4,283
 sunholo/database/alloydb.py,sha256=x1zUMB-EVWbE2Zvp4nAs2Z-tB_kOZmS45H2lwVHdYnk,11678
-sunholo/database/alloydb_client.py,sha256=uPRcdfrE1YXCUoDbaKopjlKXXP5deceEn08bJKUeWI4,27596
+sunholo/database/alloydb_client.py,sha256=yksz52Qnx8xCsW61TPKGqSGTehSukP8suKwShec4z54,27401
 sunholo/database/database.py,sha256=VqhZdkXUNdvWn8sUcUV3YNby1JDVf7IykPVXWBtxo9U,7361
 sunholo/database/lancedb.py,sha256=DyfZntiFKBlVPaFooNN1Z6Pl-LAs4nxWKKuq8GBqN58,715
 sunholo/database/static_dbs.py,sha256=8cvcMwUK6c32AS2e_WguKXWMkFf5iN3g9WHzsh0C07Q,442
@@ -168,9 +168,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.123.2.dist-info/licenses/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.123.2.dist-info/METADATA,sha256=0w8jC9b0-_LEB8VnRcKIIbSlHCIJRFPyiuqwfTnbsm8,10001
-sunholo-0.123.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-sunholo-0.123.2.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.123.2.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.123.2.dist-info/RECORD,,
+sunholo-0.123.3.dist-info/licenses/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.123.3.dist-info/METADATA,sha256=BYdslKNSlXmkUMdV4JExD5JzamsfnwVfzIepQV9PsAA,10001
+sunholo-0.123.3.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+sunholo-0.123.3.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.123.3.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.123.3.dist-info/RECORD,,