chat

Author: lzchen

instrumentation-genai/opentelemetry-instrumentation-cohere-v2/src/opentelemetry/instrumentation/cohere_v2/utils.py

@@ -69,7 +69,7 @@ def get_property_value(obj, property_name):
 
 def set_server_address_and_port(client_instance: cohere.client_v2.V2Client, attributes):
     base_client = getattr(client_instance, "_client_wrapper", None)
-    base_url = getattr(base_client, "base_url", None)
+    base_url = getattr(base_client, "_base_url", None)
     if not base_url:
         return
 

instrumentation-genai/opentelemetry-instrumentation-cohere-v2/tests/cassettes/test_chat_completion_with_content.yaml

@@ -40,7 +40,7 @@ interactions:
     body:
       string: |-
         {
-          "id": "78c201dd-0f3d-43c4-a245-5ab6d28636ef",
+          "id": "06e43724-5286-4eeb-bb10-bad73f976655",
           "message": {
             "role": "assistant",
             "content": [
@@ -77,7 +77,7 @@ interactions:
       content-type:
       - application/json
       date:
-      - Mon, 09 Dec 2024 19:22:13 GMT
+      - Mon, 09 Dec 2024 21:08:41 GMT
       expires:
       - Thu, 01 Jan 1970 00:00:00 UTC
       num_chars:
@@ -93,15 +93,15 @@ interactions:
       x-accel-expires:
       - '0'
       x-debug-trace-id:
-      - 5bf598783ce550fab3dbd3c54bd617e7
+      - 52412790708b1cfbe79a11850fbc09ec
       x-endpoint-monthly-call-limit:
       - '1000'
       x-envoy-upstream-service-time:
-      - '212'
+      - '173'
       x-trial-endpoint-call-limit:
       - '40'
       x-trial-endpoint-call-remaining:
-      - '39'
+      - '37'
     status:
       code: 200
       message: OK

instrumentation-genai/opentelemetry-instrumentation-cohere-v2/tests/test_chat_completions.py

@@ -35,7 +35,7 @@
 
 @pytest.mark.vcr()
 def test_chat_completion_with_content(
-    span_exporter, log_exporter, cohere_client, instrument_with_content, tracer_provider
+    span_exporter, log_exporter, instrument_with_content, cohere_client
 ):
     llm_model_value = "command-r-plus"
     messages_value = [{"role": "user", "content": "Say this is a test"}]
@@ -731,9 +731,8 @@ def assert_completion_attributes(
         span,
         request_model,
         response.id,
-        response.model,
-        response.usage.prompt_tokens,
-        response.usage.completion_tokens,
+        response.usage.tokens.input_tokens,
+        response.usage.tokens.output_tokens,
         operation_name,
         server_address,
     )