fix: add google as provider for openai instrumentor (#1131)

Arize-ai · Nov 22, 2024 · 3ef4243 · 3ef4243
1 parent 9e2ca3a
commit 3ef4243
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...openinference-instrumentation-openai/src/openinference/instrumentation/openai/_request.py b/...openinference-instrumentation-openai/src/openinference/instrumentation/openai/_request.py
@@ -132,6 +132,8 @@ def _get_attributes_from_instance(self, instance: Any) -> Iterator[Tuple[str, At
             yield SpanAttributes.LLM_PROVIDER, OpenInferenceLLMProviderValues.OPENAI.value
         elif host.endswith("openai.azure.com"):
             yield SpanAttributes.LLM_PROVIDER, OpenInferenceLLMProviderValues.AZURE.value
+        elif host.endswith("googleapis.com"):
+            yield SpanAttributes.LLM_PROVIDER, OpenInferenceLLMProviderValues.GOOGLE.value
 
     def _get_attributes_from_request(
         self,