langtrace-python-sdk 1.3.6__tar.gz → 2.0.0__tar.gz

{langtrace_python_sdk-1.3.6 → langtrace_python_sdk-2.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 1.3.6
+Version: 2.0.0
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>
@@ -15,7 +15,7 @@ Requires-Dist: opentelemetry-instrumentation
 Requires-Dist: opentelemetry-sdk
 Requires-Dist: pinecone-client
 Requires-Dist: tiktoken
-Requires-Dist: trace-attributes==1.0.32
+Requires-Dist: trace-attributes==3.0.2
 Provides-Extra: dev
 Requires-Dist: anthropic; extra == 'dev'
 Requires-Dist: chromadb; extra == 'dev'

{langtrace_python_sdk-1.3.6 → langtrace_python_sdk-2.0.0}/pyproject.toml

@@ -18,7 +18,7 @@ classifiers=[
     "Operating System :: OS Independent",
 ]
 dependencies = [
-  'trace-attributes==1.0.32',
+  'trace-attributes==3.0.2',
   'opentelemetry-api',
   'opentelemetry-sdk',
   'opentelemetry-instrumentation',

{langtrace_python_sdk-1.3.6 → langtrace_python_sdk-2.0.0}/src/examples/anthropic_example/completion.py

@@ -3,13 +3,13 @@
 import anthropic
 from dotenv import find_dotenv, load_dotenv
 
-from langtrace_python_sdk import langtrace
-from langtrace_python_sdk import with_langtrace_root_span
+from langtrace_python_sdk import langtrace, with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
 langtrace.init(write_to_langtrace_cloud=False)
 
+
 @with_langtrace_root_span("messages_create")
 def messages_create():
 
@@ -21,10 +21,10 @@ def messages_create():
         temperature=0.0,
         system="Respond only in Yoda-speak.",
         messages=[{"role": "user", "content": "How are you today?"}],
-        stream=False,
+        stream=True,
     )
 
-    print(message)
+    # print(message)
 
-    # for response in message:
-    #     pass
+    for response in message:
+        pass

{langtrace_python_sdk-1.3.6 → langtrace_python_sdk-2.0.0}/src/examples/cohere_example/chat.py

@@ -1,7 +1,8 @@
-from dotenv import find_dotenv, load_dotenv
 import cohere
+from dotenv import find_dotenv, load_dotenv
 
 from langtrace_python_sdk import langtrace
+
 # from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
@@ -18,9 +19,9 @@ def chat_comp():
             {"role": "USER", "message": "Who discovered gravity?"},
             {"role": "CHATBOT", "message": "The man who is widely credited with discovering gravity is Sir Isaac Newton"}
         ],
-        message="What is today's news?",
-        # preamble="answer like yoda",
+        message="Tell me a story in 3 sentences or less?",
+        preamble="answer like a pirate",
         # perform web search before answering the question. You can also use your own custom connector.
-        # connectors=[{"id": "web-search"}]
+        connectors=[{"id": "web-search"}]
     )
     print(response)

{langtrace_python_sdk-1.3.6 → langtrace_python_sdk-2.0.0}/src/examples/cohere_example/chat_stream.py

@@ -1,10 +1,8 @@
-from dotenv import find_dotenv, load_dotenv
 import cohere
+from dotenv import find_dotenv, load_dotenv
 
 from langtrace_python_sdk import langtrace
 
-# from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
-
 _ = load_dotenv(find_dotenv())
 
 langtrace.init(write_to_langtrace_cloud=False)
@@ -15,7 +13,7 @@ co = cohere.Client()
 # @with_langtrace_root_span("chat_stream")
 def chat_stream():
     result = []
-    for event in co.chat_stream(message="Tell me a short story in 2 lines"):
+    for event in co.chat_stream(message="Tell me a short story in 2 lines", preamble="Respond like a pirate", max_tokens=100):
         if event.event_type == "text-generation":
             result.append(event.text)
         elif event.event_type == "stream-end":

langtrace_python_sdk-1.3.6/src/examples/cohere_example/embed_create.py → langtrace_python_sdk-2.0.0/src/examples/cohere_example/embed.py

@@ -1,5 +1,5 @@
-from dotenv import find_dotenv, load_dotenv
 import cohere
