pass model and cost so that langfuse can show cost (#13117)

This commit is contained in:
Obada Khalili 2025-02-02 09:27:27 +02:00 committed by GitHub
parent d73d191f99
commit 941ad03f3c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1066,8 +1066,10 @@ class Message(db.Model): # type: ignore[name-defined]
"id": self.id, "id": self.id,
"app_id": self.app_id, "app_id": self.app_id,
"conversation_id": self.conversation_id, "conversation_id": self.conversation_id,
"model_id": self.model_id,
"inputs": self.inputs, "inputs": self.inputs,
"query": self.query, "query": self.query,
"total_price": self.total_price,
"message": self.message, "message": self.message,
"answer": self.answer, "answer": self.answer,
"status": self.status, "status": self.status,
@ -1088,7 +1090,9 @@ class Message(db.Model): # type: ignore[name-defined]
id=data["id"], id=data["id"],
app_id=data["app_id"], app_id=data["app_id"],
conversation_id=data["conversation_id"], conversation_id=data["conversation_id"],
model_id=data["model_id"],
inputs=data["inputs"], inputs=data["inputs"],
total_price=data["total_price"],
query=data["query"], query=data["query"],
message=data["message"], message=data["message"],
answer=data["answer"], answer=data["answer"],