✨ image to text tool v2

Author: Zhi-a

test/backend/agents/test_create_agent_info.py

@@ -1,12 +1,12 @@
 import pytest
 import sys
 import types
-from pathlib import Path
 from unittest.mock import AsyncMock, MagicMock, patch, Mock, PropertyMock
 
-PROJECT_ROOT = Path(__file__).resolve().parents[3]
-if str(PROJECT_ROOT) not in sys.path:
-    sys.path.insert(0, str(PROJECT_ROOT))
+from test.common.env_test_utils import bootstrap_env
+
+env_state = bootstrap_env()
+consts_const = env_state["mock_const"]
 
 # Utilities ---------------------------------------------------------------
 def _create_stub_module(name: str, **attrs):
@@ -17,28 +17,21 @@ def _create_stub_module(name: str, **attrs):
     return module
 
 
-# Mock consts module first to avoid ModuleNotFoundError
-consts_mock = MagicMock()
-consts_mock.const = MagicMock()
-# Set required constants in consts.const
-consts_mock.const.MINIO_ENDPOINT = "http://localhost:9000"
-consts_mock.const.MINIO_ACCESS_KEY = "test_access_key"
-consts_mock.const.MINIO_SECRET_KEY = "test_secret_key"
-consts_mock.const.MINIO_REGION = "us-east-1"
-consts_mock.const.MINIO_DEFAULT_BUCKET = "test-bucket"
-consts_mock.const.POSTGRES_HOST = "localhost"
-consts_mock.const.POSTGRES_USER = "test_user"
-consts_mock.const.NEXENT_POSTGRES_PASSWORD = "test_password"
-consts_mock.const.POSTGRES_DB = "test_db"
-consts_mock.const.POSTGRES_PORT = 5432
-consts_mock.const.DEFAULT_TENANT_ID = "default_tenant"
-consts_mock.const.LOCAL_MCP_SERVER = "http://localhost:5011"
-consts_mock.const.MODEL_CONFIG_MAPPING = {"llm": "llm_config"}
-consts_mock.const.LANGUAGE = {"ZH": "zh"}
-
-# Add the mocked consts module to sys.modules
-sys.modules['consts'] = consts_mock
-sys.modules['consts.const'] = consts_mock.const
+# Configure required constants via shared bootstrap env
+consts_const.MINIO_ENDPOINT = "http://localhost:9000"
+consts_const.MINIO_ACCESS_KEY = "test_access_key"
+consts_const.MINIO_SECRET_KEY = "test_secret_key"
+consts_const.MINIO_REGION = "us-east-1"
+consts_const.MINIO_DEFAULT_BUCKET = "test-bucket"
+consts_const.POSTGRES_HOST = "localhost"
+consts_const.POSTGRES_USER = "test_user"
+consts_const.NEXENT_POSTGRES_PASSWORD = "test_password"
+consts_const.POSTGRES_DB = "test_db"
+consts_const.POSTGRES_PORT = 5432
+consts_const.DEFAULT_TENANT_ID = "default_tenant"
+consts_const.LOCAL_MCP_SERVER = "http://localhost:5011"
+consts_const.MODEL_CONFIG_MAPPING = {"llm": "llm_config"}
+consts_const.LANGUAGE = {"ZH": "zh"}
 
 # Mock utils module
 utils_mock = MagicMock()

test/backend/app/test_image_app.py

@@ -8,7 +8,7 @@
 if str(TEST_ROOT) not in sys.path:
     sys.path.append(str(TEST_ROOT))
 
-from common.env_test_utils import bootstrap_env
+from test.common.env_test_utils import bootstrap_env
 
 helpers_env = bootstrap_env()
 

test/backend/services/test_image_service.py

@@ -8,7 +8,7 @@
 if str(TEST_ROOT) not in sys.path:
     sys.path.append(str(TEST_ROOT))
 
-from common.env_test_utils import bootstrap_env
+from test.common.env_test_utils import bootstrap_env
 
 helpers_env = bootstrap_env()
 

test/sdk/core/agents/test_nexent_agent.py

@@ -693,6 +693,46 @@ def test_create_local_tool_analyze_image_tool(nexent_agent_instance):
     assert result == mock_analyze_tool_instance
 
 
+def test_create_local_tool_analyze_image_tool(nexent_agent_instance):
+    """Test AnalyzeImageTool creation injects observer and metadata."""
+    mock_analyze_tool_class = MagicMock()
+    mock_analyze_tool_instance = MagicMock()
+    mock_analyze_tool_class.return_value = mock_analyze_tool_instance
+
+    tool_config = ToolConfig(
+        class_name="AnalyzeImageTool",
+        name="analyze_image",
+        description="desc",
+        inputs="{}",
+        output_type="string",
+        params={"prompt": "describe this"},
+        source="local",
+        metadata={
+            "vlm_model": "vlm_model_obj",
+            "storage_client": "storage_client_obj",
+        },
+    )
+
+    original_value = nexent_agent.__dict__.get("AnalyzeImageTool")
+    nexent_agent.__dict__["AnalyzeImageTool"] = mock_analyze_tool_class
+
+    try:
+        result = nexent_agent_instance.create_local_tool(tool_config)
+    finally:
+        if original_value is not None:
+            nexent_agent.__dict__["AnalyzeImageTool"] = original_value
+        elif "AnalyzeImageTool" in nexent_agent.__dict__:
+            del nexent_agent.__dict__["AnalyzeImageTool"]
+
+    mock_analyze_tool_class.assert_called_once_with(
+        observer=nexent_agent_instance.observer,
+        vlm_model="vlm_model_obj",
+        storage_client="storage_client_obj",
+        prompt="describe this",
+    )
+    assert result == mock_analyze_tool_instance
+
+
 def test_create_local_tool_with_observer_attribute(nexent_agent_instance):
     """Test create_local_tool sets observer attribute on tool if it exists."""
     mock_tool_class = MagicMock()

test/sdk/core/tools/test_analyze_image_tool.py

@@ -119,3 +119,4 @@ def test_forward_impl_wraps_model_errors(
             tool._forward_impl([b"img"], "question")
 
         mock_vlm_model.analyze_image.assert_called_once()
+