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

more test fixes

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-12-30 13:27:43 -08:00
parent 40439509ca
commit c2dd0cdc78
7 changed files with 63 additions and 120 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/tests/agents/test_agents.py
View file

@ -22,6 +22,8 @@ from llama_stack.apis.agents import (
Turn,
)
from llama_stack.apis.inference import CompletionMessage, SamplingParams, UserMessage
from llama_stack.apis.memory import MemoryBankDocument
from llama_stack.apis.memory_banks import VectorMemoryBankParams
from llama_stack.apis.safety import ViolationLevel
from llama_stack.providers.datatypes import Api