+from dotenv import find_dotenv, load_dotenv
 
 from langtrace_python_sdk import langtrace
 
@@ -13,10 +13,11 @@ co = cohere.Client()
 
 
 # @with_langtrace_root_span("embed_create")
-def embed_create():
+def embed():
     response = co.embed(
         texts=["hello", "goodbye"],
         model="embed-english-v3.0",
         input_type="classification",
     )
-    # print(response)
+    print(response)
+    return response

langtrace_python_sdk-2.0.0/src/examples/cohere_example/rerank.py

@@ -0,0 +1,31 @@
+import cohere
+from dotenv import find_dotenv, load_dotenv
+
+from langtrace_python_sdk import langtrace
+
+# from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
+
+_ = load_dotenv(find_dotenv())
+
+langtrace.init(write_to_langtrace_cloud=False)
+
+co = cohere.Client()
+
+
+# @with_langtrace_root_span("embed_create")
+def rerank():
+    docs = [
+        "Carson City is the capital city of the American state of Nevada.",
+        "The Commonwealth of the Northern Mariana Islands is a group of islands in the Pacific Ocean. Its capital is Saipan.",
+        "Washington, D.C. (also known as simply Washington or D.C., and officially as the District of Columbia) is the capital of the United States. It is a federal district.",
+        "Capital punishment (the death penalty) has existed in the United States since beforethe United States was a country. As of 2017, capital punishment is legal in 30 of the 50 states.",
+    ]
+
+    response = co.rerank(
+        model="rerank-english-v2.0",
+        query="What is the capital of the United States?",
+        documents=docs,
+        top_n=3,
+    )
+    print(response)
+    return response

langtrace_python_sdk-2.0.0/src/examples/cohere_example/tools.py

@@ -0,0 +1,40 @@
+# import json
+
+import cohere
+from dotenv import find_dotenv, load_dotenv
+
+from langtrace_python_sdk import langtrace
+
+_ = load_dotenv(find_dotenv())
+
+langtrace.init(write_to_langtrace_cloud=False)
+
+co = cohere.Client()
+
+
+student_custom_functions = [
+    {
+        "name": "extract_student_info",
+        "description": "Get the student information from the body of the input text",
+        "parameter_definitions": {
+            "name": {"type": "string", "description": "Name of the person", "required": True},
+            "major": {"type": "string", "description": "Major subject.", "required": True},
+            "school": {"type": "string", "description": "The university name.", "required": True},
+            "grades": {"type": "integer", "description": "GPA of the student.", "required": True},
+            "club": {
+                "type": "string",
+                "description": "School club for extracurricular activities. ",
+                "required": False,
+            },
+        },
+    }
+]
+
+
+def tool_calling():
+    response = co.chat(
+        message="John is a grad student in computer science at Stanford University. He is an American and has a 3.8 GPA. John is known for his programming skills and is an active member of the university's Robotics Club. He hopes to pursue a career in artificial intelligence after graduating.",
+        tools=student_custom_functions,
+    )
+    print(response)
+    return response

langtrace_python_sdk-2.0.0/src/examples/openai_example/chat_completion.py

@@ -0,0 +1,41 @@
+from dotenv import find_dotenv, load_dotenv
+from openai import OpenAI
+
+from langtrace_python_sdk import langtrace
+from langtrace_python_sdk.utils.with_root_span import (
+    with_additional_attributes, with_langtrace_root_span)
+
+_ = load_dotenv(find_dotenv())
+
+langtrace.init(write_to_langtrace_cloud=False)
+client = OpenAI()
+
+
+# @with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
+def api():
+    response = client.chat.completions.create(
+        model="gpt-4",
+        messages=[{"role": "system", "content": "Talk like a pirate"}, {"role": "user", "content": "Tell me a story in 3 sentences or less."}],
+        stream=True,
+        # stream=False,
+    )
+    return response
+
+
+# @with_langtrace_root_span()
+def chat_completion():
+    response = api()
+    # print(response)
+    # Uncomment this for streaming
+    result = []
+    for chunk in response:
+        if chunk.choices[0].delta.content is not None:
+            content = [
+                choice.delta.content if choice.delta and
+                choice.delta.content else ""
+                for choice in chunk.choices]
+            result.append(
+                content[0] if len(content) > 0 else "")
+
+    print("".join(result))
+    return response

{langtrace_python_sdk-1.3.6/src/examples/openai → langtrace_python_sdk-2.0.0/src/examples/openai_example}/embeddings_create.py

@@ -6,7 +6,7 @@ from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(batch=True, log_spans_to_console=True, write_to_remote_url=False)
+langtrace.init(write_to_langtrace_cloud=False)
 client = OpenAI()
 
 
@@ -14,5 +14,6 @@ client = OpenAI()
 def embeddings_create():
     result = client.embeddings.create(
         model="text-embedding-ada-002",
-        input="Once upon a time, there was a frog.",
+        input="Once upon a time, there was a pirate.",
     )
+    return result

{langtrace_python_sdk-1.3.6/src/examples/openai → langtrace_python_sdk-2.0.0/src/examples/openai_example}/function_calling.py

@@ -1,10 +1,9 @@
-import json
+# import json
 
 from dotenv import find_dotenv, load_dotenv
 from openai import OpenAI
 
 from langtrace_python_sdk import langtrace
-from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
@@ -34,20 +33,19 @@ student_custom_functions = [
 ]
 
 
-# @with_langtrace_root_span()
 def function_calling():
     response = client.chat.completions.create(
         model="gpt-3.5-turbo",
         messages=[
             {
                 "role": "user",
-                "content": "David Nguyen is a sophomore majoring in computer science at Stanford University. He is Asian American and has a 3.8 GPA. David is known for his programming skills and is an active member of the university's Robotics Club. He hopes to pursue a career in artificial intelligence after graduating.",
+                "content": "John is a grad student in computer science at Stanford University. He is an American and has a 3.8 GPA. John is known for his programming skills and is an active member of the university's Robotics Club. He hopes to pursue a career in artificial intelligence after graduating.",
             }
         ],
         functions=student_custom_functions,
-        function_call="auto",
         stream=True,
     )
+    # return response
 
     result = []
     for chunk in response:

{langtrace_python_sdk-1.3.6/src/examples/openai → langtrace_python_sdk-2.0.0/src/examples/openai_example}/images_generate.py

@@ -6,7 +6,7 @@ from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(batch=True, log_spans_to_console=True, write_to_remote_url=False)
+langtrace.init(write_to_langtrace_cloud=False)
 
 client = OpenAI()
 

langtrace_python_sdk-2.0.0/src/examples/openai_example/tool_calling.py

@@ -0,0 +1,67 @@
+# import json
+
+from dotenv import find_dotenv, load_dotenv
+from openai import OpenAI
+
+from langtrace_python_sdk import langtrace
+
+_ = load_dotenv(find_dotenv())
+
+langtrace.init(write_to_langtrace_cloud=False)
+
+client = OpenAI()
+
+
+student_custom_functions = [
+    {
+        "type": "function",
+        "function": {
+            "name": "extract_student_info",
+            "description": "Get the student information from the body of the input text",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "name": {"type": "string", "description": "Name of the person"},
+                    "major": {"type": "string", "description": "Major subject."},
+                    "school": {"type": "string", "description": "The university name."},
+                    "grades": {"type": "integer", "description": "GPA of the student."},
+                    "club": {
+                        "type": "string",
+                        "description": "School club for extracurricular activities. ",
+                    },
+                },
+            },
+        },
+    }
+]
+
+
+def tool_calling():
+    response = client.chat.completions.create(
+        model="gpt-3.5-turbo",
+        messages=[
+            {
+                "role": "user",
+                "content": "John is a grad student in computer science at Stanford University. He is an American and has a 3.8 GPA. John is known for his programming skills and is an active member of the university's Robotics Club. He hopes to pursue a career in artificial intelligence after graduating.",
+            }
+        ],
+        tools=student_custom_functions,
+        stream=False,
+    )
+    return response
+
+    # result = []
+    # for chunk in response:
+    #     if chunk.choices[0].delta.function_call is not None:
+    #         content = [
+    #             choice.delta.function_call.arguments if choice.delta.function_call and
+    #             choice.delta.function_call.arguments else ""
+    #             for choice in chunk.choices]
+    #         result.append(
+    #             content[0] if len(content) > 0 else "")
+
+    # print("".join(result))
+
+    # Loading the response as a JSON object
+    # json_response = json.loads(response.choices[0].message.function_call.arguments)
+    # print(json_response)

