banko-ai-assistant 1.0.9__py3-none-any.whl → 1.0.11__py3-none-any.whl

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/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.dist-info → banko_ai_assistant-1.0.11.dist-info}/METADATA

@@ -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.dist-info → banko_ai_assistant-1.0.11.dist-info}/RECORD

@@ -1,6 +1,6 @@
 banko_ai/__init__.py,sha256=G1InyKemqQxP9xx6yGZgolBmrmOLSpBXqGYY8LaFOeo,568
 banko_ai/__main__.py,sha256=vySUZt0uAzogVu6yURIlLJ1Ig5bdxT_55oiiaiZKqA8,185
-banko_ai/cli.py,sha256=LgnGaUqsbtrymOru1IJsgTW98rN3zoq6UGKkbR8ojs8,13732
+banko_ai/cli.py,sha256=SsCsikOykmTM_uId-n0ivilKXu1hKch17XXwhMHfFnU,13760
 banko_ai/ai_providers/__init__.py,sha256=JdBgw5Mji2pe9nU-aiRYUmJuZk0q8KbcMtbpMJC5Dq8,483
 banko_ai/ai_providers/aws_provider.py,sha256=-tR-8tlEeSL-Fspx05tTMFguvQylkW_pz0PI2XJEByM,13074
 banko_ai/ai_providers/base.py,sha256=zbuAgkHIfJ0YkG83LXzieJuvXBcB2-nx7NhbL-I4Pf0,4725
@@ -151,14 +151,14 @@ banko_ai/utils/database.py,sha256=sJYAFTApkWReEJuMbbBDiz7XfgiiEd6lPSSyF6BQDpk,77
 banko_ai/utils/migration.py,sha256=j1lYUVZyYMcMvxZUOFymoK19QTPqkDZFXD-iysVCnQo,4764
 banko_ai/vector_search/__init__.py,sha256=vYksnkUU4FA8XBNzYZIH4FoGjXCx9oIbrDeapSzrNuE,621
 banko_ai/vector_search/enrichment.py,sha256=sRnFLNG9WGfq8j44T7krxHI-Lc2RSH68mt0IT0GTHBA,10203
-banko_ai/vector_search/generator.py,sha256=l5DdeAiPBHLkbMnFvX03rTSn4b9JS2jiP_Pf_A6Lxpk,16437
+banko_ai/vector_search/generator.py,sha256=BAs26Uwt-8VP9xPsTdooxznLnr674uX_jr41_oKZKL8,16912
 banko_ai/vector_search/search.py,sha256=k3wo3zFJH9o-kBc-vAS-bdmM3LGX2vIk3u4cRA5QPXo,16915
 banko_ai/web/__init__.py,sha256=hjWVVxYpIZhOAN1qBf4xTd36a5AUHM03Q8BF8pykhJQ,363
 banko_ai/web/app.py,sha256=9xBe6RsyEUQvpDhsRckJ4pzk5x2RZWEhVs8DiJg4zqs,27675
 banko_ai/web/auth.py,sha256=js6qIixSFHyLbETDm8GNLCPrDkCDcaQZPFOrqtZP1uw,2125
-banko_ai_assistant-1.0.9.dist-info/licenses/LICENSE,sha256=skG0LkywIClj8fgSIXiG6o9vUDJ678BKBObIyJ19OMw,1075
-banko_ai_assistant-1.0.9.dist-info/METADATA,sha256=fRuFpSdfmLnibvN9_9DAN-AyTDY6a2hTsAIFOp2E1wc,13243
-banko_ai_assistant-1.0.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-banko_ai_assistant-1.0.9.dist-info/entry_points.txt,sha256=IxPjBjMvbpCp-ikCA43bOSbYboTGPX4HYcZlvu2_vcA,47
-banko_ai_assistant-1.0.9.dist-info/top_level.txt,sha256=xNMa9Z67UssefOQ2ubFObtqUYIfYmCIclfz0xdo5OPE,9
-banko_ai_assistant-1.0.9.dist-info/RECORD,,
+banko_ai_assistant-1.0.11.dist-info/licenses/LICENSE,sha256=skG0LkywIClj8fgSIXiG6o9vUDJ678BKBObIyJ19OMw,1075
+banko_ai_assistant-1.0.11.dist-info/METADATA,sha256=FSL0Rqnj0OMpI_pM6mxE0VReiQblRa3tT5AZ-1QpLQA,13244
+banko_ai_assistant-1.0.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+banko_ai_assistant-1.0.11.dist-info/entry_points.txt,sha256=IxPjBjMvbpCp-ikCA43bOSbYboTGPX4HYcZlvu2_vcA,47
+banko_ai_assistant-1.0.11.dist-info/top_level.txt,sha256=xNMa9Z67UssefOQ2ubFObtqUYIfYmCIclfz0xdo5OPE,9
+banko_ai_assistant-1.0.11.dist-info/RECORD,,