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

Add a test for CLI, but not fully done so disabled

Browse source
This commit is contained in:
Ashwin Bharambe 2024-09-19 13:27:07 -07:00
parent 8b3ffa33de
commit 132f9429b1
3 changed files with 124 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

34
llama_stack/providers/impls/meta_reference/agents/tests/test_chat_agent.py
View file

@ -49,19 +49,19 @@ class MockInferenceAPI:
delta="AI is a fascinating field...",
)
)
yield ChatCompletionResponseStreamChunk(
event=ChatCompletionResponseEvent(
event_type="progress",
delta=ToolCallDelta(
content=ToolCall(
call_id="123",
tool_name=BuiltinTool.brave_search.value,
arguments={"query": "AI history"},
),
parse_status="success",
),
)
)
# yield ChatCompletionResponseStreamChunk(
# event=ChatCompletionResponseEvent(
# event_type="progress",
# delta=ToolCallDelta(
# content=ToolCall(
# call_id="123",
# tool_name=BuiltinTool.brave_search.value,
# arguments={"query": "AI history"},
# ),
# parse_status="success",
# ),
# )
# )
yield ChatCompletionResponseStreamChunk(
event=ChatCompletionResponseEvent(
event_type="complete",
@ -179,10 +179,10 @@ async def chat_agent(mock_inference_api, mock_safety_api, mock_memory_api):
instructions="You are a helpful assistant.",
sampling_params=SamplingParams(),
tools=[
SearchToolDefinition(
name="brave_search",
api_key="test_key",
),
# SearchToolDefinition(
# name="brave_search",
# api_key="test_key",
# ),
],
tool_choice=ToolChoice.auto,
input_shields=[],