fix(langchain): add bedrock chat and non chat types (#187) (#192)

maxdeichmann · chaseWillden · web-flow · commit fe084be8204e · 2023-11-22T13:12:56.000Z
Co-authored-by: Chase Willden &lt;chase.willden@gmail.com&gt;
diff --git a/langfuse/callback.py b/langfuse/callback.py
@@ -477,7 +477,7 @@ def __on_llm_action(
                 )
             if kwargs["invocation_params"]["_type"] in ["anthropic-llm", "anthropic-chat"]:
                 model_name = "anthropic"  # unfortunately no model info by anthropic provided.
-            elif kwargs["invocation_params"]["_type"] == "amazon_bedrock":
+            elif kwargs["invocation_params"]["_type"] in ["amazon_bedrock", "amazon_bedrock_chat"]:
                 # langchain only provides string representation of the model class. Hence have to parse it out.
                 model_name = self.extract_second_part(self.extract_model_id("model_id", serialized["repr"]))
             elif kwargs["invocation_params"]["_type"] == "cohere-chat":

Original file line number	Diff line number	Diff line change
`@@ -477,7 +477,7 @@ def __on_llm_action(`
`477`	`477`	`)`
`478`	`478`	`if kwargs["invocation_params"]["_type"] in ["anthropic-llm", "anthropic-chat"]:`
`479`	`479`	`model_name = "anthropic" # unfortunately no model info by anthropic provided.`
`480`		`- elif kwargs["invocation_params"]["_type"] == "amazon_bedrock":`
	`480`	`+ elif kwargs["invocation_params"]["_type"] in ["amazon_bedrock", "amazon_bedrock_chat"]:`
`481`	`481`	`# langchain only provides string representation of the model class. Hence have to parse it out.`
`482`	`482`	`model_name = self.extract_second_part(self.extract_model_id("model_id", serialized["repr"]))`
`483`	`483`	`elif kwargs["invocation_params"]["_type"] == "cohere-chat":`