phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-23 15:43:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor component path

Browse source 
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-07-29 21:20:21 -04:00
parent a29b62f820
commit 19d2555424
116 changed files with 7064 additions and 1265 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/integration/tool_runtime/test_mcp.py
View file

@ -10,7 +10,6 @@ import pytest
from llama_stack_client import Agent
from llama_stack import LlamaStackAsLibraryClient
from llama_stack.apis.models import ModelType
from llama_stack.distribution.datatypes import AuthenticationRequiredError
AUTH_TOKEN = "test-token"
@ -24,7 +23,7 @@ def mcp_server():
yield mcp_server_info
def test_mcp_invocation(llama_stack_client, mcp_server):
def test_mcp_invocation(llama_stack_client, text_model_id, mcp_server):
if not isinstance(llama_stack_client, LlamaStackAsLibraryClient):
pytest.skip("The local MCP server only reliably reachable from library client.")
@ -69,14 +68,10 @@ def test_mcp_invocation(llama_stack_client, mcp_server):
assert content[0].type == "text"
assert content[0].text == "Hello, world!"
models = [
m for m in llama_stack_client.models.list() if m.model_type == ModelType.llm and "guard" not in m.identifier
]
model_id = models[0].identifier
print(f"Using model: {model_id}")
print(f"Using model: {text_model_id}")
agent = Agent(
client=llama_stack_client,
model=model_id,
model=text_model_id,
instructions="You are a helpful assistant.",
tools=[test_toolgroup_id],
)