move code review to sonnet

core/agents/code_monkey.py

@@ -8,7 +8,7 @@ from pydantic import BaseModel, Field
 from core.agents.base import BaseAgent
 from core.agents.convo import AgentConvo
 from core.agents.response import AgentResponse, ResponseType
-from core.config import CODE_MONKEY_AGENT_NAME, DESCRIBE_FILES_AGENT_NAME
+from core.config import CODE_MONKEY_AGENT_NAME, CODE_REVIEW_AGENT_NAME, DESCRIBE_FILES_AGENT_NAME
 from core.llm.parser import JSONParser, OptionalCodeBlockParser
 from core.log import get_logger
 
@@ -256,7 +256,7 @@ class CodeMonkey(BaseAgent):
 
         hunks = self.get_diff_hunks(file_name, old_content, new_content)
 
-        llm = self.get_llm()
+        llm = self.get_llm(CODE_REVIEW_AGENT_NAME)
         convo = (
             self._get_task_convo()
             .template(

core/config/__init__.py

@@ -35,6 +35,7 @@ IGNORE_SIZE_THRESHOLD = 50000  # 50K+ files are ignored by default
 # Agents with sane setup in the default configuration
 DEFAULT_AGENT_NAME = "default"
 CODE_MONKEY_AGENT_NAME = "CodeMonkey"
+CODE_REVIEW_AGENT_NAME = "CodeMonkey.code_review"
 DESCRIBE_FILES_AGENT_NAME = "CodeMonkey.describe_files"
 CHECK_LOGS_AGENT_NAME = "BugHunter.check_logs"
 PARSE_TASK_AGENT_NAME = "Developer.parse_task"
@@ -328,6 +329,11 @@ class Config(_StrictModel):
                 temperature=0.5,
             ),
             CODE_MONKEY_AGENT_NAME: AgentLLMConfig(model="gpt-4-0125-preview", temperature=0.0),
+            CODE_REVIEW_AGENT_NAME: AgentLLMConfig(
+                provider=LLMProvider.ANTHROPIC,
+                model="claude-3-5-sonnet-20240620",
+                temperature=0.0,
+            ),
             DESCRIBE_FILES_AGENT_NAME: AgentLLMConfig(model="gpt-3.5-turbo", temperature=0.0),
             GET_RELEVANT_FILES_AGENT_NAME: AgentLLMConfig(
                 provider=LLMProvider.ANTHROPIC,