sunholo 0.127.1__py3-none-any.whl → 0.127.3__py3-none-any.whl

sunholo/database/alloydb_client.py

@@ -823,7 +823,7 @@ class AlloyDBClient:
         return await self._insert_single_row(table_name, regular_data, metadata)
 
 
-    async def _insert_single_row(self, table_name: str, data: dict, metadata: dict = None):
+    async def _insert_single_row(self, table_name: str, data: dict, metadata: dict = None, primary_key_column:str = "id"):
         """
         Inserts a single row of data into the specified table.
         
@@ -869,7 +869,7 @@ class AlloyDBClient:
         sql = f'''
         INSERT INTO "{table_name}" ({columns_str})
         VALUES ({placeholders_str})
-        RETURNING id
+        RETURNING {primary_key_column}
         '''
         
         # Execute SQL to insert data based on engine type
@@ -883,6 +883,64 @@ class AlloyDBClient:
         log.info(f"Inserted data into table {table_name}")
         
         return result
+
+    async def update_row(self, table_name: str, primary_key_column: str, primary_key_value: str, 
+                        update_data: dict, condition: str = None):
+        """
+        Updates a row in the specified table based on the primary key.
+        
+        Args:
+            table_name (str): Name of the table to update
+            primary_key_column (str): Name of the primary key column (e.g., 'acdid')
+            primary_key_value (str): Value of the primary key for the row to update
+            update_data (dict): Dictionary containing column names and values to update
+            condition (str, optional): Additional condition for the WHERE clause
+            
+        Returns:
+            Result of SQL execution
+        """
+        if not update_data:
+            raise ValueError("No update data provided")
+        
+        # Generate SET clause parts
+        set_parts = []
+        processed_values = {}
+        
+        for i, (key, value) in enumerate(update_data.items()):
+            # Create a unique parameter name
+            param_name = f"param_{i}"
+            # For JSON values, convert to string
+            if isinstance(value, (dict, list)):
+                processed_values[param_name] = json.dumps(value)
+            else:
+                processed_values[param_name] = value
+            
+            set_parts.append(f'"{key}" = :{param_name}')
+        
+        # Create the WHERE clause
+        where_clause = f'"{primary_key_column}" = :pk_value'
+        processed_values['pk_value'] = primary_key_value
+        
+        if condition:
+            where_clause += f" AND ({condition})"
+        
+        # Construct the SQL statement
+        set_clause = ", ".join(set_parts)
+        sql = f'UPDATE "{table_name}" SET {set_clause} WHERE {where_clause} RETURNING {primary_key_column}'
+        
+        log.info(f"Executing update on {table_name} for {primary_key_column}={primary_key_value}")
+        
+        # Execute SQL based on engine type
+        if self.engine_type == "pg8000":
+            # Use the synchronous method for pg8000
+            result = self._execute_sql_pg8000(sql, processed_values)
+        else:
+            # Use the async method for langchain
+            result = await self._execute_sql_async_langchain(sql, processed_values)
+        
+        log.info(f"Updated row in {table_name} with {primary_key_column}={primary_key_value}")
+        
+        return result
     
     async def get_table_columns(self, table_name, schema="public"):
         """
@@ -1075,7 +1133,8 @@ class AlloyDBClient:
             log.debug(f"Conversion error for value '{value}' to {target_type}: {e}")
             return None
 
-    async def insert_rows_safely(self, table_name, rows, metadata=None, continue_on_error=False):
+    async def insert_rows_safely(self, table_name, rows, metadata=None, continue_on_error=False, primary_key_column="id"  # Specify the correct primary key column here
+):
         """
         Insert multiple rows into a table with error handling for individual rows.
         
@@ -1084,6 +1143,7 @@ class AlloyDBClient:
             rows (list): List of dictionaries containing row data
             metadata (dict, optional): Additional metadata to include in each row
             continue_on_error (bool): Whether to continue if some rows fail
+            primary_key_column (str): The primary key in the table, default 'id'
             
         Returns:
             dict: {
@@ -1136,7 +1196,7 @@ class AlloyDBClient:
                             filtered_row[col_name] = value
                 
                 # Insert the row
-                result = await self._insert_single_row(table_name, filtered_row)
+                result = await self._insert_single_row(table_name, filtered_row, primary_key_column=primary_key_column)
                 results['inserted_rows'] += 1
                 
             except Exception as e:
@@ -1148,7 +1208,7 @@ class AlloyDBClient:
                 results['errors'].append(error_info)
                 results['failed_rows'] += 1
                 
-                log.error(f"Error inserting row {i}: {e}")
+                log.error(f"Error inserting row {i}: {e} for data: {row}")
                 
                 if not continue_on_error:
                     results['success'] = False
@@ -1158,7 +1218,7 @@ class AlloyDBClient:
         results['success'] = results['inserted_rows'] > 0
         return results
 
-    async def create_table_with_columns(self, table_name, column_definitions, if_not_exists=True):
+    async def create_table_with_columns(self, table_name, column_definitions, if_not_exists=True, primary_key_column="id"):
         """
         Create a table with explicit column definitions.
         
@@ -1171,6 +1231,8 @@ class AlloyDBClient:
                 - default: Default value expression (optional)
                 - primary_key: Whether this is a primary key (default False)
             if_not_exists (bool): Whether to use IF NOT EXISTS clause
+            primary_key_column (str): default name of primary key if not specified in column_definitions
+            
             
         Returns:
             Result of the execution
@@ -1186,7 +1248,7 @@ class AlloyDBClient:
         
         if not has_primary_key:
             # Add an ID column as primary key
-            column_strs.append("id SERIAL PRIMARY KEY")
+            column_strs.append(f'"{primary_key_column}" SERIAL PRIMARY KEY')
         
         for col in column_definitions:
             col_name = col.get('name')

{sunholo-0.127.1.dist-info → sunholo-0.127.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sunholo
-Version: 0.127.1
+Version: 0.127.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.127.1.dist-info → sunholo-0.127.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=B_vCN9d2wQj77TGoyHAMryCNKljKt0ehtXNTdASqTIk,50297
+sunholo/database/alloydb_client.py,sha256=pZ6n6Is1NOw1lMqDlKm7XyyFxoAirkbPt6in4XPjBjE,53148
 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.127.1.dist-info/licenses/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.127.1.dist-info/METADATA,sha256=kKvoiijhfyGLL7CfJOEYZGYiCuIAZ6m2hGagaNjfCAQ,10084
-sunholo-0.127.1.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-sunholo-0.127.1.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.127.1.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.127.1.dist-info/RECORD,,
+sunholo-0.127.3.dist-info/licenses/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.127.3.dist-info/METADATA,sha256=Cw-b89amd_DtSRdUVOV-kPMUnVKfMzHj1m9oAnvzam4,10084
+sunholo-0.127.3.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+sunholo-0.127.3.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.127.3.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.127.3.dist-info/RECORD,,