banko-ai-assistant 1.0.16__py3-none-any.whl → 1.0.18__py3-none-any.whl

banko_ai/ai_providers/watsonx_provider.py

@@ -137,13 +137,15 @@ class WatsonxProvider(AIProvider):
     
     def get_available_models(self) -> List[str]:
         """Get list of available Watsonx models."""
+        # Only include models that are actually supported by Watsonx API
         return [
             'openai/gpt-oss-120b',
             'meta-llama/llama-2-70b-chat',
             'meta-llama/llama-2-13b-chat',
-            'meta-llama/llama-2-7b-chat',
-            'ibm/granite-13b-chat-v2',
-            'ibm/granite-13b-instruct-v2'
+            'meta-llama/llama-2-7b-chat'
+            # Note: IBM Granite models may not be available in all regions/projects
+            # 'ibm/granite-13b-chat-v2',
+            # 'ibm/granite-13b-instruct-v2'
         ]
     
     def set_model(self, model_id: str) -> bool:
@@ -159,10 +161,12 @@ class WatsonxProvider(AIProvider):
             return False
         
         try:
-            test_messages = [{"role": "user", "content": "Hello, this is a test message."}]
-            response = self._call_watsonx_api(test_messages)
-            return True
+            # Just test if we can get an access token - this is faster and more reliable
+            # than making a full API call for status checks
+            access_token = self._get_access_token()
+            return access_token is not None
         except Exception as e:
+            # If token request fails, the connection is not available
             return False
     
     def _get_access_token(self):

banko_ai/cli.py

@@ -260,6 +260,12 @@ def help():
 
 This is a modern AI-powered expense analysis application with RAG capabilities.
 
+PREREQUISITES:
+--------------
+- CockroachDB v25.2.4+ (recommended: v25.3.1)
+- Vector index feature enabled: SET CLUSTER SETTING feature.vector_index.enabled = true;
+- Start single node: cockroach start-single-node --insecure --store=./cockroach-data --listen-addr=localhost:26257 --http-addr=localhost:8080 --background
+
 QUICK START:
 -----------
 1. Set up your environment variables:

banko_ai/templates/dashboard.html

@@ -515,6 +515,7 @@
                                     <div class="flex items-center space-x-2">
                                         <i class="fas fa-${data.ai_provider === 'connected' ? 'check-circle text-green-500' : 'times-circle text-red-500'}"></i>
                                         <span class="text-sm text-cockroach-text-light">AI Provider (${data.ai_service}): ${data.ai_provider}</span>
+                                        ${data.ai_provider === 'disconnected' && data.ai_provider_available ? '<span class="text-xs text-yellow-500 ml-1">(Model switched, connection test failed)</span>' : ''}
                                     </div>
                                     <div class="flex items-center space-x-2">
                                         <i class="fas fa-robot text-blue-500"></i>

banko_ai/web/app.py

@@ -452,7 +452,9 @@ def create_app() -> Flask:
             # Check AI provider
             ai_status = "unknown"
             current_model = "unknown"
+            ai_provider_available = False
             if ai_provider:
+                ai_provider_available = True
                 ai_status = "connected" if ai_provider.test_connection() else "disconnected"
                 current_model = ai_provider.get_current_model()
             
@@ -461,7 +463,8 @@ def create_app() -> Flask:
                 'database': 'connected',
                 'ai_provider': ai_status,
                 'ai_service': config.ai_service,
-                'current_model': current_model
+                'current_model': current_model,
+                'ai_provider_available': ai_provider_available
             })
             
         except Exception as e:

