add logging callback for workflow

2026-03-06 16:16:38 +08:00 · 2024-03-19 04:37:29 +08:00
parent 2da7cc6928
commit 1c7573a686
4 changed files with 145 additions and 9 deletions
--- a/api/core/model_runtime/model_providers/__base/large_language_model.py
+++ b/api/core/model_runtime/model_providers/__base/large_language_model.py
@ -63,7 +63,7 @@ class LargeLanguageModel(AIModel):

        callbacks = callbacks or []

-        if bool(os.environ.get("DEBUG")):
+        if bool(os.environ.get("DEBUG", 'False').lower() == 'true'):
            callbacks.append(LoggingCallback())

        # trigger before invoke callbacks