sunholo 0.127.1__py3-none-any.whl → 0.127.2__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
@@ -1075,7 +1075,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 +1085,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 +1138,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:
@@ -1158,7 +1160,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 +1173,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 +1190,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.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sunholo
-Version: 0.127.1
+Version: 0.127.2
 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.2.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=bVP91jz6gizJrzgAize5EIx64htHybDVN9UGZpGSXi4,50703
 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.2.dist-info/licenses/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.127.2.dist-info/METADATA,sha256=vflFwcPWETDwMH45GRhr5McMoghZpm8-hspdiP3qNZs,10084
+sunholo-0.127.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+sunholo-0.127.2.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.127.2.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.127.2.dist-info/RECORD,,