Culture Metrics

1"""Culture model token usage tracked under the "culture_model" detail key."""
2
3from kern.agent import Agent
4from kern.culture.manager import CultureManager
5from kern.db.postgres import PostgresDb
6from kern.models.openai import OpenAIChat
7from rich.pretty import pprint
8
9db = PostgresDb(db_url="postgresql+psycopg://ai:ai@localhost:5532/ai")
10
11agent = Agent(
12    model=OpenAIChat(id="gpt-4o-mini"),
13    culture_manager=CultureManager(model=OpenAIChat(id="gpt-4o-mini"), db=db),
14    update_cultural_knowledge=True,
15    db=db,
16)
17
18if __name__ == "__main__":
19    run_response = agent.run(
20        "Our team always does code reviews before merging. We pair program on complex features."
21    )
22
23    print("=" * 50)
24    print("RUN METRICS")
25    print("=" * 50)
26    pprint(run_response.metrics)
27
28    print("=" * 50)
29    print("MODEL DETAILS")
30    print("=" * 50)
31    if run_response.metrics and run_response.metrics.details:
32        for model_type, model_metrics_list in run_response.metrics.details.items():
33            print(f"\n{model_type}:")
34            for model_metric in model_metrics_list:
35                pprint(model_metric)

1# Clone and setup repo
2git clone https://github.com/kern-ai/kern.git
3cd kern/cookbook/02_agents/14_advanced
4
5# Create and activate virtual environment
6./scripts/demo_setup.sh
7source .venvs/demo/bin/activate
8
9python culture_metrics.py