This commit is contained in:
Edward Ma 2025-01-28 09:14:57 -08:00
parent 97c1406773
commit a31b5da5cc

View file

@ -25,6 +25,7 @@ from pydantic import BaseModel, ConfigDict, Field
from llama_stack.apis.common.content_types import ContentDelta, InterleavedContent, URL from llama_stack.apis.common.content_types import ContentDelta, InterleavedContent, URL
from llama_stack.apis.inference import ( from llama_stack.apis.inference import (
CompletionMessage, CompletionMessage,
ResponseFormat,
SamplingParams, SamplingParams,
ToolCall, ToolCall,
ToolChoice, ToolChoice,
@ -32,7 +33,6 @@ from llama_stack.apis.inference import (
ToolResponse, ToolResponse,
ToolResponseMessage, ToolResponseMessage,
UserMessage, UserMessage,
ResponseFormat
) )
from llama_stack.apis.safety import SafetyViolation from llama_stack.apis.safety import SafetyViolation
from llama_stack.apis.tools import ToolDef from llama_stack.apis.tools import ToolDef