Просмотр исходного кода

feat: optimize error logging (#808)

tags/0.3.13
takatost 2 лет назад
Родитель
Сommit
f18ce203b5
Аккаунт пользователя с таким Email не найден

+ 2
- 2
api/core/callback_handler/agent_loop_gather_callback_handler.py Просмотреть файл

def on_llm_error( def on_llm_error(
self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any
) -> None: ) -> None:
logging.error(error)
logging.exception(error)
self._agent_loops = [] self._agent_loops = []
self._current_loop = None self._current_loop = None
self._message_agent_thought = None self._message_agent_thought = None
self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any
) -> None: ) -> None:
"""Do nothing.""" """Do nothing."""
logging.error(error)
logging.exception(error)
self._agent_loops = [] self._agent_loops = []
self._current_loop = None self._current_loop = None
self._message_agent_thought = None self._message_agent_thought = None

+ 1
- 1
api/core/callback_handler/dataset_tool_callback_handler.py Просмотреть файл

self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any
) -> None: ) -> None:
"""Do nothing.""" """Do nothing."""
logging.error(error)
logging.exception(error)

+ 1
- 1
api/core/callback_handler/llm_callback_handler.py Просмотреть файл

) )
self.conversation_message_task.save_message(llm_message=self.llm_message, by_stopped=True) self.conversation_message_task.save_message(llm_message=self.llm_message, by_stopped=True)
else: else:
logging.error(error)
logging.exception(error)

+ 2
- 2
api/core/callback_handler/main_chain_gather_callback_handler.py Просмотреть файл

def on_chain_error( def on_chain_error(
self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any
) -> None: ) -> None:
logging.error(error)
self.clear_chain_results()
logging.exception(error)
self.clear_chain_results()

+ 1
- 1
api/core/indexing_runner.py Просмотреть файл

qa_documents.append(qa_document) qa_documents.append(qa_document)
format_documents.extend(qa_documents) format_documents.extend(qa_documents)
except Exception as e: except Exception as e:
logging.error(str(e))
logging.exception(e)


all_qa_documents.extend(format_documents) all_qa_documents.extend(format_documents)



Загрузка…
Отмена
Сохранить