{langtrace_python_sdk-1.3.6/src/examples/openai → langtrace_python_sdk-2.0.0/src/examples/openai_example}/tool_calling_nonstreaming.py

@@ -13,6 +13,7 @@ langtrace.init(write_to_langtrace_cloud=False)
 
 client = OpenAI()
 
+
 # Example dummy function hard coded to return the same weather
 # In production, this could be your backend API or an external API
 def get_current_weather(location, unit="fahrenheit"):
@@ -89,4 +90,4 @@ def run_conversation():
             messages=messages,
         )  # get a new response from the model where it can see the function response
         # print(second_response)
-        return second_response
+        return second_response

langtrace_python_sdk-2.0.0/src/langtrace_python_sdk/__init__.py

@@ -0,0 +1,20 @@
+"""
+Copyright (c) 2024 Scale3 Labs
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+from langtrace_python_sdk import langtrace
+from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
+
+__all__ = ["langtrace", "with_langtrace_root_span"]

{langtrace_python_sdk-1.3.6 → langtrace_python_sdk-2.0.0}/src/langtrace_python_sdk/constants/instrumentation/cohere.py

@@ -4,7 +4,7 @@ APIS = {
         "METHOD": "cohere.client.chat",
         "ENDPOINT": "/v1/chat",
     },
-    "EMBED_CREATE": {
+    "EMBED": {
         "URL": "https://api.cohere.ai",
         "METHOD": "cohere.client.embed",
         "ENDPOINT": "/v1/embed",
@@ -14,4 +14,9 @@ APIS = {
         "METHOD": "cohere.client.chat_stream",
         "ENDPOINT": "/v1/messages",
     },
+    "RERANK": {
+        "URL": "https://api.cohere.ai",
+        "METHOD": "cohere.client.rerank",
+        "ENDPOINT": "/v1/rerank",
+    },
 }

{langtrace_python_sdk-1.3.6 → langtrace_python_sdk-2.0.0}/src/langtrace_python_sdk/instrumentation/anthropic/instrumentation.py

@@ -1,17 +1,29 @@
 """
-Instrumentation for Anthropic
+Copyright (c) 2024 Scale3 Labs
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
 """
 
 import importlib.metadata
+import logging
 from typing import Collection
 
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.trace import get_tracer
 from wrapt import wrap_function_wrapper
 
-from langtrace_python_sdk.instrumentation.anthropic.patch import messages_create
-
-import logging
+from langtrace_python_sdk.instrumentation.anthropic.patch import \
+    messages_create
 
 logging.basicConfig(level=logging.FATAL)
 

{langtrace_python_sdk-1.3.6 → langtrace_python_sdk-2.0.0}/src/langtrace_python_sdk/instrumentation/anthropic/patch.py

@@ -1,5 +1,18 @@
 """
