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