consistent logging

RasaHQ · Oct 13, 2023 · 34fb174 · 34fb174
1 parent 5602e3a
commit 34fb174
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/rasa/dialogue_understanding/generator/llm_command_generator.py b/rasa/dialogue_understanding/generator/llm_command_generator.py
@@ -140,11 +140,12 @@ def _generate_action_list_using_llm(self, prompt: str) -> Optional[str]:
             end = time.time()
 
             structlogger.info(
-                "rasa.llm_command_generator.llm_api_call.llm_output",
+                "rasa.llm_api_call.llm_command_generator.llm_output",
                 llm_output=llm_result.llm_output,
             )
             structlogger.info(
-                "rasa.llm_command_generator.llm_api_call.duration", duration=end - start
+                "rasa.llm_api_call.llm_command_generator.latency",
+                latency=end - start,
             )
 
             return llm_result.generations[0][0].text

diff --git a/rasa/shared/core/flows/flow.py b/rasa/shared/core/flows/flow.py
@@ -1120,10 +1120,10 @@ def generate(self, tracker: DialogueStateTracker) -> Optional[Text]:
             llm_result = llm.generate([prompt])
             end = time.time()
 
-            logger.info(
-                f"rasa.flow.llm_api_call.llm_output: llm_output = {llm_result.llm_output}"
+            structlogger.info(
+                f"rasa.llm_api_call.flow.llm_output", llm_output=llm_result.llm_output
             )
-            logger.info(f"rasa.flow.llm_api_call.duration: duration = {end - start}")
+            structlogger.info(f"rasa.llm_api_call.flow.latency", latency=end - start)
 
             return llm_result.generations[0][0].text
         except Exception as e:
@@ -1201,9 +1201,9 @@ def _generate_using_llm(self, prompt: str) -> Optional[str]:
             end = time.time()
 
             structlogger.info(
-                "rasa.flow.llm_api_call.llm_output", llm_output=llm_result.llm_output
+                "rasa.llm_api_call.flow.llm_output", llm_output=llm_result.llm_output
             )
-            structlogger.info("rasa.flow.llm_api_call.duration", duration=end - start)
+            structlogger.info("rasa.llm_api_call.flow.latency", latency=end - start)
 
             return llm_result.generations[0][0].text
         except Exception as e: