github/gpt-pilot
1
1
Fork 0
mirror of https://github.com/Pythagora-io/gpt-pilot.git synced 2026-01-10 13:37:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

change default agent config

Browse Source
This commit is contained in:
LeonOstrez
2024-07-25 12:42:56 +02:00
parent d8d4b5bcec
commit a0feb850be
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
core/agents/bug_hunter.py
View File

@@ -5,7 +5,7 @@ from pydantic import BaseModel, Field
from core.agents.base import BaseAgent
from core.agents.convo import AgentConvo
from core.agents.response import AgentResponse
from core.config import BUG_HUNT_AGENT_NAME, magic_words
from core.config import CHECK_LOGS_AGENT_NAME, magic_words
from core.db.models.project_state import IterationStatus
from core.llm.parser import JSONParser
from core.log import get_logger
@@ -61,7 +61,7 @@ class BugHunter(BaseAgent):
self.next_state.current_iteration["bug_reproduction_description"] = bug_reproduction_instructions
async def check_logs(self, logs_message: str = None):
llm = self.get_llm(BUG_HUNT_AGENT_NAME)
llm = self.get_llm(CHECK_LOGS_AGENT_NAME)
convo = AgentConvo(self).template(
"iteration",
current_task=self.current_state.current_task,

4
core/config/__init__.py
View File

@@ -36,7 +36,7 @@ IGNORE_SIZE_THRESHOLD = 50000 # 50K+ files are ignored by default
DEFAULT_AGENT_NAME = "default"
DESCRIBE_FILES_AGENT_NAME = "CodeMonkey.describe_files"
ROUTE_FILES_AGENT_NAME = "Troubleshooter.get_route_files"
BUG_HUNT_AGENT_NAME = "BugHunter.logs_or_fix"
CHECK_LOGS_AGENT_NAME = "BugHunter.check_logs"
# Endpoint for the external documentation
EXTERNAL_DOCUMENTATION_API = "http://docs-pythagora-io-439719575.us-east-1.elb.amazonaws.com"
@@ -311,7 +311,7 @@ class Config(_StrictModel):
default={
DEFAULT_AGENT_NAME: AgentLLMConfig(),
DESCRIBE_FILES_AGENT_NAME: AgentLLMConfig(model="gpt-3.5-turbo", temperature=0.0),
BUG_HUNT_AGENT_NAME: AgentLLMConfig(model="claude-3-5-sonnet-20240620", temperature=0.0),
CHECK_LOGS_AGENT_NAME: AgentLLMConfig(model="claude-3-5-sonnet-20240620", temperature=0.0),
ROUTE_FILES_AGENT_NAME: AgentLLMConfig(model="gpt-4o", temperature=0.0),
}
)