phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-28 02:11:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add function tools to openai responses

Browse source
This commit is contained in:
Ashwin Bharambe 2025-04-30 13:06:33 -07:00
parent 5a2bfd6ad5
commit 248a4a3f72
5 changed files with 220 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

26
llama_stack/apis/agents/openai_responses.py
View file

@ -4,7 +4,7 @@
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from typing import List, Literal, Optional, Union
from typing import Any, Dict, List, Literal, Optional, Union
from pydantic import BaseModel, Field
from typing_extensions import Annotated
@ -133,8 +133,30 @@ class OpenAIResponseInputToolWebSearch(BaseModel):
# TODO: add user_location
@json_schema_type
class OpenAIResponseInputToolFunction(BaseModel):
type: Literal["function"] = "function"
name: str
description: Optional[str] = None
parameters: Optional[Dict[str, Any]]
strict: Optional[bool]
class FileSearchRankingOptions(BaseModel):
ranker: Optional[str] = None
score_threshold: Optional[float] = Field(default=0.0, ge=0.0, le=1.0)
@json_schema_type
class OpenAIResponseInputToolFileSearch(BaseModel):
type: Literal["file_search"] = "file_search"
vector_store_id: List[str]
ranking_options: Optional[FileSearchRankingOptions] = None
# TODO: add filters
OpenAIResponseInputTool = Annotated[
Union[OpenAIResponseInputToolWebSearch,],
Union[OpenAIResponseInputToolWebSearch, OpenAIResponseInputToolFileSearch, OpenAIResponseInputToolFunction],
Field(discriminator="type"),
]
register_schema(OpenAIResponseInputTool, name="OpenAIResponseInputTool")