This commit is contained in:
Xi Yan 2025-02-04 10:54:53 -08:00
parent b1492ecb4e
commit 4d07460398
2 changed files with 3 additions and 0 deletions

View file

@ -165,6 +165,7 @@ class AgentConfig(AgentConfigCommon):
instructions: str
enable_session_persistence: bool
response_format: Optional[ResponseFormat] = None
output_parser: Optional[OutputParser] = Field(default=OutputParser.auto)
class AgentConfigOverridablePerTurn(AgentConfigCommon):

View file

@ -19,6 +19,7 @@ from typing import (
from llama_models.llama3.api.datatypes import (
BuiltinTool,
ResponseOutputParser,
SamplingParams,
StopReason,
ToolCall,
@ -319,6 +320,7 @@ class ChatCompletionRequest(BaseModel):
response_format: Optional[ResponseFormat] = None
stream: Optional[bool] = False
logprobs: Optional[LogProbConfig] = None
response_output_parser: Optional[ResponseOutputParser] = Field(default=ResponseOutputParser.auto)
@json_schema_type