fix: prevent test collection failures from optional modules

Fixes test collection issues when optional dependencies (praisonaiui,
google.generativeai) are not installed in base setup.

Changes:
- praisonai/inc/models.py: Add try/catch guards around importlib.util.find_spec()
  calls to handle missing parent packages gracefully
- praisonai/ui/_aiui_datastore.py: Move dependency imports to lazy loading
  in __init__ to allow module import during test collection

Resolves #1541

Co-authored-by: praisonai-triage-agent[bot] <praisonai-triage-agent[bot]@users.noreply.github.com>

Author: praisonai-triage-agent[bot]

src/praisonai/praisonai/inc/models.py

@@ -16,9 +16,22 @@
 # Use find_spec for fast availability checks (no actual import)
 # This avoids the ~3200ms langchain_openai import at module load
 OPENAI_AVAILABLE = importlib.util.find_spec("openai") is not None
-GOOGLE_GENAI_AVAILABLE = importlib.util.find_spec("google.generativeai") is not None
-ANTHROPIC_AVAILABLE = importlib.util.find_spec("anthropic") is not None
-COHERE_AVAILABLE = importlib.util.find_spec("cohere") is not None
+
+# Guard against missing parent packages (fixes test collection in base install)
+try:
+    GOOGLE_GENAI_AVAILABLE = importlib.util.find_spec("google.generativeai") is not None
+except (ImportError, AttributeError, ValueError):
+    GOOGLE_GENAI_AVAILABLE = False
+
+try:
+    ANTHROPIC_AVAILABLE = importlib.util.find_spec("anthropic") is not None
+except (ImportError, AttributeError, ValueError):
+    ANTHROPIC_AVAILABLE = False
+
+try:
+    COHERE_AVAILABLE = importlib.util.find_spec("cohere") is not None
+except (ImportError, AttributeError, ValueError):
+    COHERE_AVAILABLE = False
 
 # Lazy import helpers for provider SDKs
 def _get_openai_client():

src/praisonai/praisonai/ui/_aiui_datastore.py

@@ -11,30 +11,40 @@
 
 logger = logging.getLogger(__name__)
 
-# Fail loudly on missing optional dependencies per AGENTS.md §4.2
-try:
-    from praisonaiui.datastore import BaseDataStore
-except ImportError as e:
-    raise ImportError(
-        "praisonaiui is required for PraisonAISessionDataStore. "
-        "Install with: pip install 'praisonai[ui]'"
-    ) from e
 
-try:
-    from praisonaiagents.session import SessionStoreProtocol
-    from praisonaiagents.session import get_hierarchical_session_store
-except ImportError as e:
-    raise ImportError(
-        "praisonaiagents is required for PraisonAISessionDataStore. "
-        "Install with: pip install praisonaiagents"
-    ) from e
-
-
-class PraisonAISessionDataStore(BaseDataStore):
-    """Adapter that bridges PraisonAI SessionStoreProtocol to aiui BaseDataStore."""
+class PraisonAISessionDataStore:
+    """Adapter that bridges PraisonAI SessionStoreProtocol to aiui BaseDataStore.
     
-    def __init__(self, store: Optional[SessionStoreProtocol] = None):
-        """Initialize with an optional session store, defaults to hierarchical store."""
+    Note: This class defers dependency imports until instantiation to allow
+    test collection without optional modules installed.
+    """
+    
+    def __init__(self, store: Optional[Any] = None):
+        """Initialize with an optional session store, defaults to hierarchical store.
+        
+        Args:
+            store: Optional SessionStoreProtocol implementation
+        """
+        # Lazy import dependencies only when class is instantiated
+        try:
+            from praisonaiui.datastore import BaseDataStore
+            # Make this class inherit from BaseDataStore at runtime
+            if BaseDataStore not in self.__class__.__bases__:
+                self.__class__.__bases__ = (BaseDataStore,) + self.__class__.__bases__
+        except ImportError as e:
+            raise ImportError(
+                "praisonaiui is required for PraisonAISessionDataStore. "
+                "Install with: pip install 'praisonai[ui]'"
+            ) from e
+
+        try:
+            from praisonaiagents.session import get_hierarchical_session_store
+        except ImportError as e:
+            raise ImportError(
+                "praisonaiagents is required for PraisonAISessionDataStore. "
+                "Install with: pip install praisonaiagents"
+            ) from e
+            
         self._store = store or get_hierarchical_session_store()
 
     def _new_id(self) -> str: