agent-dev-cli 0.0.1b260128__tar.gz → 0.0.1b260205__tar.gz

{agent_dev_cli-0.0.1b260128 → agent_dev_cli-0.0.1b260205}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-dev-cli
-Version: 0.0.1b260128
+Version: 0.0.1b260205
 Summary: AI Toolkit - CLI and SDK for agent debugging and workflow visualization
 Keywords: ai,agent,toolkit,debugging,workflow,visualization
 Author-email: Microsoft <aitkfeedback@microsoft.com>
@@ -28,8 +28,8 @@ Requires-Dist: starlette>=0.49.3
 Requires-Dist: openai>=1.109.1
 Requires-Dist: agent-framework-azure-ai>=1.0.0b251211,<=1.0.0b260107
 Requires-Dist: agent-framework-core>=1.0.0b251211,<=1.0.0b260107
-Requires-Dist: azure-ai-agentserver-agentframework==1.0.0b7
-Requires-Dist: azure-ai-agentserver-core==1.0.0b7
+Requires-Dist: azure-ai-agentserver-agentframework==1.0.0b10
+Requires-Dist: azure-ai-agentserver-core==1.0.0b10
 Requires-Dist: click>=8.1.0
 Requires-Dist: azure-ai-projects>=2.0.0b2
 Project-URL: Documentation, https://github.com/microsoft/vscode-ai-toolkit#readme
@@ -79,7 +79,7 @@ The CLI automatically:
 If you prefer explicit control, you can integrate agentdev directly:
 
 ```python
-from agentdev import setup_test_tool
+from agentdev import configure_agent_server
 from azure.ai.agentserver.agentframework import from_agent_framework
 
 # Create your agent
@@ -89,7 +89,7 @@ agent = build_agent(chat_client)
 agent_server = from_agent_framework(agent)
 
 # Setup workflow visualization
-setup_test_tool(agent_server)
+configure_agent_server(agent_server)
 
 # Run the server
 await agent_server.run_async()

{agent_dev_cli-0.0.1b260128 → agent_dev_cli-0.0.1b260205}/README.md

@@ -40,7 +40,7 @@ The CLI automatically:
 If you prefer explicit control, you can integrate agentdev directly:
 
 ```python
-from agentdev import setup_test_tool
+from agentdev import configure_agent_server
 from azure.ai.agentserver.agentframework import from_agent_framework
 
 # Create your agent
@@ -50,7 +50,7 @@ agent = build_agent(chat_client)
 agent_server = from_agent_framework(agent)
 
 # Setup workflow visualization
-setup_test_tool(agent_server)
+configure_agent_server(agent_server)
 
 # Run the server
 await agent_server.run_async()

{agent_dev_cli-0.0.1b260128 → agent_dev_cli-0.0.1b260205}/pyproject.toml

@@ -33,8 +33,8 @@ dependencies = [
     "openai>=1.109.1",
     "agent-framework-azure-ai>=1.0.0b251211,<=1.0.0b260107",
     "agent-framework-core>=1.0.0b251211,<=1.0.0b260107",
-    "azure-ai-agentserver-agentframework==1.0.0b7",
-    "azure-ai-agentserver-core==1.0.0b7",
+    "azure-ai-agentserver-agentframework==1.0.0b10",
+    "azure-ai-agentserver-core==1.0.0b10",
     "click>=8.1.0",
     "azure-ai-projects>=2.0.0b2",
 ]

{agent_dev_cli-0.0.1b260128 → agent_dev_cli-0.0.1b260205}/src/agentdev/__init__.py

@@ -12,11 +12,11 @@ This package provides two ways to use agentdev:
    
 2. Programmatic API (Requires code modification):
    
-   from agentdev import setup_test_tool
-   setup_test_tool(agent_server)
+   from agentdev import configure_agent_server
+   configure_agent_server(agent_server)
 """
 
-from .localdebug import setup_test_tool
+from .localdebug import configure_agent_server
 from ._version import __version__
 
-__all__ = ["setup_test_tool", "__version__"]
+__all__ = ["configure_agent_server", "__version__"]

{agent_dev_cli-0.0.1b260128 → agent_dev_cli-0.0.1b260205}/src/agentdev/_hooks.py

@@ -7,7 +7,7 @@ capabilities without requiring user code modifications.
 
 The hook intercepts:
 - `azure.ai.agentserver.agentframework.from_agent_framework()` 
-  to automatically call `setup_test_tool()` on the created server.
+  to automatically call `configure_agent_server()` on the created server.
 """
 
 import sys
@@ -31,7 +31,7 @@ def _create_patched_from_agent_framework(original_func: Callable) -> Callable:
         original_func: The original from_agent_framework function
         
     Returns:
