Prechádzať zdrojové kódy

feat: optimize error logging (#808)

takatost 1 rok pred
rodič
commit
f18ce203b5

+ 2 - 2
api/core/callback_handler/agent_loop_gather_callback_handler.py

@@ -85,7 +85,7 @@ class AgentLoopGatherCallbackHandler(BaseCallbackHandler):
     def on_llm_error(
         self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any
     ) -> None:
-        logging.error(error)
+        logging.exception(error)
         self._agent_loops = []
         self._current_loop = None
         self._message_agent_thought = None
@@ -164,7 +164,7 @@ class AgentLoopGatherCallbackHandler(BaseCallbackHandler):
         self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any
     ) -> None:
         """Do nothing."""
-        logging.error(error)
+        logging.exception(error)
         self._agent_loops = []
         self._current_loop = None
         self._message_agent_thought = None

+ 1 - 1
api/core/callback_handler/dataset_tool_callback_handler.py

@@ -68,4 +68,4 @@ class DatasetToolCallbackHandler(BaseCallbackHandler):
         self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any
     ) -> None:
         """Do nothing."""
-        logging.error(error)
+        logging.exception(error)

+ 1 - 1
api/core/callback_handler/llm_callback_handler.py

@@ -96,4 +96,4 @@ class LLMCallbackHandler(BaseCallbackHandler):
                 )
                 self.conversation_message_task.save_message(llm_message=self.llm_message, by_stopped=True)
         else:
-            logging.error(error)
+            logging.exception(error)

+ 2 - 2
api/core/callback_handler/main_chain_gather_callback_handler.py

@@ -72,5 +72,5 @@ class MainChainGatherCallbackHandler(BaseCallbackHandler):
     def on_chain_error(
         self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any
     ) -> None:
-        logging.error(error)
-        self.clear_chain_results()
+        logging.exception(error)
+        self.clear_chain_results()

+ 1 - 1
api/core/indexing_runner.py

@@ -549,7 +549,7 @@ class IndexingRunner:
                 qa_documents.append(qa_document)
             format_documents.extend(qa_documents)
         except Exception as e:
-            logging.error(str(e))
+            logging.exception(e)
 
         all_qa_documents.extend(format_documents)