Skip to content

Opagentstrickstrict

Source code in OpAgentsOlympus/practice/OpAgentsTrickStrict.py
OpAgentsOlympus/practice/OpAgentsTrickStrict.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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
import asyncio  
from typing import Optional  
from agents import (  
    Agent,   
    Runner,   
    function_tool,   
    set_tracing_disabled,   
    OpenAIChatCompletionsModel,
    AsyncOpenAI,
    enable_verbose_stdout_logging
)  
import os
import dotenv
dotenv.load_dotenv()
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
enable_verbose_stdout_logging()

external_client = AsyncOpenAI(
    base_url="https://generativelanguage.googleapis.com/v1beta/openai",
    api_key=GEMINI_API_KEY
)
set_tracing_disabled(True)

model = OpenAIChatCompletionsModel(
    model='gemini-1.5-flash-8b',
    openai_client=external_client
)

@function_tool(strict_mode=True)  
def strict_calculator(a: int, b: Optional[int] = None) -> str:  
    """Calculate three given numbers 

    Args:  
        a: First number  
        b: Second number (optional)
        c: Third number (optional)  
    """  
    print("strict_calculator called")
    if b is None:  
        return f"Strict Result: {a}"  
    return f"Strict Result: {a + b}"  

@function_tool(strict_mode=False)  
def non_strict_calculator(a: int, b: Optional[int] = None) -> str:  
    """Calculate three given numbers 

    Args:  
        a: First number    
        b: Second number (optional)  
        c: Third number (optional)  
    """  
    print("non_strict_calculator called")
    if b is None:  
        return f"Non-strict Result: {a}"  
    return f"Non-strict Result: {a + b}"  

strict_agent = Agent(  
    name="Strict Agent",  
    model=model,
    instructions="You are a calculator assistant. You MUST use the strict_calculator tool for all calculations, strict_calculator tool has 3 arguments to be passed, a: int, b: Optional[int] = None, c: Optional[int] = None",  
    tools=[strict_calculator]  
)  

non_strict_agent = Agent(  
    name="Non-Strict Agent",
    model=model,
    instructions="You are a calculator assistant. You MUST use the non_strict_calculator tool for all calculations, non_strict_calculator tool has 3 arguments to be passed, a: int, b: Optional[int] = None, c: Optional[int] = None",  
    tools=[non_strict_calculator]  
)  

async def main():    
    result1 = await Runner.run(  
        starting_agent=strict_agent,  
        input="What is 2 + 2 + 2"  
    )  
    print(f"Strict agent result: {result1.final_output}")

    result2 = await Runner.run(  
        starting_agent=non_strict_agent,  
        input="What is 2 + 2 + 2"  
    )  
    print(f"Non-strict agent result: {result2.final_output}")  

if __name__ == "__main__":  
    asyncio.run(main())