-        Patched function that calls setup_test_tool automatically
+        Patched function that calls configure_agent_server automatically
     """
     @functools.wraps(original_func)
     def patched_from_agent_framework(agent: Any, **kwargs) -> Any:
@@ -48,8 +48,8 @@ def _create_patched_from_agent_framework(original_func: Callable) -> Callable:
         
         # Auto-inject agentdev visualization
         try:
-            from agentdev.localdebug import setup_test_tool
-            setup_test_tool(server)
+            from agentdev.localdebug import configure_agent_server
+            configure_agent_server(server)
             
             if verbose:
                 print(f"agentdev Hook: Successfully injected agentdev instrumentation", file=sys.stderr)

{agent_dev_cli-0.0.1b260128 → agent_dev_cli-0.0.1b260205}/src/agentdev/_version.py

@@ -1,2 +1,2 @@
 """Version information for agentdev package."""
-__version__ = "0.0.1b260128"
+__version__ = "0.0.1b260205"

{agent_dev_cli-0.0.1b260128 → agent_dev_cli-0.0.1b260205}/src/agentdev/backend/event_mapper.py

@@ -321,16 +321,26 @@ class EventMapper():
         
         for content in event.contents:
             if isinstance(content, FunctionCallContent):
-                # Arguments are always streamed in OpenAI. But not always in Agent Framework.
-                # Argument streaming use last call_id to track the function call instance.
-                if ctx.call_id and content.arguments:
-                    results.append({
-                        "type": "response.function_call_arguments.delta",
-                        "output_index": ctx.output_index,
-                        # DevUI set OpenAI's item_id to the value of Agent Framework's call_id, which might be a bug
-                        "item_id": ctx.call_id,
-                        "delta": content.arguments,
-                    })
+                is_same_call = ctx.call_id and content.call_id == ctx.call_id
+                
+                if is_same_call and content.arguments:
+                    delta_args = ""
+                    if isinstance(content.arguments, str):
+                        delta_args = content.arguments
+                    elif content.arguments:
+                        try:
+                            delta_args = json.dumps(serialize_data(content.arguments))
+                        except Exception as e:
+                            logging.warning(f"Failed to serialize arguments: {e}")
+                            delta_args = str(content.arguments)
+                    
+                    if delta_args:
+                        results.append({
+                            "type": "response.function_call_arguments.delta",
+                            "output_index": ctx.output_index,
+                            "item_id": ctx.call_id,
+                            "delta": delta_args,
+                        })
                 else:
                     if content.call_id != ctx.call_id:
                         ctx.call_id = content.call_id
@@ -347,6 +357,24 @@ class EventMapper():
                             "status":"in_progress",
                         },
                     })
+                    if content.arguments:
+                        initial_args = ""
+                        if isinstance(content.arguments, str):
+                            initial_args = content.arguments
+                        elif content.arguments:
+                            try:
+                                initial_args = json.dumps(serialize_data(content.arguments))
+                            except Exception as e:
+                                logging.warning(f"Failed to serialize initial arguments: {e}")
+                                initial_args = str(content.arguments)
+                        
+                        if initial_args:
+                            results.append({
+                                "type": "response.function_call_arguments.delta",
+                                "output_index": ctx.output_index,
+                                "item_id": ctx.call_id,
+                                "delta": initial_args,
+                            })
             elif isinstance(content, FunctionResultContent):
                 results.append({
                     "type":"response.function_result.complete",           

{agent_dev_cli-0.0.1b260128 → agent_dev_cli-0.0.1b260205}/src/agentdev/backend/server.py

@@ -18,6 +18,8 @@ from starlette.websockets import WebSocket, WebSocketDisconnect
 
 from agent_framework import Workflow, ChatMessage, AgentProtocol, Executor
 
+from .._version import __version__
+
 from .event_mapper import EventMapper, MapperContext
 from .structs.request import AgentFrameworkRequest
 from .structs.entity_response import EntityResponse
@@ -81,6 +83,7 @@ class TestToolServer:
         app = Starlette(routes=[
             # Responses API for workflow and agent
             Route("/v1/responses", endpoint=self.responses, methods=["POST"]),
+            Route("/version", endpoint=self.get_version, methods=["GET"]),
             Route("/entities", endpoint=self.list_entities, methods=["GET"]),
             Route("/entities/{entity_id}/info", endpoint=self.get_entity_info, methods=["GET"]),
             Route("/entities/{entity_id}/executor/{executor_id}/location", endpoint=self.get_executor_location, methods=["GET"]),
@@ -90,6 +93,10 @@ class TestToolServer:
         ])
         root_app.mount("/agentdev/", app)
     
+    async def get_version(self, raw_request: Request):
+        """Return the CLI version for compatibility checking."""
+        return JSONResponse({"version": __version__})
+    
     async def list_entities(self, raw_request: Request):
         entities_info = []
         for entity in self._entities:

agent_dev_cli-0.0.1b260205/src/agentdev/localdebug.py

@@ -0,0 +1,100 @@
+"""
+Workflow visualization setup module.
+
+This module provides functionality to set up workflow or agent visualization
+"""
+
+from agent_framework import WorkflowAgent
+import logging
+import time
+from typing import Any, Optional
+
+
+def _get_agent_from_server(agent_server: Any) -> Optional[Any]:
+    """
+    Extract the agent from an agent server instance.
+    
+    Args:
+        agent_server: The agent server instance
+        
+    Returns:
+        The agent instance, or None if not found
+    """
+    if hasattr(agent_server, '_agent') and agent_server._agent is not None:
+        return agent_server._agent
+    
+    if hasattr(agent_server, 'agent') and agent_server.agent is not None:
+        return agent_server.agent
+    
+    if hasattr(agent_server, '_workflow_factory') and agent_server._workflow_factory is not None:
+        try:
+            return agent_server._workflow_factory().as_agent()
+        except Exception as e:
+            logging.debug("Failed to build workflow agent (%s): %s", type(e).__name__, e)
+    
+    if hasattr(agent_server, '_build_agent'):
+        try:
+            return agent_server._build_agent()
+        except Exception as e:
+            logging.debug("Failed to call _build_agent (%s): %s", type(e).__name__, e)
+    
+    return None
+
+
+def configure_agent_server(agent_server):
+    """
+    Set up workflow or agent visualization for an agent server.
+    
+    This function configures a health check endpoint and starts a visualization
+    server for workflow agents on port 8090.
+    
+    Args:
+        agent_server: The agent server instance to set up visualization for.
+                     Should have 'app' (Starlette) attribute. The agent is 
+                     extracted using `_get_agent_from_server()` which supports
+                     both old and new SDK versions.
+    
+    Example:
+        >>> from azure.ai.agentserver.agentframework import from_agent_framework
+        >>> agent_server = from_agent_framework(agent)
+        >>> configure_agent_server(agent_server)
+        >>> await agent_server.run_async()
+    """
+    from starlette.applications import Starlette
+    from starlette.responses import JSONResponse
+    from starlette.routing import Route
+    from threading import Thread
+    from agentdev.backend.server import TestToolServer
+    
+    async def health_check(request):
+        """Health check endpoint handler."""
+        return JSONResponse({"status": "ok"}, status_code=200)
+    
+    app = agent_server.app
+    agent = _get_agent_from_server(agent_server)
+    
+    if agent is None:
+        logging.warning("agentdev: Could not extract agent from server, visualization may be limited")
+
+    app.mount(
+        "/agentdev/health",
+        Starlette(routes=[Route("/", health_check)])
+    )
+
+    entities = []
+    if agent is not None:
+        if isinstance(agent, WorkflowAgent):
+            entities.append(agent.workflow)
+        else:
+            entities.append(agent)
+
+    test_tool_server = TestToolServer(entities)
+    test_tool_server.mount_backend(app)
+    def show_startup_message():
+        time.sleep(2)
+        logging.info("agentdev: Application startup complete")
+    thread = Thread(target=show_startup_message)
+    thread.daemon = True
+    thread.start()
+    
+    logging.debug("Agent server configured: %s", agent)

agent_dev_cli-0.0.1b260128/src/agentdev/localdebug.py

@@ -1,64 +0,0 @@
-"""
-Workflow visualization setup module.
-
-This module provides functionality to set up workflow or agent visualization
-"""
-
-from agent_framework import WorkflowAgent
-import logging
-import time
-
-
-def setup_test_tool(agent_server):
-    """
-    Set up workflow or agent visualization for an agent server.
-    
-    This function configures a health check endpoint and starts a visualization
-    server for workflow agents on port 8090.
-    
-    Args:
-        agent_server: The agent server instance to set up visualization for.
-                     Should have 'app' (Starlette) and 'agent' attributes.
-    
-    Example:
-        >>> from azure.ai.agentserver.agentframework import from_agent_framework
-        >>> agent_server = from_agent_framework(agent)
-        >>> setup_test_tool(agent_server)
-        >>> await agent_server.run_async()
-    """
-    from starlette.applications import Starlette
-    from starlette.responses import JSONResponse
-    from starlette.routing import Route
-    from threading import Thread
-    from agentdev.backend.server import TestToolServer
-    
-    async def health_check(request):
-        """Health check endpoint handler."""
-        return JSONResponse({"status": "ok"}, status_code=200)
-    
-    app = agent_server.app
-    agent = agent_server.agent
-
-    # Mount health check endpoint
-    app.mount(
-        "/agentdev/health",
-        Starlette(routes=[Route("/", health_check)])
-    )
-
-    # Prepare entities for visualization
-    entities = []
-    if isinstance(agent, WorkflowAgent):
-        entities.append(agent.workflow)
-    else:
-        entities.append(agent)
-
-    test_tool_server = TestToolServer(entities)
-    test_tool_server.mount_backend(app)
-    def show_startup_message():
-        time.sleep(2)
-        logging.info("agentdev: Application startup complete")
-    thread = Thread(target=show_startup_message)
-    thread.daemon = True
-    thread.start()
-    
-    logging.debug("Agent server configured: %s", agent)