mirror of
https://github.com/Pythagora-io/gpt-pilot.git
synced 2026-01-09 13:17:55 -05:00
5b474ccc1f
This is a complete rewrite of the GPT Pilot core, from the ground up, making the agentic architecture front and center, and also fixing some long-standing problems with the database architecture that weren't feasible to solve without breaking compatibility. As the database structure and config file syntax have changed, we have automatic imports for projects and current configs, see the README.md file for details. This also relicenses the project to FSL-1.1-MIT license.
54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
from core.agents.base import BaseAgent
|
|
from core.agents.convo import AgentConvo
|
|
from core.agents.response import AgentResponse
|
|
from core.log import get_logger
|
|
|
|
log = get_logger(__name__)
|
|
|
|
|
|
class TaskReviewer(BaseAgent):
|
|
agent_type = "task-reviewer"
|
|
display_name = "Task Reviewer"
|
|
|
|
async def run(self) -> AgentResponse:
|
|
response = await self.review_code_changes()
|
|
self.next_state.complete_step()
|
|
return response
|
|
|
|
async def review_code_changes(self) -> AgentResponse:
|
|
"""
|
|
Review all the code changes during current task.
|
|
"""
|
|
|
|
log.debug(f"Reviewing code changes for task {self.current_state.current_task['description']}")
|
|
await self.send_message("Reviewing the task implementation ...")
|
|
all_feedbacks = [
|
|
iteration["user_feedback"].replace("```", "").strip()
|
|
for iteration in self.current_state.iterations
|
|
# Some iterations are created by the task reviewer and have no user feedback
|
|
if iteration["user_feedback"]
|
|
]
|
|
|
|
files_before_modification = self.current_state.modified_files
|
|
files_after_modification = [
|
|
(file.path, file.content.content)
|
|
for file in self.current_state.files
|
|
if (file.path in files_before_modification)
|
|
]
|
|
|
|
llm = self.get_llm()
|
|
# TODO instead of sending files before and after maybe add nice way to show diff for multiple files
|
|
convo = AgentConvo(self).template(
|
|
"review_task",
|
|
current_task=self.current_state.current_task,
|
|
all_feedbacks=all_feedbacks,
|
|
files_before_modification=files_before_modification,
|
|
files_after_modification=files_after_modification,
|
|
)
|
|
llm_response: str = await llm(convo, temperature=0.7)
|
|
|
|
if llm_response.strip().lower() == "done":
|
|
return AgentResponse.done(self)
|
|
else:
|
|
return AgentResponse.task_review_feedback(self, llm_response)
|