main.py

import os
from orchestrator import SelfImprovingOrchestrator
from regression_runner import RegressionRunner, TestCase

os.environ["OPENAI_API_KEY"] = "your-key-here"


def seed_test_cases():
    """Pre-load test cases for tools we expect the agent to create."""
    runner = RegressionRunner()

    runner.store_test_case("fetch_exchange_rate", TestCase(
        input_args={"from_currency": "USD", "to_currency": "EUR"},
        expected_type="float",
        max_latency_ms=3000,
        description="USD to EUR returns a float",
    ))
    runner.store_test_case("fetch_exchange_rate", TestCase(
        input_args={"from_currency": "USD", "to_currency": "ILS"},
        expected_type="float",
        max_latency_ms=3000,
        description="USD to ILS returns a float",
    ))
    runner.store_test_case("convert_currency", TestCase(
        input_args={"amount": 100.0, "rate": 3.5},
        expected_type="float",
        max_latency_ms=100,
        description="Simple multiplication returns float",
    ))


def main():
    seed_test_cases()
    agent = SelfImprovingOrchestrator()

    tasks = [
        "What is the current USD to ILS exchange rate? Convert 5000 USD.",
        "What is the current EUR to GBP rate? Convert 2000 EUR.",
        "Convert 10000 JPY to USD using the latest rate.",
    ]

    for task in tasks:
        result = agent.run(task)
        print(f"\n Result: {result}\n")
        print("-" * 60)

    # Print final performance report
    print("\n" + "=" * 60)
    print("PERFORMANCE REPORT")
    print("=" * 60)
    from performance_memory import memory
    for tool_name in ["fetch_exchange_rate", "convert_currency"]:
        summary = memory.summary(tool_name)
        if summary["invocations"] > 0:
            print(f"\n{tool_name}:")
            print(f"  Invocations: {summary['invocations']}")
            print(f"  Success rate: {summary['success_rate']:.1%}")
            print(f"  Avg latency: {summary['avg_latency_ms']:.0f}ms")
            print(f"  Version: v{summary['current_version']}")


if __name__ == "__main__":
    main()