banko-ai-assistant 1.0.14__tar.gz → 1.0.15__tar.gz

{banko_ai_assistant-1.0.14/banko_ai_assistant.egg-info → banko_ai_assistant-1.0.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: banko-ai-assistant
-Version: 1.0.14
+Version: 1.0.15
 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.14 → banko_ai_assistant-1.0.15}/banko_ai/vector_search/generator.py

@@ -86,46 +86,56 @@ class EnhancedExpenseGenerator:
             "Internet Provider", "Phone Company", "Water Company"
         ]
         
-        # Use the exact categories from the original CSV
+        # Use the exact categories from the original CSV with appropriate merchants
         self._categories = {
             "Groceries": {
                 "items": ["Fresh produce", "Dairy products", "Meat and poultry", "Pantry staples", "Organic foods", "Beverages", "Snacks"],
+                "merchants": ["Whole Foods", "Local Market", "Costco", "Walmart", "Target"],
                 "amount_range": (10, 150)
             },
             "Home Improvement": {
                 "items": ["Tools", "Hardware", "Paint", "Lumber", "Electrical supplies", "Plumbing supplies", "Garden supplies"],
+                "merchants": ["Home Depot", "Lowe's", "Ace Hardware", "IKEA"],
                 "amount_range": (20, 500)
             },
             "Electronics": {
                 "items": ["Smartphone", "Laptop", "Tablet", "Headphones", "Camera", "Gaming console", "Smart home device"],
+                "merchants": ["Apple Store", "Best Buy", "Amazon", "Target", "Walmart"],
                 "amount_range": (50, 1000)
             },
             "Subscription": {
                 "items": ["Streaming service", "Software subscription", "Gym membership", "News subscription", "Cloud storage", "Music service"],
+                "merchants": ["Netflix", "Spotify", "Planet Fitness", "Electric Company", "Internet Provider"],
                 "amount_range": (10, 50)
             },
             "Shopping": {
                 "items": ["Clothing", "Shoes", "Accessories", "Home decor", "Books", "Toys", "Beauty products"],
+                "merchants": ["Amazon", "Target", "Walmart", "IKEA", "Best Buy"],
                 "amount_range": (15, 200)
             },
             "Restaurant": {
                 "items": ["Dinner", "Lunch", "Breakfast", "Takeout", "Delivery", "Catering", "Fine dining"],
+                "merchants": ["McDonald's", "Italian Bistro", "Chipotle", "Subway", "Pizza Hut", "Domino's"],
                 "amount_range": (15, 100)
             },
             "Transport": {
                 "items": ["Uber ride", "Lyft ride", "Taxi", "Bus fare", "Train ticket", "Flight", "Car rental"],
+                "merchants": ["Uber", "Lyft", "Delta Airlines"],
                 "amount_range": (5, 500)
             },
             "Fuel": {
                 "items": ["Gas fill-up", "Electric charging", "Diesel fuel", "Premium gas", "Regular gas"],
+                "merchants": ["Shell Gas Station", "Tesla Supercharger", "Exxon", "Chevron", "BP"],
                 "amount_range": (20, 100)
             },
             "Travel": {
                 "items": ["Flight", "Hotel", "Car rental", "Travel insurance", "Airport parking", "Baggage fee"],
+                "merchants": ["Delta Airlines", "Hilton Hotels"],
                 "amount_range": (100, 2000)
             },
             "Coffee": {
                 "items": ["Coffee", "Espresso", "Latte", "Cappuccino", "Pastry", "Sandwich", "Breakfast"],
+                "merchants": ["Starbucks", "Local Market", "Whole Foods", "Costco"],
                 "amount_range": (3, 25)
             }
         }
@@ -142,8 +152,8 @@ class EnhancedExpenseGenerator:
         category = random.choice(list(self.categories.keys()))
         category_data = self.categories[category]
         
-        # Select merchant from the full merchant list (matching original CSV)
-        merchant = random.choice(self.merchants)
+        # Select merchant from category-specific merchants
+        merchant = random.choice(category_data["merchants"])
         
         # Generate amount within category range
         amount = round(random.uniform(*category_data["amount_range"]), 2)
@@ -242,10 +252,12 @@ class EnhancedExpenseGenerator:
                         df = pd.DataFrame(batch)
                         df.to_sql('expenses', conn, if_exists='append', index=False, method='multi')
                         # 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
+                        
+                    # Only increment counter after successful transaction
+                    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:
                     # Check if it's a CockroachDB serialization failure (SQL state 40001)

{banko_ai_assistant-1.0.14 → banko_ai_assistant-1.0.15/banko_ai_assistant.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: banko-ai-assistant
-Version: 1.0.14
+Version: 1.0.15
 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.14 → banko_ai_assistant-1.0.15}/pyproject.toml

@@ -7,7 +7,7 @@ py-modules = []
 
 [project]
 name = "banko-ai-assistant"
-version = "1.0.14"
+version = "1.0.15"
 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" }