phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Make a couple properties optional (#963)

Browse source
This commit is contained in:
Ashwin Bharambe 2025-02-04 16:20:24 -08:00 • committed by GitHub
parent 0cbb3e401c
commit 474c4bdd7a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 4 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/apis/agents/agents.py
View file

@ -116,7 +116,7 @@ class Turn(BaseModel):
]
steps: List[Step]
output_message: CompletionMessage
output_attachments: List[Attachment] = Field(default_factory=list)
output_attachments: Optional[List[Attachment]] = Field(default_factory=list)
started_at: datetime
completed_at: Optional[datetime] = None
@ -156,7 +156,7 @@ class AgentConfigCommon(BaseModel):
tool_choice: Optional[ToolChoice] = Field(default=ToolChoice.auto)
tool_prompt_format: Optional[ToolPromptFormat] = Field(default=None)
max_infer_iters: int = 10
max_infer_iters: Optional[int] = 10
@json_schema_type

2
llama_stack/apis/inference/inference.py
View file

@ -142,7 +142,7 @@ class CompletionMessage(BaseModel):
role: Literal["assistant"] = "assistant"
content: InterleavedContent
stop_reason: StopReason
tool_calls: List[ToolCall] = Field(default_factory=list)
tool_calls: Optional[List[ToolCall]] = Field(default_factory=list)
Message = register_schema(