(session: SessionDep, log: ChatLog, full_message: Union[list[dict], dict, str],
reasoning_content: str = None,
token_usage=None)
| 851 | |
| 852 | |
| 853 | def end_log(session: SessionDep, log: ChatLog, full_message: Union[list[dict], dict, str], |
| 854 | reasoning_content: str = None, |
| 855 | token_usage=None) -> ChatLog: |
| 856 | if token_usage is None: |
| 857 | token_usage = {} |
| 858 | log.messages = full_message |
| 859 | log.token_usage = token_usage |
| 860 | log.finish_time = datetime.datetime.now() |
| 861 | log.reasoning_content = reasoning_content if reasoning_content and len(reasoning_content.strip()) > 0 else None |
| 862 | |
| 863 | stmt = update(ChatLog).where(and_(ChatLog.id == log.id)).values( |
| 864 | messages=log.messages, |
| 865 | token_usage=log.token_usage, |
| 866 | finish_time=log.finish_time, |
| 867 | reasoning_content=log.reasoning_content |
| 868 | ) |
| 869 | session.execute(stmt) |
| 870 | session.commit() |
| 871 | |
| 872 | return log |
| 873 | |
| 874 | |
| 875 | def trigger_log_error(session: SessionDep, log: ChatLog) -> ChatLog: |
no test coverage detected