banko-ai-assistant 1.0.9__tar.gz → 1.0.11__tar.gz

{banko_ai_assistant-1.0.9/banko_ai_assistant.egg-info → banko_ai_assistant-1.0.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: banko-ai-assistant
-Version: 1.0.9
+Version: 1.0.11
 Summary: AI-powered expense analysis and RAG system with CockroachDB vector search and multi-provider AI support
 Author-email: Virag Tripathi <virag.tripathi@gmail.com>
 License-Expression: MIT

{banko_ai_assistant-1.0.9 → banko_ai_assistant-1.0.11}/banko_ai/cli.py

@@ -73,8 +73,8 @@ def run(host, port, debug, generate_data, no_data, clear_data, background):
                     click.echo("Clearing existing data...")
                     generator.clear_expenses()
                 
-                generator.generate_and_save(generate_data)
-                click.echo(f"✅ Successfully generated {generate_data} expense records")
+                actual_count = generator.generate_and_save(generate_data)
+                click.echo(f"✅ Successfully generated {actual_count} expense records")
         except Exception as e:
             click.echo(f"❌ Error generating data: {e}")
             return
@@ -220,8 +220,8 @@ def start(host, port, generate_data, no_data, clear_data):
                     click.echo("Clearing existing data...")
                     generator.clear_expenses()
                 
-                generator.generate_and_save(generate_data)
-                click.echo(f"✅ Successfully generated {generate_data} expense records")
+                actual_count = generator.generate_and_save(generate_data)
+                click.echo(f"✅ Successfully generated {actual_count} expense records")
         except Exception as e:
             click.echo(f"❌ Error generating data: {e}")
             return

{banko_ai_assistant-1.0.9 → banko_ai_assistant-1.0.11}/banko_ai/vector_search/generator.py

@@ -250,6 +250,9 @@ class EnhancedExpenseGenerator:
         # Insert in smaller batches to reduce transaction conflicts
         batch_size = 50  # Reduced from 100 to minimize conflicts
         total_inserted = 0
+        total_batches = (len(data_to_insert) + batch_size - 1) // batch_size
+        
+        print(f"📊 Inserting {len(data_to_insert)} records in {total_batches} batches of {batch_size}")
         
         for i in range(0, len(data_to_insert), batch_size):
             batch = data_to_insert[i:i + batch_size]
@@ -260,12 +263,14 @@ class EnhancedExpenseGenerator:
             
             while retry_count < max_retries:
                 try:
-                    with self.engine.connect() as conn:
+                    with self.engine.begin() as conn:
                         # Use pandas to insert the batch
                         df = pd.DataFrame(batch)
                         df.to_sql('expenses', conn, if_exists='append', index=False, method='multi')
-                        conn.commit()
+                        # Transaction is automatically committed when exiting the context
                         total_inserted += len(batch)
+                        batch_num = i//batch_size + 1
+                        print(f"✅ Batch {batch_num}/{total_batches}: {len(batch)} records inserted (Total: {total_inserted})")
                         break  # Success, exit retry loop
                         
                 except OperationalError as e:
@@ -306,9 +311,9 @@ class EnhancedExpenseGenerator:
         
         while retry_count < max_retries:
             try:
-                with self.engine.connect() as conn:
+                with self.engine.begin() as conn:
                     conn.execute(text("DELETE FROM expenses"))
-                    conn.commit()
+                    # Transaction is automatically committed when exiting the context
                     return True
                     
             except OperationalError as e:

{banko_ai_assistant-1.0.9 → banko_ai_assistant-1.0.11/banko_ai_assistant.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: banko-ai-assistant
-Version: 1.0.9
+Version: 1.0.11
 Summary: AI-powered expense analysis and RAG system with CockroachDB vector search and multi-provider AI support
 Author-email: Virag Tripathi <virag.tripathi@gmail.com>
 License-Expression: MIT

{banko_ai_assistant-1.0.9 → banko_ai_assistant-1.0.11}/pyproject.toml

@@ -7,7 +7,7 @@ py-modules = []
 
 [project]
 name = "banko-ai-assistant"
-version = "1.0.9"
+version = "1.0.11"
 description = "AI-powered expense analysis and RAG system with CockroachDB vector search and multi-provider AI support"
 authors = [
     { name="Virag Tripathi", email="virag.tripathi@gmail.com" }