dhisana 0.0.1.dev308__tar.gz → 0.0.1.dev309__tar.gz

{dhisana-0.0.1.dev308 → dhisana-0.0.1.dev309}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dhisana
-Version: 0.0.1.dev308
+Version: 0.0.1.dev309
 Summary: A Python SDK for Dhisana AI Platform
 Home-page: https://github.com/dhisana-ai/dhisana-python-sdk
 Author: Admin

{dhisana-0.0.1.dev308 → dhisana-0.0.1.dev309}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='dhisana',  
-    version='0.0.1-dev308',
+    version='0.0.1-dev309',
     description='A Python SDK for Dhisana AI Platform',
     author='Admin',
     author_email='contact@dhisana.ai',

{dhisana-0.0.1.dev308 → dhisana-0.0.1.dev309}/src/dhisana/utils/test_connect.py

@@ -3,7 +3,7 @@ import logging
 import asyncio
 import os
 from datetime import datetime, timedelta, timezone
-from typing import Awaitable, Callable, Dict, List, Any, Optional
+from typing import Awaitable, Callable, Dict, List, Any, Optional, Union
 import requests
 
 try:
@@ -2106,17 +2106,17 @@ async def test_dataverse(
 
 
 async def test_gemini(
-    api_key: str = "",
+    api_key: Union[str, Dict[str, Any]] = "",
     use_vertex_ai: bool = False,
     project: str = "",
     location: str = "us-central1",
 ) -> Dict[str, Any]:
-    """Verify Gemini connectivity by issuing a minimal generate_content call.
+    """Verify Gemini connectivity by listing available models.
 
     Args:
-        api_key: A Gemini API key *or* a service-account JSON string.
-            When a service-account JSON is detected, Vertex AI mode is
-            enabled automatically.
+        api_key: A Gemini API key, a service-account JSON string, or an
+            already-parsed service-account ``dict``.  When a service-account
+            JSON is detected, Vertex AI mode is enabled automatically.
         use_vertex_ai: Explicitly enable Vertex AI mode.
         project: GCP project ID (required for Vertex AI).
         location: GCP region for Vertex AI (default ``us-central1``).
@@ -2131,15 +2131,28 @@ async def test_gemini(
         }
 
     try:
-        # Try to parse service account JSON from the api_key field
-        sa_info = None
-        if api_key and api_key.strip().startswith("{"):
-            try:
-                parsed = json.loads(api_key)
-                if isinstance(parsed, dict) and parsed.get("type") == "service_account":
-                    sa_info = parsed
-            except (json.JSONDecodeError, TypeError):
-                pass
+        # Normalise api_key: it may arrive as a dict (already-parsed
+        # service-account JSON from Pydantic config deserialization).
+        if isinstance(api_key, dict):
+            if api_key.get("type") == "service_account":
+                sa_info = api_key
+            else:
+                return {
+                    "success": False,
+                    "status_code": 0,
+                    "error_message": "api_key is a dict but not a valid service account JSON (missing \"type\": \"service_account\").",
+                }
+            api_key = ""  # not a plain API key string
+        else:
+            # Try to parse service account JSON from the api_key field
+            sa_info = None
+            if api_key and api_key.strip().startswith("{"):
+                try:
+                    parsed = json.loads(api_key)
+                    if isinstance(parsed, dict) and parsed.get("type") == "service_account":
+                        sa_info = parsed
+                except (json.JSONDecodeError, TypeError):
+                    pass
 
         # Validate required fields in service-account JSON
         if sa_info:
@@ -2195,17 +2208,27 @@ async def test_gemini(
                 "error_message": "No API key or Vertex AI project configured.",
             }
 
-        response = await client.aio.models.generate_content(
-            model="gemini-2.0-flash-lite",
-            contents="Reply with exactly: OK",
-        )
+        # Use models.list to verify credentials without depending on a
+        # specific model being available in the configured location.
+        found_any = False
+        models_iter = await client.aio.models.list(config={"page_size": 1})
+        async for _ in models_iter:
+            found_any = True
+            break
+        if not found_any:
+            return {
+                "success": False,
+                "status_code": 0,
+                "error_message": "Authentication succeeded but no models are accessible. "
+                                 "Check your GCP project permissions or API key.",
+            }
         # Any successful API response proves connectivity, regardless of
         # whether the model returned text (safety filters may suppress it).
         return {"success": True, "status_code": 200, "error_message": None}
     except Exception as exc:
         err_msg = str(exc)
         # Avoid leaking credentials in error messages
-        if api_key and api_key in err_msg:
+        if isinstance(api_key, str) and api_key and api_key in err_msg:
             err_msg = err_msg.replace(api_key, "<REDACTED>")
         logger.error(f"Gemini connectivity test failed: {err_msg}")
         return {"success": False, "status_code": 0, "error_message": err_msg}

{dhisana-0.0.1.dev308 → dhisana-0.0.1.dev309}/src/dhisana.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dhisana
-Version: 0.0.1.dev308
+Version: 0.0.1.dev309
 Summary: A Python SDK for Dhisana AI Platform
 Home-page: https://github.com/dhisana-ai/dhisana-python-sdk
 Author: Admin