Skip to content

Commit 80a6066

Browse files
authored
fix(langchain): deployment version in azure model (#1203)
1 parent 4dc9258 commit 80a6066

File tree

1 file changed

+16
-3
lines changed

1 file changed

+16
-3
lines changed

langfuse/extract_model.py

Lines changed: 16 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -53,12 +53,25 @@ def _extract_model_name(
5353
return kwargs.get("invocation_params").get("model_name")
5454

5555
deployment_name = None
56-
if serialized.get("kwargs").get("openai_api_version"):
57-
deployment_name = serialized.get("kwargs").get("deployment_version")
5856
deployment_version = None
57+
58+
if serialized.get("kwargs").get("openai_api_version"):
59+
deployment_version = serialized.get("kwargs").get("deployment_version")
60+
5961
if serialized.get("kwargs").get("deployment_name"):
6062
deployment_name = serialized.get("kwargs").get("deployment_name")
61-
return deployment_name + "-" + deployment_version
63+
64+
if not isinstance(deployment_name, str):
65+
return None
66+
67+
if not isinstance(deployment_version, str):
68+
return deployment_name
69+
70+
return (
71+
f"{deployment_name}-{deployment_version}"
72+
if deployment_version and deployment_version not in deployment_name
73+
else deployment_name
74+
)
6275

6376
# 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
6477
# We use regex to extract the model from the repr string

0 commit comments

Comments
 (0)