Skip to content

Commit 6c538fe

Browse files
committed
support system mesage behavior
# What does this PR do? Updates AgentConfig to support llamastack/llama-stack#938 - [ ] Addresses issue (#issue) ## Test Plan Tested in llamastack/llama-stack#938 ## Sources Please link relevant resources if necessary. ## Before submitting - [ ] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case). - [ ] Ran pre-commit to handle lint / formatting issues. - [ ] Read the [contributor guideline](https://github.com/meta-llama/llama-stack/blob/main/CONTRIBUTING.md), Pull Request section? - [ ] Updated relevant documentation. - [ ] Wrote necessary unit or integration tests.
1 parent 51fd947 commit 6c538fe

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

src/llama_stack_client/types/shared_params/agent_config.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,11 @@ class ToolgroupUnionMember1(TypedDict, total=False):
2020

2121
Toolgroup: TypeAlias = Union[str, ToolgroupUnionMember1]
2222

23+
class ToolConfig(TypedDict, total=False):
24+
tool_choice: Literal["auto", "required"]
25+
tool_prompt_format: Literal["json", "function_tag", "python_list"]
26+
system_message_behavior: Literal["append", "replace"]
27+
2328

2429
class AgentConfig(TypedDict, total=False):
2530
enable_session_persistence: Required[bool]
@@ -40,8 +45,12 @@ class AgentConfig(TypedDict, total=False):
4045

4146
sampling_params: SamplingParams
4247

48+
# DEPRECATED: use tool_config instead
4349
tool_choice: Literal["auto", "required"]
4450

51+
# DEPRECATED: use tool_config instead
4552
tool_prompt_format: Literal["json", "function_tag", "python_list"]
4653

4754
toolgroups: List[Toolgroup]
55+
56+
tool_config: ToolConfig

0 commit comments

Comments
 (0)