openlit 1.33.1__tar.gz → 1.33.4__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (122) hide show
  1. {openlit-1.33.1 → openlit-1.33.4}/PKG-INFO +5 -4
  2. {openlit-1.33.1 → openlit-1.33.4}/README.md +2 -1
  3. {openlit-1.33.1 → openlit-1.33.4}/pyproject.toml +1 -1
  4. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/astra/astra.py +1 -1
  5. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/astra/async_astra.py +1 -1
  6. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/langchain/langchain.py +25 -10
  7. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/openai/async_openai.py +2 -1
  8. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/openai/openai.py +2 -1
  9. {openlit-1.33.1 → openlit-1.33.4}/LICENSE +0 -0
  10. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/__helpers.py +0 -0
  11. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/__init__.py +0 -0
  12. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/evals/__init__.py +0 -0
  13. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/evals/all.py +0 -0
  14. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/evals/bias_detection.py +0 -0
  15. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/evals/hallucination.py +0 -0
  16. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/evals/toxicity.py +0 -0
  17. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/evals/utils.py +0 -0
  18. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/guard/__init__.py +0 -0
  19. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/guard/all.py +0 -0
  20. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/guard/prompt_injection.py +0 -0
  21. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/guard/restrict_topic.py +0 -0
  22. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/guard/sensitive_topic.py +0 -0
  23. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/guard/utils.py +0 -0
  24. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/ag2/__init__.py +0 -0
  25. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/ag2/ag2.py +0 -0
  26. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/ai21/__init__.py +0 -0
  27. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/ai21/ai21.py +0 -0
  28. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/ai21/async_ai21.py +0 -0
  29. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/anthropic/__init__.py +0 -0
  30. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/anthropic/anthropic.py +0 -0
  31. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/anthropic/async_anthropic.py +0 -0
  32. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/assemblyai/__init__.py +0 -0
  33. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/assemblyai/assemblyai.py +0 -0
  34. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/astra/__init__.py +0 -0
  35. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/azure_ai_inference/__init__.py +0 -0
  36. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/azure_ai_inference/async_azure_ai_inference.py +0 -0
  37. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/azure_ai_inference/azure_ai_inference.py +0 -0
  38. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/bedrock/__init__.py +0 -0
  39. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/bedrock/bedrock.py +0 -0
  40. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/chroma/__init__.py +0 -0
  41. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/chroma/chroma.py +0 -0
  42. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/cohere/__init__.py +0 -0
  43. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/cohere/cohere.py +0 -0
  44. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/controlflow/__init__.py +0 -0
  45. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/controlflow/controlflow.py +0 -0
  46. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/crawl4ai/__init__.py +0 -0
  47. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/crawl4ai/async_crawl4ai.py +0 -0
  48. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/crawl4ai/crawl4ai.py +0 -0
  49. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/crewai/__init__.py +0 -0
  50. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/crewai/crewai.py +0 -0
  51. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/dynamiq/__init__.py +0 -0
  52. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/dynamiq/dynamiq.py +0 -0
  53. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/elevenlabs/__init__.py +0 -0
  54. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/elevenlabs/async_elevenlabs.py +0 -0
  55. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/elevenlabs/elevenlabs.py +0 -0
  56. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/embedchain/__init__.py +0 -0
  57. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/embedchain/embedchain.py +0 -0
  58. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/firecrawl/__init__.py +0 -0
  59. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/firecrawl/firecrawl.py +0 -0
  60. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/google_ai_studio/__init__.py +0 -0
  61. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/google_ai_studio/async_google_ai_studio.py +0 -0
  62. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/google_ai_studio/google_ai_studio.py +0 -0
  63. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/gpt4all/__init__.py +0 -0
  64. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/gpt4all/gpt4all.py +0 -0
  65. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/gpu/__init__.py +0 -0
  66. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/groq/__init__.py +0 -0
  67. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/groq/async_groq.py +0 -0
  68. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/groq/groq.py +0 -0
  69. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/haystack/__init__.py +0 -0
  70. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/haystack/haystack.py +0 -0
  71. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/julep/__init__.py +0 -0
  72. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/julep/async_julep.py +0 -0
  73. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/julep/julep.py +0 -0
  74. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/langchain/__init__.py +0 -0
  75. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/letta/__init__.py +0 -0
  76. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/letta/letta.py +0 -0
  77. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/litellm/__init__.py +0 -0
  78. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/litellm/async_litellm.py +0 -0
  79. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/litellm/litellm.py +0 -0
  80. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/llamaindex/__init__.py +0 -0
  81. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/llamaindex/llamaindex.py +0 -0
  82. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/mem0/__init__.py +0 -0
  83. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/mem0/mem0.py +0 -0
  84. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/milvus/__init__.py +0 -0
  85. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/milvus/milvus.py +0 -0
  86. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/mistral/__init__.py +0 -0
  87. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/mistral/async_mistral.py +0 -0
  88. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/mistral/mistral.py +0 -0
  89. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/multion/__init__.py +0 -0
  90. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/multion/async_multion.py +0 -0
  91. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/multion/multion.py +0 -0
  92. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/ollama/__init__.py +0 -0
  93. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/ollama/async_ollama.py +0 -0
  94. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/ollama/ollama.py +0 -0
  95. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/openai/__init__.py +0 -0
  96. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/openai/async_azure_openai.py +0 -0
  97. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/openai/azure_openai.py +0 -0
  98. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/phidata/__init__.py +0 -0
  99. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/phidata/phidata.py +0 -0
  100. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/pinecone/__init__.py +0 -0
  101. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/pinecone/pinecone.py +0 -0
  102. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/premai/__init__.py +0 -0
  103. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/premai/premai.py +0 -0
  104. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/qdrant/__init__.py +0 -0
  105. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/qdrant/async_qdrant.py +0 -0
  106. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/qdrant/qdrant.py +0 -0
  107. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/reka/__init__.py +0 -0
  108. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/reka/async_reka.py +0 -0
  109. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/reka/reka.py +0 -0
  110. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/together/__init__.py +0 -0
  111. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/together/async_together.py +0 -0
  112. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/together/together.py +0 -0
  113. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/transformers/__init__.py +0 -0
  114. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/transformers/transformers.py +0 -0
  115. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/vertexai/__init__.py +0 -0
  116. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/vertexai/async_vertexai.py +0 -0
  117. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/vertexai/vertexai.py +0 -0
  118. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/vllm/__init__.py +0 -0
  119. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/instrumentation/vllm/vllm.py +0 -0
  120. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/otel/metrics.py +0 -0
  121. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/otel/tracing.py +0 -0
  122. {openlit-1.33.1 → openlit-1.33.4}/src/openlit/semcov/__init__.py +0 -0
@@ -1,8 +1,7 @@
1
- Metadata-Version: 2.1
1
+ Metadata-Version: 2.3
2
2
  Name: openlit
3
- Version: 1.33.1
3
+ Version: 1.33.4
4
4
  Summary: OpenTelemetry-native Auto instrumentation library for monitoring LLM Applications and GPUs, facilitating the integration of observability into your GenAI-driven projects
5
- Home-page: https://github.com/openlit/openlit/tree/main/openlit/python
6
5
  Keywords: OpenTelemetry,otel,otlp,llm,tracing,openai,anthropic,claude,cohere,llm monitoring,observability,monitoring,gpt,Generative AI,chatGPT,gpu
7
6
  Author: OpenLIT
8
7
  Requires-Python: >=3.7.1,<4.0.0
@@ -26,6 +25,7 @@ Requires-Dist: requests (>=2.26.0,<3.0.0)
26
25
  Requires-Dist: schedule (>=1.2.2,<2.0.0)
27
26
  Requires-Dist: tiktoken (>=0.7.0,<0.8.0)
28
27
  Requires-Dist: xmltodict (>=0.13.0,<0.14.0)
28
+ Project-URL: Homepage, https://github.com/openlit/openlit/tree/main/openlit/python
29
29
  Project-URL: Repository, https://github.com/openlit/openlit/tree/main/openlit/python
30
30
  Description-Content-Type: text/markdown
31
31
 
@@ -83,7 +83,7 @@ This project proudly follows and maintains the [Semantic Conventions](https://gi
83
83
  | [✅ vLLM](https://docs.openlit.io/latest/integrations/vllm) | | [✅ Crawl4AI](https://docs.openlit.io/latest/integrations/crawl4ai) | |
84
84
  | [✅ OLA Krutrim](https://docs.openlit.io/latest/integrations/krutrim) | | [✅ FireCrawl](https://docs.openlit.io/latest/integrations/firecrawl) | |
85
85
  | [✅ Google AI Studio](https://docs.openlit.io/latest/integrations/google-ai-studio) | | [✅ Letta](https://docs.openlit.io/latest/integrations/letta) | |
86
- | [✅ NVIDIA NIM](https://docs.openlit.io/latest/integrations/nvidia-nim) | | | |
86
+ | [✅ NVIDIA NIM](https://docs.openlit.io/latest/integrations/nvidia-nim) | | [✅ SwarmZero](https://docs.openlit.io/latest/integrations/swarmzero) | |
87
87
  | [✅ Titan ML](https://docs.openlit.io/latest/integrations/titan-ml) | | | |
88
88
  | [✅ Reka AI](https://docs.openlit.io/latest/integrations/reka) | | | |
89
89
  | [✅ xAI](https://docs.openlit.io/latest/integrations/xai) | | | |
@@ -106,6 +106,7 @@ This project proudly follows and maintains the [Semantic Conventions](https://gi
106
106
  - [✅ Dynatrace](https://docs.openlit.io/latest/connections/dynatrace)
107
107
  - [✅ OpenObserve](https://docs.openlit.io/latest/connections/openobserve)
108
108
  - [✅ Highlight.io](https://docs.openlit.io/latest/connections/highlight)
109
+ - [✅ SigLens](https://docs.openlit.io/latest/connections/siglens)
109
110
 
110
111
  ## 💿 Installation
111
112
 
@@ -52,7 +52,7 @@ This project proudly follows and maintains the [Semantic Conventions](https://gi
52
52
  | [✅ vLLM](https://docs.openlit.io/latest/integrations/vllm) | | [✅ Crawl4AI](https://docs.openlit.io/latest/integrations/crawl4ai) | |
53
53
  | [✅ OLA Krutrim](https://docs.openlit.io/latest/integrations/krutrim) | | [✅ FireCrawl](https://docs.openlit.io/latest/integrations/firecrawl) | |
54
54
  | [✅ Google AI Studio](https://docs.openlit.io/latest/integrations/google-ai-studio) | | [✅ Letta](https://docs.openlit.io/latest/integrations/letta) | |
55
- | [✅ NVIDIA NIM](https://docs.openlit.io/latest/integrations/nvidia-nim) | | | |
55
+ | [✅ NVIDIA NIM](https://docs.openlit.io/latest/integrations/nvidia-nim) | | [✅ SwarmZero](https://docs.openlit.io/latest/integrations/swarmzero) | |
56
56
  | [✅ Titan ML](https://docs.openlit.io/latest/integrations/titan-ml) | | | |
57
57
  | [✅ Reka AI](https://docs.openlit.io/latest/integrations/reka) | | | |
58
58
  | [✅ xAI](https://docs.openlit.io/latest/integrations/xai) | | | |
@@ -75,6 +75,7 @@ This project proudly follows and maintains the [Semantic Conventions](https://gi
75
75
  - [✅ Dynatrace](https://docs.openlit.io/latest/connections/dynatrace)
76
76
  - [✅ OpenObserve](https://docs.openlit.io/latest/connections/openobserve)
77
77
  - [✅ Highlight.io](https://docs.openlit.io/latest/connections/highlight)
78
+ - [✅ SigLens](https://docs.openlit.io/latest/connections/siglens)
78
79
 
79
80
  ## 💿 Installation
80
81
 
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "openlit"
3
- version = "1.33.1"
3
+ version = "1.33.4"
4
4
  description = "OpenTelemetry-native Auto instrumentation library for monitoring LLM Applications and GPUs, facilitating the integration of observability into your GenAI-driven projects"
5
5
  authors = ["OpenLIT"]
6
6
  repository = "https://github.com/openlit/openlit/tree/main/openlit/python"
@@ -87,7 +87,7 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
87
87
  span.set_attribute(SemanticConvetion.DB_INDEX_DIMENSION,
88
88
  kwargs.get("dimension", ""))
89
89
  span.set_attribute(SemanticConvetion.DB_INDEX_METRIC,
90
- kwargs.get("metric", ""))
90
+ str(kwargs.get("metric", "")))
91
91
  span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
92
92
  instance.api_endpoint)
93
93
 
@@ -87,7 +87,7 @@ def general_wrap(gen_ai_endpoint, version, environment, application_name,
87
87
  span.set_attribute(SemanticConvetion.DB_INDEX_DIMENSION,
88
88
  kwargs.get("dimension", ""))
89
89
  span.set_attribute(SemanticConvetion.DB_INDEX_METRIC,
90
- kwargs.get("metric", ""))
90
+ str(kwargs.get("metric", "")))
91
91
  span.set_attribute(SemanticConvetion.DB_OPERATION_API_ENDPOINT,
92
92
  instance.api_endpoint)
93
93
 
@@ -508,11 +508,11 @@ def chat(gen_ai_endpoint, version, environment, application_name,
508
508
 
509
509
  try:
510
510
  prompt = ""
511
- if hasattr(response, 'response_metadata') and response.response_metadata:
512
- token_usage = response.response_metadata.get("token_usage", {})
513
- input_tokens = token_usage.get("prompt_tokens", 0)
514
- output_tokens = token_usage.get("completion_tokens", 0)
515
- model = response.response_metadata.get("model_name", "gpt-4")
511
+ if hasattr(response, 'usage_metadata') and response.usage_metadata:
512
+ token_usage = response.usage_metadata
513
+ input_tokens = token_usage.get("input_tokens", 0)
514
+ output_tokens = token_usage.get("output_tokens", 0)
515
+ model = instance.model_id
516
516
  prompt = "" if isinstance(args[0], list) else args[0]
517
517
  else:
518
518
  if not isinstance(response, dict) or "output_text" not in response:
@@ -661,12 +661,27 @@ def achat(gen_ai_endpoint, version, environment, application_name,
661
661
  response = await wrapped(*args, **kwargs)
662
662
 
663
663
  try:
664
- token_usage = response.response_metadata.get("token_usage", {})
665
- input_tokens = token_usage.get("prompt_tokens", 0)
666
- output_tokens = token_usage.get("completion_tokens", 0)
667
- model = response.response_metadata.get("model_name", "gpt-4")
664
+ prompt = ""
665
+ if hasattr(response, 'usage_metadata') and response.usage_metadata:
666
+ token_usage = response.usage_metadata
667
+ input_tokens = token_usage.get("input_tokens", 0)
668
+ output_tokens = token_usage.get("output_tokens", 0)
669
+ model = instance.model_id
670
+ prompt = "" if isinstance(args[0], list) else args[0]
671
+
672
+ else:
673
+ if not isinstance(response, dict) or "output_text" not in response:
674
+ return response
675
+ # Fallback: Calculate tokens manually if response_metadata is missing
676
+ model = "gpt-4o-mini" # Fallback model
677
+ input_texts = [
678
+ doc.page_content for doc in response.get("input_documents", [])
679
+ if isinstance(doc.page_content, str)
680
+ ]
681
+ input_tokens = sum(general_tokens(text) for text in input_texts)
682
+ output_text = response.get("output_text", "")
683
+ output_tokens = general_tokens(output_text)
668
684
 
669
- prompt = "" if isinstance(args[0], list) else args[0]
670
685
  # Calculate cost of the operation
671
686
  cost = get_chat_model_cost(
672
687
  model,
@@ -478,11 +478,12 @@ def async_embedding(gen_ai_endpoint, version, environment, application_name,
478
478
  response_dict.get('usage').get('total_tokens'))
479
479
  span.set_attribute(SemanticConvetion.GEN_AI_USAGE_COST,
480
480
  cost)
481
+
481
482
  if trace_content:
482
483
  span.add_event(
483
484
  name=SemanticConvetion.GEN_AI_CONTENT_PROMPT_EVENT,
484
485
  attributes={
485
- SemanticConvetion.GEN_AI_CONTENT_PROMPT: kwargs.get("input", ""),
486
+ SemanticConvetion.GEN_AI_CONTENT_PROMPT: str(kwargs.get("input", "")),
486
487
  },
487
488
  )
488
489
 
@@ -481,11 +481,12 @@ def embedding(gen_ai_endpoint, version, environment, application_name,
481
481
  response_dict.get('usage').get('total_tokens'))
482
482
  span.set_attribute(SemanticConvetion.GEN_AI_USAGE_COST,
483
483
  cost)
484
+
484
485
  if trace_content:
485
486
  span.add_event(
486
487
  name=SemanticConvetion.GEN_AI_CONTENT_PROMPT_EVENT,
487
488
  attributes={
488
- SemanticConvetion.GEN_AI_CONTENT_PROMPT: kwargs.get("input", ""),
489
+ SemanticConvetion.GEN_AI_CONTENT_PROMPT: str(kwargs.get("input", "")),
489
490
  },
490
491
  )
491
492
 
File without changes