mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-08-15 06:00:48 +00:00
remove integration test
This commit is contained in:
parent
d8177a2e4e
commit
9f14b67f5c
1 changed files with 0 additions and 59 deletions
|
@ -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
|
|
Loading…
Add table
Add a link
Reference in a new issue