Function main

demo.py:102–145

()

"""

async def main():
    print("\n" + "="*60)
    print("MyAgent Demo - Function-based Tools")
    print("="*60)
    
    # 1. 初始化
    model = OpenAIModel(
        api_key=os.getenv("OPENAI_API_KEY"),
        model_name="gpt-4o",
        temperature=0.7
    )
    
    agent = MyAgent(
        model=model,
        config={"name": "my_assistant"}
    )
    
    print(f"\n✅ Loaded {len(agent.get_registered_tools())} tools")
    
    # 2. 测试
    tests = [
        "Design a novel organocatalyst for enhancing CO2 conversion in carbon capture processes. The following reference functions are available: 1) Research existing organocatalysts using `query2smiles` to obtain their SMILES structures. 2) Generate structural variants using `modify_mol`. 3) Verify novelty with `patent_check`. Finally, propose a novel organocatalyst with detailed rationale for its design and expected performance. Ensure all steps are documented and justified. ",
        # "What's the weather in Beijing?",
        # "Calculate (10 + 20) * 3",
        # "Get weather for Tokyo and Paris, then calculate the average temperature"
    ]
    
    for i, query in enumerate(tests, 1):
        print(f"\n{'─'*60}")
        print(f"Test {i}: {query}")
        print(f"{'─'*60}")
        
        result = await agent.execute(
            context={"query": query},
            params={"temperature": 0.7}
        )
        
        if result["status"] == "success":
            print(f"\n💬 {result['answer']}")
            print(f"\n🔧 Used {len(result['tool_calls'])} tool(s)")
        else:
            print(f"\n❌ Error: {result['error']}")
    
    print("\n" + "="*60 + "\n")


if __name__ == "__main__":