Skip to content

Add PR analysis with code context investigation #95

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
mohit-sheth wants to merge 1 commit into
base: main
Choose a base branch
from
Draft

Add PR analysis with code context investigation #95

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 16 additions & 2 deletions bugzooka/analysis/pr_analyzer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,10 @@
tool_not_found_error,
)
from bugzooka.core.utils import make_response
from bugzooka.core.constants import PR_ANALYSIS_MAX_TOOL_ITERATIONS
from bugzooka.integrations.inference_client import analyze_with_agentic
from bugzooka.analysis.prompts import PR_PERFORMANCE_ANALYSIS_PROMPT
from bugzooka.analysis.pr_tools import clear_pr_files_cache, create_pr_tools
import bugzooka.integrations.mcp_client as mcp_module


Expand Down Expand Up @@ -137,7 +139,11 @@ async def analyze_pr_with_gemini(text: str) -> dict:

system_prompt = PR_PERFORMANCE_ANALYSIS_PROMPT["system"]
user_prompt = PR_PERFORMANCE_ANALYSIS_PROMPT["user"].format(
org=org, repo=repo, pr_number=pr_number, pr_url=pr_url, version=version
org=org,
repo=repo,
pr_number=pr_number,
pr_url=pr_url,
version=version,
)
assistant_prompt = PR_PERFORMANCE_ANALYSIS_PROMPT["assistant"]

Expand All @@ -147,10 +153,18 @@ async def analyze_pr_with_gemini(text: str) -> dict:
{"role": "assistant", "content": assistant_prompt},
]

# Clear stale cache before creating new tools for this analysis
clear_pr_files_cache()

# Create PR investigation tools and combine with MCP tools
pr_tools = create_pr_tools(org, repo, pr_number)
all_tools = mcp_module.mcp_tools + pr_tools

# Use the agentic loop with tool calling
result = await analyze_with_agentic(
messages=messages,
tools=mcp_module.mcp_tools,
tools=all_tools,
max_iterations=PR_ANALYSIS_MAX_TOOL_ITERATIONS,
)

# Handle empty results
Expand Down
Loading