fix: callback bug

85a25148 · John Wang · c2f1be9d · 85a25148
Commit 85a25148 authored Jun 20, 2023 by John Wang
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

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

No files found.
--- a/api/core/callback_handler/llm_callback_handler.py
+++ b/api/core/callback_handler/llm_callback_handler.py
@@ -31,6 +31,7 @@ class LLMCallbackHandler(BaseCallbackHandler):
            messages: List[List[BaseMessage]],
            **kwargs: Any
    ) -> Any:
+        self.start_at = time.perf_counter()
        real_prompts = []
        for message in messages[0]:
            if message.type == 'human':