From 8618006620435152998a0bcb525d57ec70c218e9 Mon Sep 17 00:00:00 2001
From: Anton Dubovik <anton_dubovik@epam.com>
Date: Thu, 20 Feb 2025 13:11:09 +0100
Subject: [PATCH] fix: ignore `assembled_response` for endpoints other than
 `/chat/completions` (#110)

---
 aidial_analytics_realtime/app.py | 2 +-
 tests/test_app.py                | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/aidial_analytics_realtime/app.py b/aidial_analytics_realtime/app.py
index fe74af9..da8d016 100644
--- a/aidial_analytics_realtime/app.py
+++ b/aidial_analytics_realtime/app.py
@@ -209,7 +209,6 @@ async def on_log_message(
     parent_deployment = message.get("parent_deployment")
     execution_path = message.get("execution_path")
     deployment = message.get("deployment") or ""
-    response_body = get_assembled_response(message)
 
     if re.search(RATE_PATTERN, uri):
         await on_rate_message(
@@ -225,6 +224,7 @@ async def on_log_message(
         )
 
     elif re.search(CHAT_COMPLETION_PATTERN, uri):
+        response_body = get_assembled_response(message)
         await on_chat_completion_message(
             deployment,
             project_id,
diff --git a/tests/test_app.py b/tests/test_app.py
index c288010..c69de21 100644
--- a/tests/test_app.py
+++ b/tests/test_app.py
@@ -783,6 +783,7 @@ def test_rate_request():
                                 }
                             ),
                         },
+                        "assembled_response": "",
                         "response": {
                             "status": "200",
                             "body": "",