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

fix agent server tests

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-12-26 18:24:27 -08:00
parent 439f52b067
commit 18d9937500
4 changed files with 41 additions and 63 deletions
Download patch file Download diff file Expand all files Collapse all files

50
llama_stack/providers/tests/agents/fixtures.py
View file

@ -64,7 +64,7 @@ def agents_meta_reference() -> ProviderFixture:
@pytest.fixture(scope="session")
def tool_runtime_memory() -> ProviderFixture:
def tool_runtime_memory_and_search() -> ProviderFixture:
return ProviderFixture(
providers=[
Provider(
@ -72,31 +72,19 @@ def tool_runtime_memory() -> ProviderFixture:
provider_type="inline::memory-runtime",
config={},
),
Provider(
provider_id="brave-search",
provider_type="inline::brave-search",
config={
"api_key": os.environ["BRAVE_SEARCH_API_KEY"],
},
),
Provider(
provider_id="tavily-search",
provider_type="inline::tavily-search",
provider_type="remote::tavily-search",
config={
"api_key": os.environ["TAVILY_SEARCH_API_KEY"],
},
),
Provider(
provider_id="code-interpreter",
provider_type="inline::code-interpreter",
config={},
),
],
)
AGENTS_FIXTURES = ["meta_reference", "remote"]
TOOL_RUNTIME_FIXTURES = ["memory"]
TOOL_RUNTIME_FIXTURES = ["memory_and_search"]
@pytest_asyncio.fixture(scope="session")
@ -173,43 +161,25 @@ async def agents_stack(request, inference_model, safety_shield):
name="memory",
description="memory",
parameters=[
ToolParameter(
name="session_id",
description="session id",
parameter_type="string",
required=True,
),
ToolParameter(
name="input_messages",
description="messages",
parameter_type="list",
required=True,
),
ToolParameter(
name="attachments",
description="attachments",
parameter_type="list",
required=False,
),
],
metadata={},
metadata={
"config": {
"memory_bank_configs": [
{"bank_id": "test_bank", "type": "vector"}
]
}
},
)
],
),
provider_id="memory-runtime",
),
ToolGroupInput(
tool_group_id="code_interpreter_group",
tool_group=UserDefinedToolGroupDef(
tools=[
BuiltInToolDef(
built_in_type=BuiltinTool.code_interpreter,
metadata={},
)
],
),
provider_id="code-interpreter",
),
]
test_stack = await construct_stack_for_test(