fix model id

we now pass model info to langfuse for generation observations
This commit is contained in:
ther3zz 2025-03-04 13:27:58 -05:00 committed by GitHub
parent 40c32238af
commit 34a464e1d9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -244,6 +244,7 @@ class Pipeline:
# If it's an LLM generation
generation_payload = {
"name": f"{task_name}:{str(uuid.uuid4())}",
"model": body.get("model"), # <-- Include the model in LLM generation
"input": body["messages"],
"metadata": metadata,
"usage": usage,