fix(langchain): DeepInfra model parsing (#973)

hassiebp · web-flow · commit 51b55c28beb1 · 2024-10-22T20:27:08.000+02:00
diff --git a/langfuse/extract_model.py b/langfuse/extract_model.py
@@ -74,6 +74,7 @@ def _extract_model_name(
         ("VLLM", "model", None),
         ("Xinference", "model_uid", None),
         ("ChatOCIGenAI", "model_id", None),
+        ("DeepInfra", "model_id", None),
     ]
 
     for model_name, pattern, default in models_by_pattern:

Original file line number	Diff line number	Diff line change
`@@ -74,6 +74,7 @@ def _extract_model_name(`
`74`	`74`	`("VLLM", "model", None),`
`75`	`75`	`("Xinference", "model_uid", None),`
`76`	`76`	`("ChatOCIGenAI", "model_id", None),`
	`77`	`+ ("DeepInfra", "model_id", None),`
`77`	`78`	`]`
`78`	`79`
`79`	`80`	`for model_name, pattern, default in models_by_pattern:`