Team Tool Call Metrics

1from kern.agent import Agent
2from kern.models.openai import OpenAIChat
3from kern.team import Team
4from kern.tools.yfinance import YFinanceTools
5from rich.pretty import pprint
6
7stock_searcher = Agent(
8    name="Stock Searcher",
9    model=OpenAIChat(id="gpt-4o-mini"),
10    role="Searches for stock information.",
11    tools=[YFinanceTools()],
12)
13
14team = Team(
15    name="Stock Research Team",
16    model=OpenAIChat(id="gpt-4o-mini"),
17    members=[stock_searcher],
18    markdown=True,
19    show_members_responses=True,
20    store_member_responses=True,
21)
22
23if __name__ == "__main__":
24    run_output = team.run("What is the stock price of NVDA?")
25
26    # Team leader run metrics
27    print("=" * 50)
28    print("TEAM LEADER RUN METRICS")
29    print("=" * 50)
30    pprint(run_output.metrics)
31
32    # Member metrics and tool call timing
33    print("=" * 50)
34    print("MEMBER METRICS AND TOOL CALLS")
35    print("=" * 50)
36    if run_output.member_responses:
37        for member_response in run_output.member_responses:
38            print(f"\nMember: {member_response.agent_name}")
39            print("-" * 40)
40            pprint(member_response.metrics)
41
42            if member_response.tools:
43                print(f"\nTool calls ({len(member_response.tools)}):")
44                for tool_call in member_response.tools:
45                    print(f"  Tool: {tool_call.tool_name}")
46                    if tool_call.metrics:
47                        pprint(tool_call.metrics)