fix: pre-commit local

This commit is contained in:
zxmfke
2025-03-22 13:38:27 +08:00
parent f37c3c29a1
commit cdbefb5fc7
2 changed files with 6 additions and 4 deletions
+1 -1
View File
@@ -1,7 +1,7 @@
from app.agent.base import BaseAgent
from app.agent.browser import BrowserAgent
from app.agent.mcp import MCPAgent
from app.agent.cot import CoTAgent
from app.agent.mcp import MCPAgent
from app.agent.planning import PlanningAgent
from app.agent.react import ReActAgent
from app.agent.swe import SWEAgent
+5 -3
View File
@@ -1,10 +1,10 @@
from typing import List, Optional
from typing import Optional
from pydantic import Field
from app.agent.base import BaseAgent
from app.logger import logger
from app.llm import LLM
from app.logger import logger
from app.prompt.cot import NEXT_STEP_PROMPT, SYSTEM_PROMPT
from app.schema import AgentState, Message
@@ -33,7 +33,9 @@ class CoTAgent(BaseAgent):
# Use system prompt and user messages
response = await self.llm.ask(
messages=self.messages,
system_msgs=[Message.system_message(self.system_prompt)] if self.system_prompt else None
system_msgs=[Message.system_message(self.system_prompt)]
if self.system_prompt
else None,
)
# Record assistant's response