Clean up logging

xyvora · Dec 24, 2024 · 03d8290 · 03d8290
1 parent 817bd44
commit 03d8290
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 6 deletions.
diff --git a/backend/app/agents/agents.py b/backend/app/agents/agents.py
@@ -38,8 +38,6 @@ def _create_prompt(self) -> ChatPromptTemplate:
         return ChatPromptTemplate.from_template(template)
 
     async def process(self, state: AgentState) -> JsonDict:
-        logger.debug(f"DLPFC Agent processing state: {state}")
-
         response = await self.llm.ainvoke(
             self.prompt.format_messages(
                 task=state.task,

diff --git a/backend/app/agents/base.py b/backend/app/agents/base.py
@@ -33,9 +33,7 @@ def _create_prompt(self) -> ChatPromptTemplate:
     async def process(self, state: AgentState) -> JsonDict:
         """Process the current state and return updated state."""
         try:
-            logger.debug(f"Processing state with prompt: {self.prompt}")
             result = await self._process_with_timeout(state)
-            logger.debug(f"Received response: {result}")
             return result
         except TimeoutError:
             error_msg = "Request timed out. Please try again."
@@ -53,7 +51,6 @@ async def process(self, state: AgentState) -> JsonDict:
     async def _process_with_timeout(self, state: AgentState) -> JsonDict:
         """Process with timeout handling."""
         try:
-            logger.debug("Sending request to OpenAI API...")
             response = await self.llm.ainvoke(
                 self.prompt.format_messages(
                     task=state.task,
@@ -63,7 +60,6 @@ async def _process_with_timeout(self, state: AgentState) -> JsonDict:
                     feedback_history=state.feedback_history,
                 ),
             )
-            logger.debug(f"Received API response: {response}")
             return self._format_response(response.content)
         except TimeoutError:
             logger.debug("API request timed out")