bump trace attributes version and cleanup

Author: obinnascale3

pyproject.toml

@@ -18,7 +18,7 @@ classifiers=[
     "Operating System :: OS Independent",
 ]
 dependencies = [
-  'trace-attributes==7.0.1',
+  'trace-attributes==7.0.4',
   'opentelemetry-api>=1.25.0',
   'opentelemetry-sdk>=1.25.0',
   'opentelemetry-instrumentation>=0.47b0',

src/langtrace_python_sdk/instrumentation/mistral/patch.py

@@ -31,7 +31,6 @@
 )
 from langtrace_python_sdk.constants.instrumentation.mistral import APIS
 from langtrace_python_sdk.utils.llm import (
-    calculate_prompt_tokens,
     get_extra_attributes,
     get_langtrace_attributes,
     get_llm_request_attributes,
@@ -77,11 +76,9 @@ def traced_method(wrapped, instance, args, kwargs):
         try:
             result = wrapped(*args, **kwargs)
             if is_streaming:
-                prompt_tokens = 0
                 return StreamWrapper(
                     result,
                     span,
-                    prompt_tokens,
                     function_call=kwargs.get("functions") is not None,
                     tool_calls=kwargs.get("tools") is not None,
                 )