Merge pull request #1377 from Anish29801/feat/dashboard-gui

Feat/dashboard gui

tests/conftest.py

@@ -0,0 +1,4 @@
+import sys
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).parent.parent / "src"))

tests/test_builder.py

@@ -0,0 +1,69 @@
+import pytest
+from llm.core.types import LLMInput, Message, Role, ToolDefinition
+from llm.prompt import PromptBuilder, adapt_messages_for_provider
+from llm.prompt.builder import PromptConfig
+
+
+class TestPromptBuilder:
+    def test_build_without_system(self):
+        messages = [Message(role=Role.USER, content="Hello")]
+        builder = PromptBuilder()
+        result = builder.build(messages)
+
+        assert len(result) == 1
+        assert result[0].role == Role.USER
+
+    def test_build_with_system(self):
+        messages = [
+            Message(role=Role.SYSTEM, content="You are helpful."),
+            Message(role=Role.USER, content="Hello"),
+        ]
+        builder = PromptBuilder()
+        result = builder.build(messages)
+
+        assert len(result) == 2
+        assert result[0].role == Role.SYSTEM
+
+    def test_build_adds_system_from_config(self):
+        messages = [Message(role=Role.USER, content="Hello")]
+        builder = PromptBuilder(system_template="You are a pirate.")
+        result = builder.build(messages)
+
+        assert len(result) == 2
+        assert "pirate" in result[0].content
+
+    def test_build_adds_system_from_config(self):
+        messages = [Message(role=Role.USER, content="Hello")]
+        builder = PromptBuilder(config=PromptConfig(system_template="You are a pirate."))
+        result = builder.build(messages)
+
+        assert len(result) == 2
+        assert "pirate" in result[0].content
+    def test_build_with_tools(self):
+        messages = [Message(role=Role.USER, content="Search for something")]
+        tools = [
+            ToolDefinition(name="search", description="Search the web", parameters={}),
+        ]
+        builder = PromptBuilder(include_tools_in_system=True)
+        result = builder.build(messages, tools)
+
+        assert len(result) == 2
+        assert "search" in result[0].content
+        assert "Available Tools" in result[0].content
+
+
+class TestAdaptMessagesForProvider:
+    def test_adapt_for_claude(self):
+        messages = [Message(role=Role.USER, content="Hello")]
+        result = adapt_messages_for_provider(messages, "claude")
+        assert len(result) == 1
+
+    def test_adapt_for_openai(self):
+        messages = [Message(role=Role.USER, content="Hello")]
+        result = adapt_messages_for_provider(messages, "openai")
+        assert len(result) == 1
+
+    def test_adapt_for_ollama(self):
+        messages = [Message(role=Role.USER, content="Hello")]
+        result = adapt_messages_for_provider(messages, "ollama")
+        assert len(result) == 1

tests/test_executor.py

@@ -0,0 +1,86 @@
+import pytest
+from llm.core.types import ToolCall, ToolDefinition, ToolResult
+from llm.tools import ToolExecutor, ToolRegistry
+
+
+class TestToolRegistry:
+    def test_register_and_get(self):
+        registry = ToolRegistry()
+
+        def dummy_func() -> str:
+            return "result"
+
+        tool_def = ToolDefinition(
+            name="dummy",
+            description="A dummy tool",
+            parameters={"type": "object"},
+        )
+        registry.register(tool_def, dummy_func)
+
+        assert registry.has("dummy") is True
+        assert registry.get("dummy") is dummy_func
+        assert registry.get_definition("dummy") == tool_def
+
+    def test_list_tools(self):
+        registry = ToolRegistry()
+        tool_def = ToolDefinition(name="test", description="Test", parameters={})
+        registry.register(tool_def, lambda: None)
+
+        tools = registry.list_tools()
+        assert len(tools) == 1
+        assert tools[0].name == "test"
+
+
+class TestToolExecutor:
+    def test_execute_success(self):
+        registry = ToolRegistry()
+
+        def search(query: str) -> str:
+            return f"Results for: {query}"
+
+        registry.register(
+            ToolDefinition(
+                name="search",
+                description="Search",
+                parameters={"type": "object", "properties": {"query": {"type": "string"}}},
+            ),
+            search,
+        )
+
+        executor = ToolExecutor(registry)
+        result = executor.execute(ToolCall(id="1", name="search", arguments={"query": "test"}))
+
+        assert result.tool_call_id == "1"
+        assert result.content == "Results for: test"
+        assert result.is_error is False
+
+    def test_execute_unknown_tool(self):
+        registry = ToolRegistry()
+        executor = ToolExecutor(registry)
+
+        result = executor.execute(ToolCall(id="1", name="unknown", arguments={}))
+
+        assert result.is_error is True
+        assert "not found" in result.content
+
+    def test_execute_all(self):
+        registry = ToolRegistry()
+
+        def tool1() -> str:
+            return "result1"
+
+        def tool2() -> str:
+            return "result2"
+
+        registry.register(ToolDefinition(name="t1", description="", parameters={}), tool1)
+        registry.register(ToolDefinition(name="t2", description="", parameters={}), tool2)
+
+        executor = ToolExecutor(registry)
+        results = executor.execute_all([
+            ToolCall(id="1", name="t1", arguments={}),
+            ToolCall(id="2", name="t2", arguments={}),
+        ])
+
+        assert len(results) == 2
+        assert results[0].content == "result1"
+        assert results[1].content == "result2"

