feat: add support for meta vendor in openinference-instrumentation-be…

…drock (#448)
Arize-ai · May 11, 2024 · 958c385 · 958c385
1 parent b6e4cfe
commit 958c385
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...eninference-instrumentation-bedrock/src/openinference/instrumentation/bedrock/__init__.py b/...eninference-instrumentation-bedrock/src/openinference/instrumentation/bedrock/__init__.py
@@ -147,6 +147,8 @@ def instrumented_response(*args: Any, **kwargs: Any) -> Dict[str, Any]:
                         content = str(response_body.get("completion"))
                     elif vendor == "cohere":
                         content = str(response_body.get("generations"))
+                    elif vendor == "meta":
+                        content = str(response_body.get("generation"))
                     else:
                         content = ""