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

test agents imports

Browse source
This commit is contained in:
Xi Yan 2024-12-26 16:32:23 -08:00
parent 9bdb7236b2
commit 0a0c01fbc2
1 changed files with 22 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -5,11 +5,31 @@
# the root directory of this source tree. # the root directory of this source tree.
import os import os
from typing import Dict, List
import pytest import pytest
from llama_models.llama3.api.datatypes import BuiltinTool
from llama_stack.apis.agents import * # noqa: F403 from llama_stack.apis.agents import (
from llama_stack.providers.datatypes import * # noqa: F403 AgentConfig,
AgentTool,
AgentTurnResponseEventType,
AgentTurnResponseStepCompletePayload,
AgentTurnResponseStreamChunk,
AgentTurnResponseTurnCompletePayload,
Attachment,
MemoryToolDefinition,
SearchEngineType,
SearchToolDefinition,
ShieldCallStep,
StepType,
ToolChoice,
ToolExecutionStep,
Turn,
)
from llama_stack.apis.inference import CompletionMessage, SamplingParams, UserMessage
from llama_stack.apis.safety import ViolationLevel
from llama_stack.providers.datatypes import Api
# How to run this test: # How to run this test:
# #