{banko_ai_assistant-1.0.16.dist-info → banko_ai_assistant-1.0.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: banko-ai-assistant
-Version: 1.0.16
+Version: 1.0.18
 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
@@ -30,9 +30,9 @@ Requires-Dist: jinja2<4.0.0,>=3.1.0
 Requires-Dist: psycopg2-binary<3.0.0,>=2.9.0
 Requires-Dist: sqlalchemy<3.0.0,>=2.0.0
 Requires-Dist: sqlalchemy-cockroachdb<3.0.0,>=2.0.0
-Requires-Dist: sentence-transformers<3.0.0,>=2.2.0
-Requires-Dist: boto3<1.35.0,>=1.34.0
-Requires-Dist: botocore<1.35.0,>=1.34.0
+Requires-Dist: sentence-transformers<4.0.0,>=3.1.0
+Requires-Dist: boto3<1.41.0,>=1.40.0
+Requires-Dist: botocore<1.41.0,>=1.40.0
 Requires-Dist: openai<2.0.0,>=1.11.0
 Requires-Dist: requests<3.0.0,>=2.32.4
 Requires-Dist: numpy<2.0.0,>=1.26.0
@@ -76,9 +76,48 @@ A modern AI-powered expense analysis application with Retrieval-Augmented Genera
 
 ### Prerequisites
 
-- Python 3.8+
-- CockroachDB (running locally or cloud)
-- AI Provider API Key (OpenAI, AWS, IBM Watsonx, or Google Gemini)
+- **Python 3.8+**
+- **CockroachDB v25.2.4+** (recommended: [v25.3.1](https://www.cockroachlabs.com/docs/releases/v25.3#v25-3-1))
+- **Vector Index Feature Enabled** (required for vector search)
+- **AI Provider API Key** (OpenAI, AWS, IBM Watsonx, or Google Gemini)
+
+#### CockroachDB Setup
+
+1. **Download and Install CockroachDB**:
+   ```bash
+   # Download CockroachDB v25.3.1 (recommended)
+   # Visit: https://www.cockroachlabs.com/docs/releases/v25.3#v25-3-1
+   
+   # Or install via package manager
+   brew install cockroachdb/tap/cockroach  # macOS
+   ```
+
+2. **Start CockroachDB Single Node**:
+   ```bash
+   # Start a single-node cluster (for development)
+   cockroach start-single-node \
+     --insecure \
+     --store=./cockroach-data \
+     --listen-addr=localhost:26257 \
+     --http-addr=localhost:8080 \
+     --background
+   ```
+
+3. **Enable Vector Index Feature**:
+   ```sql
+   -- Connect to the database
+   cockroach sql --url="cockroachdb://root@localhost:26257/defaultdb?sslmode=disable"
+   
+   -- Enable vector index feature (required for vector search)
+   SET CLUSTER SETTING feature.vector_index.enabled = true;
+   ```
+
+4. **Verify Setup**:
+   ```sql
+   -- Check if vector index is enabled
+   SHOW CLUSTER SETTING feature.vector_index.enabled;
+   -- Should return: true
+   ```
 
 ### Installation
 
@@ -94,7 +133,7 @@ export WATSONX_MODEL_ID="openai/gpt-oss-120b"
 export DATABASE_URL="cockroachdb://root@localhost:26257/defaultdb?sslmode=disable"
 
 # Run the application
-python -m banko_ai
+python -m banko-ai
 ```
 
 #### Option 2: Development Installation
@@ -230,14 +269,14 @@ banko-ai help
 
 ## 🔌 API Endpoints
 
-| Endpoint | Method | Description |
-|----------|--------|-------------|
-| `/` | GET | Web interface |
-| `/api/health` | GET | System health check |
-| `/api/ai-providers` | GET | Available AI providers |
-| `/api/models` | GET | Available models for current provider |
-| `/api/search` | POST | Vector search expenses |
-| `/api/rag` | POST | RAG-based Q&A |
+| Endpoint            | Method | Description                           |
+|---------------------|--------|---------------------------------------|
+| `/`                 | GET    | Web interface                         |
+| `/api/health`       | GET    | System health check                   |
+| `/api/ai-providers` | GET    | Available AI providers                |
+| `/api/models`       | GET    | Available models for current provider |
+| `/api/search`       | POST   | Vector search expenses                |
+| `/api/rag`          | POST   | RAG-based Q&A                         |
 
 ### API Examples
 
@@ -366,10 +405,36 @@ banko_ai/
 
 ### Common Issues
 
+**CockroachDB Version Issues**
+```bash
+# Check CockroachDB version (must be v25.2.4+)
+cockroach version
+
+# If version is too old, download v25.3.1:
+# https://www.cockroachlabs.com/docs/releases/v25.3#v25-3-1
+```
+
+**Vector Index Feature Not Enabled**
+```bash
+# Connect to database and enable vector index feature
+cockroach sql --url="cockroachdb://root@localhost:26257/defaultdb?sslmode=disable"
+
+# Enable vector index feature
+SET CLUSTER SETTING feature.vector_index.enabled = true;
+
+# Verify it's enabled
+SHOW CLUSTER SETTING feature.vector_index.enabled;
+```
+
 **Database Connection Error**
 ```bash
-# Check CockroachDB is running
-cockroach start --insecure --listen-addr=localhost:26257
+# Start CockroachDB single node
+cockroach start-single-node \
+  --insecure \
+  --store=./cockroach-data \
+  --listen-addr=localhost:26257 \
+  --http-addr=localhost:8080 \
+  --background
 
 # Verify database exists
 cockroach sql --url="cockroachdb://root@localhost:26257/defaultdb?sslmode=disable" --execute "SHOW TABLES;"
@@ -417,4 +482,4 @@ For issues and questions:
 
 ---
 
-**Built with ❤️ using CockroachDB, Flask, and modern AI technologies**
+**Built with ❤️ using CockroachDB, Flask, and modern AI technologies such as **

{banko_ai_assistant-1.0.16.dist-info → banko_ai_assistant-1.0.18.dist-info}/RECORD

@@ -1,13 +1,13 @@
 banko_ai/__init__.py,sha256=G1InyKemqQxP9xx6yGZgolBmrmOLSpBXqGYY8LaFOeo,568
 banko_ai/__main__.py,sha256=U-KkrXtL8JNIyV25PE5v_eYhlhjR7jd6kG-txfYfs0M,709
-banko_ai/cli.py,sha256=7lMeDRIuJjeAkrVMzw2YmJp4t10KLGP7TRIywUSOkeU,13884
+banko_ai/cli.py,sha256=goTkY3qhTrGvmERbapKztYH-g7fAemxThZJSO7Ll2HI,14205
 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
 banko_ai/ai_providers/factory.py,sha256=Bqq9HcbyTfPvaOTxsHSM9eSvkB71cJoq21cMmXo4LLc,2885
 banko_ai/ai_providers/gemini_provider.py,sha256=KqzHLLl7EYnai9-zFenRmktVk0zOA8AtsYScQZIcdLU,13044
 banko_ai/ai_providers/openai_provider.py,sha256=Myu2And6kTD2EgIVcWHGak5fDIq0pu1HQzf-jj72y3k,11657
-banko_ai/ai_providers/watsonx_provider.py,sha256=Jl33LKApIED4nl2EmhpSp3x7aFYe4d2QSqq5udtGlZg,39258
+banko_ai/ai_providers/watsonx_provider.py,sha256=PlS0OiJau0SbENH3f2UTz3QqExARPEMmOFecr5dkG-g,39556
 banko_ai/config/__init__.py,sha256=YObKfKjjW89kSfARiTzXnGuSPz1C92aSMKgmO3BtQb8,133
 banko_ai/config/settings.py,sha256=6M8YoaxzyCULn6BOot4hahtDkgjsgyLmMd8BxDKVt4k,9317
 banko_ai/static/Anallytics.png,sha256=fWLddd5hlB4YEUYSIzOFVNnTny6D8VExQeVn31mywTc,80242
@@ -142,7 +142,7 @@ banko_ai/static/creditcard_files/vendor-react-76e71fd67c83934b.mjs,sha256=LCx9dM
 banko_ai/static/creditcard_files/visual-search-fe95849c7568d600.mjs,sha256=dsIqV8HCK5_oYVudHyTDnXpfXiggV1a0x6JjFPLT4eE,267636
 banko_ai/static/css/style.css,sha256=9Vz6EC3xLtqOsrSxYX3-W24ZnSk1eDSsUl6-bICjnEI,24116
 banko_ai/templates/base.html,sha256=s8Dfa0Eg9Lze6zeruVSbXcLJtv-Vo7KhKvk4M3HIhxM,2237
-banko_ai/templates/dashboard.html,sha256=-HIQ3sU86hqVn1jGeHV8_w2HlnPZ4uS2cD-BNGUaroU,27674
+banko_ai/templates/dashboard.html,sha256=jSKF4HqJHilatXsftXMZJMFPtp8jm0qWKa9r3yya8dc,27883
 banko_ai/templates/index.html,sha256=SkDXWH_ttH2T_a_46_OETgy1Q3zDfvot9eEKGB0S1x0,66973
 banko_ai/templates/login.html,sha256=YPMtJcvCzFlknwmUrG7VskeM691J4msAjZw-t4CcPn4,2063
 banko_ai/utils/__init__.py,sha256=0n1JYzZUWwgwOzV82I6OnmfUV_TOnal1V0DoEb0E2Cs,225
@@ -154,11 +154,11 @@ banko_ai/vector_search/enrichment.py,sha256=tgAImLehkp2kL46vI5GEHsE8B5E4gT3PweXZ
 banko_ai/vector_search/generator.py,sha256=6SV6_RZ4M4-TfJvxlXHZY1p-zcLKpIJm0p871nmEjmI,17837
 banko_ai/vector_search/search.py,sha256=RiDxqcRblAFSmxMvxy9nRRrw22OtVMHP5_rsvQsS_MU,18763
 banko_ai/web/__init__.py,sha256=hjWVVxYpIZhOAN1qBf4xTd36a5AUHM03Q8BF8pykhJQ,363
-banko_ai/web/app.py,sha256=Temx39TX19idHrlrUwb_2QLr1GSZZVjqhyJC299NFMY,32145
+banko_ai/web/app.py,sha256=dmr-fIKFmN93vijnTrBWbuz7rg3FT-iRZaBPvUJCYFY,32296
 banko_ai/web/auth.py,sha256=js6qIixSFHyLbETDm8GNLCPrDkCDcaQZPFOrqtZP1uw,2125
-banko_ai_assistant-1.0.16.dist-info/licenses/LICENSE,sha256=skG0LkywIClj8fgSIXiG6o9vUDJ678BKBObIyJ19OMw,1075
-banko_ai_assistant-1.0.16.dist-info/METADATA,sha256=BuVSoi6etHKRv2qq6rLW5FucRSzlUSuZip87XXi0y-g,13505
-banko_ai_assistant-1.0.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-banko_ai_assistant-1.0.16.dist-info/entry_points.txt,sha256=IxPjBjMvbpCp-ikCA43bOSbYboTGPX4HYcZlvu2_vcA,47
-banko_ai_assistant-1.0.16.dist-info/top_level.txt,sha256=xNMa9Z67UssefOQ2ubFObtqUYIfYmCIclfz0xdo5OPE,9
-banko_ai_assistant-1.0.16.dist-info/RECORD,,
+banko_ai_assistant-1.0.18.dist-info/licenses/LICENSE,sha256=skG0LkywIClj8fgSIXiG6o9vUDJ678BKBObIyJ19OMw,1075
+banko_ai_assistant-1.0.18.dist-info/METADATA,sha256=-9ZE8-5drz7Ms8IHohKrAqQT7nsLnVIX6vCd4KKmEZI,15573
+banko_ai_assistant-1.0.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+banko_ai_assistant-1.0.18.dist-info/entry_points.txt,sha256=IxPjBjMvbpCp-ikCA43bOSbYboTGPX4HYcZlvu2_vcA,47
+banko_ai_assistant-1.0.18.dist-info/top_level.txt,sha256=xNMa9Z67UssefOQ2ubFObtqUYIfYmCIclfz0xdo5OPE,9
+banko_ai_assistant-1.0.18.dist-info/RECORD,,