diff --git a/tests/integration/agents/test_agent_metrics_integration.py b/tests/integration/agents/test_agent_metrics_integration.py deleted file mode 100644 index 7c146d367..000000000 --- a/tests/integration/agents/test_agent_metrics_integration.py +++ /dev/null @@ -1,59 +0,0 @@ -# Copyright (c) Meta Platforms, Inc. and affiliates. -# All rights reserved. -# -# This source code is licensed under the terms described in the LICENSE file in -# the root directory of this source tree. - -import asyncio -from unittest.mock import AsyncMock, Mock, patch - -from llama_stack.providers.inline.agents.meta_reference.agent_instance import ChatAgent - - -class TestAgentMetricsIntegration: - """Smoke test for agent metrics integration""" - - async def test_agent_metrics_methods_exist_and_work(self): - """Test that metrics methods exist and can be called without errors""" - # Create a minimal agent instance with mocked dependencies - telemetry_api = AsyncMock() - telemetry_api.logged_events = [] - - async def mock_log_event(event): - telemetry_api.logged_events.append(event) - - telemetry_api.log_event = mock_log_event - - agent = ChatAgent( - agent_id="test-agent", - agent_config=Mock(), - inference_api=Mock(), - safety_api=Mock(), - tool_runtime_api=Mock(), - tool_groups_api=Mock(), - vector_io_api=Mock(), - telemetry_api=telemetry_api, - persistence_store=Mock(), - created_at="2025-01-01T00:00:00Z", - policy=[], - ) - - with patch("llama_stack.providers.inline.agents.meta_reference.agent_instance.get_current_span") as mock_span: - mock_span.return_value = Mock(get_span_context=Mock(return_value=Mock(trace_id=123, span_id=456))) - - # Test all metrics methods work - agent._track_step() - agent._track_workflow("completed", 2.5) - agent._track_tool("web_search") - - # Wait for async operations - await asyncio.sleep(0.01) - - # Basic verification that telemetry was called - assert len(telemetry_api.logged_events) >= 3 - - # Verify we can call the methods without exceptions - agent._track_tool("knowledge_search") # Test tool mapping - await asyncio.sleep(0.01) - - assert len(telemetry_api.logged_events) >= 4