Skip to content

Tool Choice With Tool Name Set

Source code in OpAgentsOlympus/practice/tool_choice_with_tool_name_set.py
OpAgentsOlympus/practice/tool_choice_with_tool_name_set.py
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from config import config
from agents import Agent, Runner, function_tool, ModelSettings, RunContextWrapper


@function_tool
def say_hello() -> int:
    return 'Hello, Guys!'

@function_tool
def say_bye(ctx: RunContextWrapper) -> int:
    return 'Bye, Guys!'

def main():
    assistant = Agent(
        name="assistant",
        instructions='123',
        model_settings=ModelSettings(
            tool_choice='say_bye'
        ),
        tools=[say_bye, say_hello]
    )
    result = Runner.run_sync(
        assistant, input='hello', run_config=config
    )
    print(result.final_output)

if __name__ == "__main__":
    main()