tom_swe_agent_quickstart_for_repo_bug_fixing.py

1from tom_swe import TOM
2
3# Initialize the TOM agent
4# You will need an OPENAI_API_KEY set in your environment variables
5tom = TOM()
6
7# Define the task and the path to the local repository you want to work on
8repo_path = "./my-python-project"
9instruction = "Fix the bug in the calculation logic in models.py where tax is calculated incorrectly."
10
11# Run TOM to solve the issue
12# TOM will explore the codebase, identify the bug, and propose/apply a fix
13result = tom.solve(
14    repo_path=repo_path,
15    instruction=instruction
16)
17
18print(f"Task Status: {result['status']}")
19print(f"Summary of changes: {result['summary']}")