-This module contains the patching logic for the Anthropic library."""
+Copyright (c) 2024 Scale3 Labs
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
 
 import json
 
@@ -58,6 +71,8 @@ def messages_create(original_method, version, tracer):
             attributes.llm_top_p = kwargs.get("top_k")
         if kwargs.get("user") is not None:
             attributes.llm_user = kwargs.get("user")
+        if kwargs.get("max_tokens") is not None:
+            attributes.llm_max_tokens = str(kwargs.get("max_tokens"))
 
         span = tracer.start_span(
             APIS["MESSAGES_CREATE"]["METHOD"], kind=SpanKind.CLIENT
@@ -70,12 +85,14 @@ def messages_create(original_method, version, tracer):
             result = wrapped(*args, **kwargs)
             if kwargs.get("stream") is False:
                 if hasattr(result, "content") and result.content is not None:
+                    span.set_attribute("llm.model", result.model if result.model else kwargs.get("model"))
                     span.set_attribute(
                         "llm.responses",
                         json.dumps(
                             [
                                 {
-                                    "text": result.content[0].text,
+                                    "role": result.role if result.role else "assistant",
+                                    "content": result.content[0].text,
                                     "type": result.content[0].type,
                                 }
                             ]
@@ -106,11 +123,11 @@ def messages_create(original_method, version, tracer):
                 return result
             else:
                 return handle_streaming_response(result, span)
-        except Exception as e:
+        except Exception as err:
             # Record the exception in the span
-            span.record_exception(e)
+            span.record_exception(err)
             # Set the span status to indicate an error
-            span.set_status(Status(StatusCode.ERROR, str(e)))
+            span.set_status(Status(StatusCode.ERROR, str(err)))
             # Reraise the exception to ensure it's not swallowed
             span.end()
             raise
@@ -123,6 +140,8 @@ def messages_create(original_method, version, tracer):
         output_tokens = 0
         try:
             for chunk in result:
+                if hasattr(chunk, "message") and chunk.message is not None and hasattr(chunk.message, "model") and chunk.message.model is not None:
+                    span.set_attribute("llm.model", chunk.message.model)
                 content = ""
                 if hasattr(chunk, "delta") and chunk.delta is not None:
                     content = chunk.delta.text if hasattr(chunk.delta, "text") else ""
@@ -165,7 +184,8 @@ def messages_create(original_method, version, tracer):
                 ),
             )
             span.set_attribute(
-                "llm.responses", json.dumps([{"text": "".join(result_content)}])
+                "llm.responses",
+                json.dumps([{"role": "assistant", "content": "".join(result_content)}]),
             )
             span.set_status(StatusCode.OK)
             span.end()

{langtrace_python_sdk-1.3.6 → langtrace_python_sdk-2.0.0}/src/langtrace_python_sdk/instrumentation/chroma/instrumentation.py

@@ -1,8 +1,21 @@
 """
-Instrumentation for ChromaDB
+Copyright (c) 2024 Scale3 Labs
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
 """
 
 import importlib.metadata
+import logging
 from typing import Collection
 
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
@@ -11,7 +24,6 @@ from wrapt import wrap_function_wrapper
 
 from langtrace_python_sdk.constants.instrumentation.chroma import APIS
 from langtrace_python_sdk.instrumentation.chroma.patch import collection_patch
-import logging
 
 logging.basicConfig(level=logging.FATAL)
 

{langtrace_python_sdk-1.3.6 → langtrace_python_sdk-2.0.0}/src/langtrace_python_sdk/instrumentation/chroma/patch.py

@@ -1,5 +1,17 @@
 """
-This module contains the patching logic for the Chroma client.
+Copyright (c) 2024 Scale3 Labs
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
 """
 
 from langtrace.trace_attributes import DatabaseSpanAttributes
@@ -47,12 +59,12 @@ def collection_patch(method, version, tracer):
                 result = wrapped(*args, **kwargs)
                 span.set_status(StatusCode.OK)
                 return result
-            except Exception as e:
+            except Exception as err:
                 # Record the exception in the span
-                span.record_exception(e)
+                span.record_exception(err)
 
                 # Set the span status to indicate an error
-                span.set_status(Status(StatusCode.ERROR, str(e)))
+                span.set_status(Status(StatusCode.ERROR, str(err)))
 
                 # Reraise the exception to ensure it's not swallowed
                 raise

{langtrace_python_sdk-1.3.6 → langtrace_python_sdk-2.0.0}/src/langtrace_python_sdk/instrumentation/cohere/instrumentation.py

@@ -1,5 +1,17 @@
 """
-Instrumentation for Cohere
+Copyright (c) 2024 Scale3 Labs
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
 """
 
 import importlib.metadata
@@ -9,11 +21,10 @@ from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.trace import get_tracer
 from wrapt import wrap_function_wrapper
 
-from langtrace_python_sdk.instrumentation.cohere.patch import (
-    chat_create,
-    chat_stream,
-    embed_create
-)
+from langtrace_python_sdk.instrumentation.cohere.patch import (chat_create,
+                                                               chat_stream,
+                                                               embed, rerank)
+
 
 class CohereInstrumentation(BaseInstrumentor):
     """
@@ -43,7 +54,13 @@ class CohereInstrumentation(BaseInstrumentor):
         wrap_function_wrapper(
             "cohere.client",
             "Client.embed",
-            embed_create("cohere.client.embed", version, tracer),
+            embed("cohere.client.embed", version, tracer),
+        )
+
+        wrap_function_wrapper(
+            "cohere.client",
+            "Client.rerank",
+            rerank("cohere.client.rerank", version, tracer),
         )
 
     def _instrument_module(self, module_name):