phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

fix(telemetry): library client does not log span (#1833)

Browse source
This commit is contained in:
ehhuang 2025-03-29 14:55:31 -07:00 • committed by GitHub
parent d8a8a734b5
commit 3a2314dcef
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 2 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/integration/telemetry/test_telemetry.py
View file

@ -7,16 +7,10 @@
import time
from uuid import uuid4
import pytest
from llama_stack_client import Agent
from llama_stack.distribution.library_client import LlamaStackAsLibraryClient
def test_agent_query_spans(llama_stack_client, text_model_id):
if isinstance(llama_stack_client, LlamaStackAsLibraryClient):
pytest.mark.xfail(reason="Need to fix LlamaStackAsLibraryClient to log spans")
agent = Agent(llama_stack_client, model=text_model_id, instructions="You are a helpful assistant")
session_id = agent.create_session(f"test-session-{uuid4()}")
agent.create_turn(
@ -41,7 +35,6 @@ def test_agent_query_spans(llama_stack_client, text_model_id):
],
attributes_to_return=["input", "output"],
):
print(span.attributes)
if span.attributes["output"] != "no shields":
agent_logs.append(span.attributes)