Python: Add DevUI to AgentFramework (#781)

* add initial backend service code for devui * add tests * add frontendcode * ui updates * update readme * ui updates and tweaks * update ui bundle * improve ui, add react flow base * add react flow ui, fix background * update ui, fix introspection bug * update readme * update ui build * add support for multimodal input - both backend and frontend * update ui build * refactor as main framework package * backend and tests refactor * ui build update * ui build update and refactor * update pyproject.toml, update uv.lock * update ui build * ui update to fit oai responses types * add backend updat and readme update * mypy and other fixes * add intial dev guide * update ui and fix workflow bug * update ui build, add thread support * type fixes * update workflow view * update uv.lock * fix workflow iport errors * lint and other fixes * mypy fixes * minor update * update ui build * refactor to use oai dependencies directly, update examples to samples, improve typing * readme update * update ui and ui build * fix workflow pyright error * update ui, fix issues with run workflow placement, miniamp menu, etc * make samples integrate serve --------- Co-authored-by: Chris <66376200+crickman@users.noreply.github.com> Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com>
2026-06-16 21:04:09 +08:00 · 2025-09-22 16:30:08 -07:00
parent adb6dcd2af
commit 1ef24d3e91
98 changed files with 18045 additions and 4 deletions
@@ -0,0 +1,135 @@
+# Copyright (c) Microsoft. All rights reserved.
+
+"""Focused tests for server functionality."""
+
+import asyncio
+import tempfile
+from pathlib import Path
+
+import pytest
+
+from agent_framework_devui import DevServer
+from agent_framework_devui.models._openai_custom import AgentFrameworkExtraBody, AgentFrameworkRequest
+
+
+@pytest.fixture
+def test_entities_dir():
+    """Use the samples directory which has proper entity structure."""
+    current_dir = Path(__file__).parent
+    samples_dir = current_dir.parent / "samples"
+    return str(samples_dir.resolve())
+
+
+@pytest.mark.asyncio
+async def test_server_health_endpoint(test_entities_dir):
+    """Test /health endpoint."""
+    server = DevServer(entities_dir=test_entities_dir)
+    executor = await server._ensure_executor()
+
+    # Test entity count
+    entities = await executor.discover_entities()
+    assert len(entities) > 0
+    # Framework name is now hardcoded since we simplified to single framework
+
+
+@pytest.mark.asyncio
+async def test_server_entities_endpoint(test_entities_dir):
+    """Test /v1/entities endpoint."""
+    server = DevServer(entities_dir=test_entities_dir)
+    executor = await server._ensure_executor()
+
+    entities = await executor.discover_entities()
+    assert len(entities) >= 1
+    # Should find at least the weather agent
+    agent_entities = [e for e in entities if e.type == "agent"]
+    assert len(agent_entities) >= 1
+    agent_names = [e.name for e in agent_entities]
+    assert "WeatherAgent" in agent_names
+
+
+@pytest.mark.asyncio
+async def test_server_execution_sync(test_entities_dir):
+    """Test sync execution endpoint."""
+    server = DevServer(entities_dir=test_entities_dir)
+    executor = await server._ensure_executor()
+
+    entities = await executor.discover_entities()
+    agent_id = entities[0].id
+
+    request = AgentFrameworkRequest(
+        model="agent-framework",
+        input="San Francisco",
+        stream=False,
+        extra_body=AgentFrameworkExtraBody(entity_id=agent_id),
+    )
+
+    response = await executor.execute_sync(request)
+    assert response.model == "agent-framework"
+    assert len(response.output) > 0
+
+
+@pytest.mark.asyncio
+async def test_server_execution_streaming(test_entities_dir):
+    """Test streaming execution endpoint."""
+    server = DevServer(entities_dir=test_entities_dir)
+    executor = await server._ensure_executor()
+
+    entities = await executor.discover_entities()
+    agent_id = entities[0].id
+
+    request = AgentFrameworkRequest(
+        model="agent-framework", input="New York", stream=True, extra_body=AgentFrameworkExtraBody(entity_id=agent_id)
+    )
+
+    event_count = 0
+    async for _event in executor.execute_streaming(request):
+        event_count += 1
+        if event_count > 5:  # Limit for testing
+            break
+
+    assert event_count > 0
+
+
+def test_configuration():
+    """Test basic configuration."""
+    server = DevServer(entities_dir="test", port=9000, host="localhost")
+    assert server.port == 9000
+    assert server.host == "localhost"
+    assert server.entities_dir == "test"
+    assert server.cors_origins == ["*"]
+    assert server.ui_enabled
+
+
+if __name__ == "__main__":
+    # Simple test runner
+    async def run_tests():
+        with tempfile.TemporaryDirectory() as temp_dir:
+            temp_path = Path(temp_dir)
+
+            # Create test agent
+            agent_file = temp_path / "weather_agent.py"
+            agent_file.write_text("""
+class WeatherAgent:
+    name = "Weather Agent"
+    description = "Gets weather information"
+
+    def run_stream(self, input_str):
+        return f"Weather in {input_str} is sunny"
+""")
+
+            server = DevServer(entities_dir=str(temp_path))
+            executor = await server._ensure_executor()
+
+            entities = await executor.discover_entities()
+
+            if entities:
+                request = AgentFrameworkRequest(
+                    model="agent-framework",
+                    input="test location",
+                    stream=False,
+                    extra_body=AgentFrameworkExtraBody(entity_id=entities[0].id),
+                )
+
+                await executor.execute_sync(request)
+
+    asyncio.run(run_tests())