banko-ai-assistant 1.0.17__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/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.17.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.17
+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

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

@@ -7,7 +7,7 @@ banko_ai/ai_providers/base.py,sha256=zbuAgkHIfJ0YkG83LXzieJuvXBcB2-nx7NhbL-I4Pf0
 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.17.dist-info/licenses/LICENSE,sha256=skG0LkywIClj8fgSIXiG6o9vUDJ678BKBObIyJ19OMw,1075
-banko_ai_assistant-1.0.17.dist-info/METADATA,sha256=s8CC5JnedQVmaO-rZgulNvFekvxR1KYcUsbYysVzdoY,15573
-banko_ai_assistant-1.0.17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-banko_ai_assistant-1.0.17.dist-info/entry_points.txt,sha256=IxPjBjMvbpCp-ikCA43bOSbYboTGPX4HYcZlvu2_vcA,47
-banko_ai_assistant-1.0.17.dist-info/top_level.txt,sha256=xNMa9Z67UssefOQ2ubFObtqUYIfYmCIclfz0xdo5OPE,9
-banko_ai_assistant-1.0.17.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,,