phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 02:53:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: test_registration was borked somehow

Browse source
This commit is contained in:
Ashwin Bharambe 2025-04-12 12:04:01 -07:00
parent 1e5bf6c19d
commit ef3dc143ec
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/integration/tool_runtime/test_registration.py
View file

@ -12,7 +12,6 @@ import httpx
import mcp.types as types import mcp.types as types
import pytest import pytest
import uvicorn import uvicorn
from llama_stack_client.types.shared_params.url import URL
from mcp.server.fastmcp import Context, FastMCP from mcp.server.fastmcp import Context, FastMCP
from mcp.server.sse import SseServerTransport from mcp.server.sse import SseServerTransport
from starlette.applications import Starlette from starlette.applications import Starlette
@ -97,7 +96,7 @@ def test_register_and_unregister_toolgroup(llama_stack_client, mcp_server):
llama_stack_client.toolgroups.register( llama_stack_client.toolgroups.register(
toolgroup_id=test_toolgroup_id, toolgroup_id=test_toolgroup_id,
provider_id=provider_id, provider_id=provider_id,
mcp_endpoint=URL(uri=f"http://localhost:{port}/sse"), mcp_endpoint=dict(uri=f"http://localhost:{port}/sse"),
) )
# Verify registration # Verify registration