youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-01-19 11:45:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: optimize error logging (#808)

Browse Source
This commit is contained in:
takatost
2023-08-12 02:22:43 +08:00
committed by GitHub
parent b81b8637ec
commit f18ce203b5
5 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/callback_handler/agent_loop_gather_callback_handler.py
View File

@ -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

2
api/core/callback_handler/dataset_tool_callback_handler.py
View File

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

2
api/core/callback_handler/llm_callback_handler.py
View File

@ -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)

4
api/core/callback_handler/main_chain_gather_callback_handler.py
View File

@ -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()