fix: ignore assembled_response for endpoints other than `/chat/comp…

…letions` (#110)
epam · Feb 20, 2025 · 55b74ba · 55b74ba
1 parent 6aa91d3
commit 55b74ba
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -783,6 +783,7 @@ def test_rate_request():
                                 }
                             ),
                         },
+                        "assembled_response": "",
                         "response": {
                             "status": "200",
                             "body": "",