From 941ad03f3c27951345fc60884c60dd5235143d8f Mon Sep 17 00:00:00 2001
From: Obada Khalili <54270856+obadakhalili@users.noreply.github.com>
Date: Sun, 2 Feb 2025 09:27:27 +0200
Subject: [PATCH] pass model and cost so that langfuse can show cost (#13117)

---
 api/models/model.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/api/models/model.py b/api/models/model.py
index d6f73c5ede..2780b79c98 100644
--- a/api/models/model.py
+++ b/api/models/model.py
@@ -1066,8 +1066,10 @@ class Message(db.Model):  # type: ignore[name-defined]
             "id": self.id,
             "app_id": self.app_id,
             "conversation_id": self.conversation_id,
+            "model_id": self.model_id,
             "inputs": self.inputs,
             "query": self.query,
+            "total_price": self.total_price,
             "message": self.message,
             "answer": self.answer,
             "status": self.status,
@@ -1088,7 +1090,9 @@ class Message(db.Model):  # type: ignore[name-defined]
             id=data["id"],
             app_id=data["app_id"],
             conversation_id=data["conversation_id"],
+            model_id=data["model_id"],
             inputs=data["inputs"],
+            total_price=data["total_price"],
             query=data["query"],
             message=data["message"],
             answer=data["answer"],