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

fix the rag query generator types

Browse source
This commit is contained in:
Dinesh Yeduguru 2025-01-07 21:33:25 -08:00
parent efe3189728
commit 82395ba654
4 changed files with 19 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/client-sdk/agents/test_agents.py
View file

@ -206,7 +206,7 @@ def test_builtin_tool_code_execution(llama_stack_client, agent_config):
def test_code_execution(llama_stack_client):
agent_config = AgentConfig(
model="meta-llama/Llama-3.1-70B-Instruct",
model="meta-llama/Llama-3.1-8B-Instruct",
instructions="You are a helpful assistant",
toolgroups=[
"builtin::code_interpreter",
@ -297,7 +297,7 @@ def test_rag_agent(llama_stack_client, agent_config):
agent_config["toolgroups"].append(
dict(
name="builtin::memory",
args={"memory_bank_id": memory_bank_id},
args={"memory_bank_ids": [memory_bank_id]},
)
)
agent = Agent(llama_stack_client, agent_config)