tests/test_resolver.py

@@ -0,0 +1,28 @@
+import pytest
+from llm.core.types import ProviderType
+from llm.providers import ClaudeProvider, OpenAIProvider, OllamaProvider, get_provider
+
+
+class TestGetProvider:
+    def test_get_claude_provider(self):
+        provider = get_provider("claude")
+        assert isinstance(provider, ClaudeProvider)
+        assert provider.provider_type == ProviderType.CLAUDE
+
+    def test_get_openai_provider(self):
+        provider = get_provider("openai")
+        assert isinstance(provider, OpenAIProvider)
+        assert provider.provider_type == ProviderType.OPENAI
+
+    def test_get_ollama_provider(self):
+        provider = get_provider("ollama")
+        assert isinstance(provider, OllamaProvider)
+        assert provider.provider_type == ProviderType.OLLAMA
+
+    def test_get_provider_by_enum(self):
+        provider = get_provider(ProviderType.CLAUDE)
+        assert isinstance(provider, ClaudeProvider)
+
+    def test_invalid_provider_raises(self):
+        with pytest.raises(ValueError, match="Unknown provider type"):
+            get_provider("invalid")

tests/test_types.py

@@ -0,0 +1,117 @@
+import pytest
+from llm.core.types import (
+    LLMInput,
+    LLMOutput,
+    Message,
+    ModelInfo,
+    ProviderType,
+    Role,
+    ToolCall,
+    ToolDefinition,
+    ToolResult,
+)
+
+
+class TestRole:
+    def test_role_values(self):
+        assert Role.SYSTEM.value == "system"
+        assert Role.USER.value == "user"
+        assert Role.ASSISTANT.value == "assistant"
+        assert Role.TOOL.value == "tool"
+
+
+class TestProviderType:
+    def test_provider_values(self):
+        assert ProviderType.CLAUDE.value == "claude"
+        assert ProviderType.OPENAI.value == "openai"
+        assert ProviderType.OLLAMA.value == "ollama"
+
+
+class TestMessage:
+    def test_create_message(self):
+        msg = Message(role=Role.USER, content="Hello")
+        assert msg.role == Role.USER
+        assert msg.content == "Hello"
+        assert msg.name is None
+        assert msg.tool_call_id is None
+
+    def test_message_to_dict(self):
+        msg = Message(role=Role.USER, content="Hello", name="test")
+        result = msg.to_dict()
+        assert result["role"] == "user"
+        assert result["content"] == "Hello"
+        assert result["name"] == "test"
+
+
+class TestToolDefinition:
+    def test_create_tool(self):
+        tool = ToolDefinition(
+            name="search",
+            description="Search the web",
+            parameters={"type": "object", "properties": {}},
+        )
+        assert tool.name == "search"
+        assert tool.strict is True
+
+    def test_tool_to_dict(self):
+        tool = ToolDefinition(
+            name="search",
+            description="Search",
+            parameters={"type": "object"},
+        )
+        result = tool.to_dict()
+        assert result["name"] == "search"
+        assert result["strict"] is True
+
+
+class TestToolCall:
+    def test_create_tool_call(self):
+        tc = ToolCall(id="1", name="search", arguments={"query": "test"})
+        assert tc.id == "1"
+        assert tc.name == "search"
+        assert tc.arguments == {"query": "test"}
+
+
+class TestToolResult:
+    def test_create_tool_result(self):
+        result = ToolResult(tool_call_id="1", content="result")
+        assert result.tool_call_id == "1"
+        assert result.is_error is False
+
+
+class TestLLMInput:
+    def test_create_input(self):
+        messages = [Message(role=Role.USER, content="Hello")]
+        input_obj = LLMInput(messages=messages, temperature=0.7)
+        assert len(input_obj.messages) == 1
+        assert input_obj.temperature == 0.7
+
+    def test_input_to_dict(self):
+        messages = [Message(role=Role.USER, content="Hello")]
+        input_obj = LLMInput(messages=messages)
+        result = input_obj.to_dict()
+        assert "messages" in result
+        assert result["temperature"] == 1.0
+
+
+class TestLLMOutput:
+    def test_create_output(self):
+        output = LLMOutput(content="Hello!")
+        assert output.content == "Hello!"
+        assert output.has_tool_calls is False
+
+    def test_output_with_tool_calls(self):
+        tc = ToolCall(id="1", name="search", arguments={})
+        output = LLMOutput(content="", tool_calls=[tc])
+        assert output.has_tool_calls is True
+
+
+class TestModelInfo:
+    def test_create_model_info(self):
+        info = ModelInfo(
+            name="gpt-4",
+            provider=ProviderType.OPENAI,
+        )
+        assert info.name == "gpt-4"
+        assert info.supports_tools is True
+        assert info.supports_vision is False