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

remove imports 1/n

Browse source
This commit is contained in:
Xi Yan 2024-12-26 15:19:06 -08:00
parent 28ce511986
commit da26d22f90
8 changed files with 43 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

12
llama_stack/apis/batch_inference/batch_inference.py
View file

@ -10,8 +10,16 @@ from llama_models.schema_utils import json_schema_type, webmethod
from pydantic import BaseModel, Field
from llama_models.llama3.api.datatypes import * # noqa: F403
from llama_stack.apis.inference import * # noqa: F403
from llama_stack.apis.inference import (
CompletionMessage,
InterleavedContent,
LogProbConfig,
Message,
SamplingParams,
ToolChoice,
ToolDefinition,
ToolPromptFormat,
)
@json_schema_type