more imports, pylint fixes

Author: kovtcharov

src/gaia/agents/blender/agent.py

@@ -80,7 +80,7 @@ def _create_console(self) -> AgentConsole:
     def _get_system_prompt(self) -> str:
         """Generate the system prompt for the Blender agent."""
         # Get formatted tools from registry
-        return f"""
+        return """
 You are a specialized Blender 3D assistant that can create and modify 3D scenes.
 You will use a set of tools to accomplish tasks based on the user's request.
 
@@ -283,7 +283,7 @@ def set_material_color(
                 return {"status": "error", "error": str(e)}
 
         # @tool
-        def get_object_info(name: str) -> Dict[str, Any]:
+        def _get_object_info(name: str) -> Dict[str, Any]:
             """
             Get information about an object in the scene.
 
@@ -354,7 +354,7 @@ def modify_object(
                 return {"status": "error", "error": str(e)}
 
         # @tool
-        def delete_object(name: str) -> Dict[str, Any]:
+        def _delete_object(name: str) -> Dict[str, Any]:
             """
             Delete an object from the scene.
 
@@ -407,7 +407,7 @@ def get_scene_info() -> Dict[str, Any]:
                 return {"status": "error", "error": str(e)}
 
         # @tool
-        def execute_blender_code(code: str) -> Dict[str, Any]:
+        def _execute_blender_code(code: str) -> Dict[str, Any]:
             """
             Execute arbitrary Python code in Blender with error handling.
 
@@ -436,7 +436,7 @@ def execute_blender_code(code: str) -> Dict[str, Any]:
                 return {"status": "error", "error": str(e)}
 
         # @tool
-        def diagnose_scene() -> Dict[str, Any]:
+        def _diagnose_scene() -> Dict[str, Any]:
             """
             Diagnose the current Blender scene for common issues.
             Returns information about objects, materials, and potential problems.

src/gaia/agents/blender/core/view.py

@@ -13,13 +13,13 @@ def __init__(self, mcp: MCPClient):
         self.mcp = mcp
 
     def adjust_for_large_scale(
-        self, clip_end: float = 100000, orbit_selection: bool = True
+        self, clip_end: float = 100000, _orbit_selection: bool = True
     ) -> Dict:
         """Adjust viewport settings to properly view large-scale objects like Earth.
 
         Args:
             clip_end: The maximum view distance to set for the 3D viewport (default: 100000)
-            orbit_selection: Whether to enable orbit around selection (default: True, but may not work in all Blender versions)
+            _orbit_selection: Whether to enable orbit around selection (default: True, but may not work in all Blender versions)
         """
 
         def generate_code():

src/gaia/agents/emr/agent.py

@@ -325,7 +325,7 @@ def _get_vlm(self):
         """Get or create VLM client (lazy initialization)."""
         if self._vlm is None:
             try:
-                from gaia.llm.vlm_client import VLMClient
+                from gaia.llm import VLMClient
 
                 self.console.print_model_loading(self._vlm_model)
                 self._vlm = VLMClient(vlm_model=self._vlm_model)

src/gaia/agents/emr/cli.py

@@ -636,7 +636,7 @@ def cmd_test(args):
 
     from PIL import Image
 
-    from gaia.llm.vlm_client import VLMClient
+    from gaia.llm import VLMClient
 
     file_path = Path(args.file)
     if not file_path.exists():

src/gaia/rag/sdk.py

@@ -432,7 +432,7 @@ def _extract_text_from_pdf(self, pdf_path: str) -> tuple:
             vlm = None
             vlm_available = False
             try:
-                from gaia.llm.vlm_client import VLMClient
+                from gaia.llm import VLMClient
                 from gaia.rag.pdf_utils import (
                     count_images_in_page,
                     extract_images_from_page_pymupdf,

util/lint.ps1

@@ -273,16 +273,21 @@ function Invoke-ImportTests {
         @{Import="from gaia.llm import LLMClient"; Desc="LLM client class"},
         @{Import="from gaia.llm import VLMClient"; Desc="Vision LLM client"},
         @{Import="from gaia.llm import create_client"; Desc="LLM factory"},
+        @{Import="from gaia.llm import NotSupportedError"; Desc="LLM exception"},
 
         # Chat SDK
         @{Module="gaia.chat.sdk"; Desc="Chat SDK module"},
         @{Import="from gaia.chat.sdk import ChatSDK"; Desc="Chat SDK class"},
         @{Import="from gaia.chat.sdk import ChatConfig"; Desc="Chat configuration"},
+        @{Import="from gaia.chat.sdk import ChatSession"; Desc="Chat session"},
+        @{Import="from gaia.chat.sdk import ChatResponse"; Desc="Chat response"},
         @{Import="from gaia.chat.sdk import quick_chat"; Desc="Quick chat function"},
 
         # RAG SDK
         @{Module="gaia.rag.sdk"; Desc="RAG SDK module"},
         @{Import="from gaia.rag.sdk import RAGSDK"; Desc="RAG SDK class"},
+        @{Import="from gaia.rag.sdk import RAGConfig"; Desc="RAG configuration"},
+        @{Import="from gaia.rag.sdk import quick_rag"; Desc="Quick RAG function"},
 
         # Base Agent System
         @{Module="gaia.agents.base.agent"; Desc="Base agent module"},
@@ -295,13 +300,17 @@ function Invoke-ImportTests {
         @{Import="from gaia.agents.code import CodeAgent"; Desc="Code agent"},
         @{Import="from gaia.agents.jira import JiraAgent"; Desc="Jira agent"},
         @{Import="from gaia.agents.docker import DockerAgent"; Desc="Docker agent"},
+        @{Import="from gaia.agents.blender import BlenderAgent"; Desc="Blender agent"},
+        @{Import="from gaia.agents.emr import MedicalIntakeAgent"; Desc="Medical intake agent"},
+        @{Import="from gaia.agents.routing import RoutingAgent"; Desc="Routing agent"},
 
         # Database
         @{Import="from gaia.database import DatabaseAgent"; Desc="Database agent"},
         @{Import="from gaia.database import DatabaseMixin"; Desc="Database mixin"},
 
         # Utilities
-        @{Import="from gaia.utils import FileWatcher"; Desc="File watcher"}
+        @{Import="from gaia.utils import FileWatcher"; Desc="File watcher"},
+        @{Import="from gaia.utils import FileWatcherMixin"; Desc="File watcher mixin"}
     )
 
     $failed = $false