phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-09 13:14:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add tool prompt formats

Browse source
This commit is contained in:
Hardik Shah 2024-08-13 16:00:47 -07:00
parent 0c3e754453
commit 48b78430eb
5 changed files with 32 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

9
llama_toolchain/agentic_system/api/datatypes.py
View file

@ -110,6 +110,12 @@ class Session(BaseModel):
started_at: datetime
@json_schema_type
class ToolPromptFormat(Enum):
json = "json"
function_tag = "function_tag"
@json_schema_type
class AgenticSystemInstanceConfig(BaseModel):
instructions: str
@ -127,6 +133,9 @@ class AgenticSystemInstanceConfig(BaseModel):
# if you completely want to replace the messages prefixed by the system,
# this is debug only
debug_prefix_messages: Optional[List[Message]] = Field(default_factory=list)
tool_prompt_format: Optional[ToolPromptFormat] = Field(
default=ToolPromptFormat.function_tag
)
class AgenticSystemTurnResponseEventType(Enum):