phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-01 02:34:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix tests

Browse source
This commit is contained in:
Hardik Shah 2025-01-14 17:38:10 -08:00 • committed by Ashwin Bharambe
parent d9d827ff25
commit 0edd3ce78b
3 changed files with 7 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -7,6 +7,7 @@
import os
import pytest
from llama_models.datatypes import SamplingParams, TopPSamplingStrategy
from llama_models.llama3.api.datatypes import BuiltinTool
from llama_stack.apis.agents import (
@ -22,12 +23,8 @@ from llama_stack.apis.agents import (
ToolExecutionStep,
Turn,
)
from llama_stack.apis.inference import (
CompletionMessage,
SamplingParams,
TopPSamplingStrategy,
UserMessage,
)
from llama_stack.apis.inference import CompletionMessage, UserMessage
from llama_stack.apis.safety import ViolationLevel
from llama_stack.providers.datatypes import Api