Skip to content

Commit ba01184

Browse files
committed
fix(langchain): deployment version added to azure model
1 parent 4dc9258 commit ba01184

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

langfuse/extract_model.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,12 @@ def _extract_model_name(
5858
deployment_version = None
5959
if serialized.get("kwargs").get("deployment_name"):
6060
deployment_name = serialized.get("kwargs").get("deployment_name")
61-
return deployment_name + "-" + deployment_version
61+
62+
return (
63+
deployment_name + "-" + deployment_version
64+
if deployment_version not in deployment_name
65+
else deployment_name
66+
)
6267

6368
# Third, for some models, we are unable to extract the model by a path in an object. Langfuse provides us with a string representation of the model pbjects
6469
# We use regex to extract the model from the repr string

0 commit comments

Comments
 (0)