From 826b7ab81b0cd9d0cfa551e13a8dd12cbad87cf7 Mon Sep 17 00:00:00 2001 From: Roger Yang Date: Thu, 21 Nov 2024 21:45:58 -0800 Subject: [PATCH] feat: add google as provider for openai instrumentor --- .../src/openinference/instrumentation/openai/_request.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/instrumentation/openinference-instrumentation-openai/src/openinference/instrumentation/openai/_request.py b/python/instrumentation/openinference-instrumentation-openai/src/openinference/instrumentation/openai/_request.py index 13b64de7e..6c3668f88 100644 --- a/python/instrumentation/openinference-instrumentation-openai/src/openinference/instrumentation/openai/_request.py +++ b/python/instrumentation/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,