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

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,