openlit 1.33.10__py3-none-any.whl → 1.33.11__py3-none-any.whl

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 (112) hide show
  1. openlit/__helpers.py +73 -0
  2. openlit/__init__.py +38 -11
  3. openlit/instrumentation/ag2/__init__.py +9 -10
  4. openlit/instrumentation/ag2/ag2.py +133 -68
  5. openlit/instrumentation/ai21/__init__.py +6 -5
  6. openlit/instrumentation/ai21/ai21.py +71 -534
  7. openlit/instrumentation/ai21/async_ai21.py +71 -534
  8. openlit/instrumentation/ai21/utils.py +407 -0
  9. openlit/instrumentation/anthropic/__init__.py +3 -3
  10. openlit/instrumentation/anthropic/anthropic.py +4 -4
  11. openlit/instrumentation/anthropic/async_anthropic.py +4 -4
  12. openlit/instrumentation/assemblyai/__init__.py +2 -2
  13. openlit/instrumentation/assemblyai/assemblyai.py +3 -3
  14. openlit/instrumentation/astra/__init__.py +25 -25
  15. openlit/instrumentation/astra/astra.py +2 -2
  16. openlit/instrumentation/astra/async_astra.py +2 -2
  17. openlit/instrumentation/azure_ai_inference/__init__.py +5 -5
  18. openlit/instrumentation/azure_ai_inference/async_azure_ai_inference.py +8 -8
  19. openlit/instrumentation/azure_ai_inference/azure_ai_inference.py +8 -8
  20. openlit/instrumentation/bedrock/__init__.py +2 -2
  21. openlit/instrumentation/bedrock/bedrock.py +3 -3
  22. openlit/instrumentation/chroma/__init__.py +9 -9
  23. openlit/instrumentation/chroma/chroma.py +2 -2
  24. openlit/instrumentation/cohere/__init__.py +7 -7
  25. openlit/instrumentation/cohere/async_cohere.py +9 -9
  26. openlit/instrumentation/cohere/cohere.py +9 -9
  27. openlit/instrumentation/controlflow/__init__.py +4 -4
  28. openlit/instrumentation/controlflow/controlflow.py +2 -2
  29. openlit/instrumentation/crawl4ai/__init__.py +3 -3
  30. openlit/instrumentation/crawl4ai/async_crawl4ai.py +2 -2
  31. openlit/instrumentation/crawl4ai/crawl4ai.py +2 -2
  32. openlit/instrumentation/crewai/__init__.py +3 -3
  33. openlit/instrumentation/crewai/crewai.py +2 -2
  34. openlit/instrumentation/dynamiq/__init__.py +5 -5
  35. openlit/instrumentation/dynamiq/dynamiq.py +2 -2
  36. openlit/instrumentation/elevenlabs/__init__.py +5 -5
  37. openlit/instrumentation/elevenlabs/async_elevenlabs.py +3 -3
  38. openlit/instrumentation/elevenlabs/elevenlabs.py +3 -3
  39. openlit/instrumentation/embedchain/__init__.py +2 -2
  40. openlit/instrumentation/embedchain/embedchain.py +4 -4
  41. openlit/instrumentation/firecrawl/__init__.py +3 -3
  42. openlit/instrumentation/firecrawl/firecrawl.py +2 -2
  43. openlit/instrumentation/google_ai_studio/__init__.py +3 -3
  44. openlit/instrumentation/google_ai_studio/async_google_ai_studio.py +3 -3
  45. openlit/instrumentation/google_ai_studio/google_ai_studio.py +3 -3
  46. openlit/instrumentation/gpt4all/__init__.py +3 -3
  47. openlit/instrumentation/gpt4all/gpt4all.py +7 -7
  48. openlit/instrumentation/groq/__init__.py +3 -3
  49. openlit/instrumentation/groq/async_groq.py +5 -5
  50. openlit/instrumentation/groq/groq.py +5 -5
  51. openlit/instrumentation/haystack/__init__.py +2 -2
  52. openlit/instrumentation/haystack/haystack.py +2 -2
  53. openlit/instrumentation/julep/__init__.py +7 -7
  54. openlit/instrumentation/julep/async_julep.py +3 -3
  55. openlit/instrumentation/julep/julep.py +3 -3
  56. openlit/instrumentation/langchain/__init__.py +2 -2
  57. openlit/instrumentation/langchain/async_langchain.py +13 -9
  58. openlit/instrumentation/langchain/langchain.py +13 -8
  59. openlit/instrumentation/letta/__init__.py +7 -7
  60. openlit/instrumentation/letta/letta.py +5 -5
  61. openlit/instrumentation/litellm/__init__.py +5 -5
  62. openlit/instrumentation/litellm/async_litellm.py +8 -8
  63. openlit/instrumentation/litellm/litellm.py +8 -8
  64. openlit/instrumentation/llamaindex/__init__.py +2 -2
  65. openlit/instrumentation/llamaindex/llamaindex.py +2 -2
  66. openlit/instrumentation/mem0/__init__.py +2 -2
  67. openlit/instrumentation/mem0/mem0.py +2 -2
  68. openlit/instrumentation/milvus/__init__.py +2 -2
  69. openlit/instrumentation/milvus/milvus.py +2 -2
  70. openlit/instrumentation/mistral/__init__.py +7 -7
  71. openlit/instrumentation/mistral/async_mistral.py +10 -10
  72. openlit/instrumentation/mistral/mistral.py +10 -10
  73. openlit/instrumentation/multion/__init__.py +7 -7
  74. openlit/instrumentation/multion/async_multion.py +5 -5
  75. openlit/instrumentation/multion/multion.py +5 -5
  76. openlit/instrumentation/ollama/__init__.py +11 -9
  77. openlit/instrumentation/ollama/async_ollama.py +71 -465
  78. openlit/instrumentation/ollama/ollama.py +71 -465
  79. openlit/instrumentation/ollama/utils.py +333 -0
  80. openlit/instrumentation/openai/__init__.py +11 -11
  81. openlit/instrumentation/openai/async_openai.py +18 -18
  82. openlit/instrumentation/openai/openai.py +18 -18
  83. openlit/instrumentation/phidata/__init__.py +2 -2
  84. openlit/instrumentation/phidata/phidata.py +2 -2
  85. openlit/instrumentation/pinecone/__init__.py +6 -6
  86. openlit/instrumentation/pinecone/pinecone.py +2 -2
  87. openlit/instrumentation/premai/__init__.py +3 -3
  88. openlit/instrumentation/premai/premai.py +7 -7
  89. openlit/instrumentation/qdrant/__init__.py +2 -2
  90. openlit/instrumentation/qdrant/async_qdrant.py +2 -2
  91. openlit/instrumentation/qdrant/qdrant.py +2 -2
  92. openlit/instrumentation/reka/__init__.py +3 -3
  93. openlit/instrumentation/reka/async_reka.py +3 -3
  94. openlit/instrumentation/reka/reka.py +3 -3
  95. openlit/instrumentation/together/__init__.py +5 -5
  96. openlit/instrumentation/together/async_together.py +8 -8
  97. openlit/instrumentation/together/together.py +8 -8
  98. openlit/instrumentation/transformers/__init__.py +2 -2
  99. openlit/instrumentation/transformers/transformers.py +4 -4
  100. openlit/instrumentation/vertexai/__init__.py +9 -9
  101. openlit/instrumentation/vertexai/async_vertexai.py +4 -4
  102. openlit/instrumentation/vertexai/vertexai.py +4 -4
  103. openlit/instrumentation/vllm/__init__.py +2 -2
  104. openlit/instrumentation/vllm/vllm.py +3 -3
  105. openlit/otel/events.py +85 -0
  106. openlit/otel/tracing.py +3 -13
  107. openlit/semcov/__init__.py +13 -1
  108. {openlit-1.33.10.dist-info → openlit-1.33.11.dist-info}/METADATA +2 -2
  109. openlit-1.33.11.dist-info/RECORD +125 -0
  110. openlit-1.33.10.dist-info/RECORD +0 -122
  111. {openlit-1.33.10.dist-info → openlit-1.33.11.dist-info}/LICENSE +0 -0
  112. {openlit-1.33.10.dist-info → openlit-1.33.11.dist-info}/WHEEL +0 -0
openlit/instrumentation/multion/async_multion.py CHANGED
@@ -15,7 +15,7 @@ from openlit.semcov import SemanticConvetion
15
15
  logger = logging.getLogger(__name__)
16
16
 
17
17
  def async_multion_wrap(gen_ai_endpoint, version, environment, application_name,
18
- tracer, pricing_info, trace_content, metrics, disable_metrics):
18
+ tracer, pricing_info, capture_message_content, metrics, disable_metrics):
19
19
  """
20
20
  Generates a telemetry wrapper for chat completions to collect metrics.
21
21
 
@@ -26,7 +26,7 @@ def async_multion_wrap(gen_ai_endpoint, version, environment, application_name,
26
26
  application_name: Name of the application using the multion Agent.
27
27
  tracer: OpenTelemetry tracer for creating spans.
28
28
  pricing_info: Information used for calculating the cost of multion usage.
29
- trace_content: Flag indicating whether to trace the actual content.
29
+ capture_message_content: Flag indicating whether to trace the actual content.
30
30
 
31
31
  Returns:
32
32
  A function that wraps the chat completions method to add telemetry.
@@ -75,7 +75,7 @@ def async_multion_wrap(gen_ai_endpoint, version, environment, application_name,
75
75
  span.set_attribute(SemanticConvetion.GEN_AI_AGENT_RESPONSE_TIME,
76
76
  response.metadata.processing_time)
77
77
 
78
- if trace_content:
78
+ if capture_message_content:
79
79
  span.add_event(
80
80
  name=SemanticConvetion.GEN_AI_CONTENT_PROMPT_EVENT,
81
81
  attributes={
@@ -92,7 +92,7 @@ def async_multion_wrap(gen_ai_endpoint, version, environment, application_name,
92
92
  span.set_attribute(SemanticConvetion.GEN_AI_AGENT_BROWSE_URL,
93
93
  kwargs.get("url", ""))
94
94
 
95
- if trace_content:
95
+ if capture_message_content:
96
96
  span.add_event(
97
97
  name=SemanticConvetion.GEN_AI_CONTENT_PROMPT_EVENT,
98
98
  attributes={
@@ -110,7 +110,7 @@ def async_multion_wrap(gen_ai_endpoint, version, environment, application_name,
110
110
  span.set_attribute(SemanticConvetion.GEN_AI_AGENT_BROWSE_URL,
111
111
  kwargs.get("url", ""))
112
112
 
113
- if trace_content:
113
+ if capture_message_content:
114
114
  span.add_event(
115
115
  name=SemanticConvetion.GEN_AI_CONTENT_COMPLETION_EVENT,
116
116
  attributes={
openlit/instrumentation/multion/multion.py CHANGED
@@ -15,7 +15,7 @@ from openlit.semcov import SemanticConvetion
15
15
  logger = logging.getLogger(__name__)
16
16
 
17
17
  def multion_wrap(gen_ai_endpoint, version, environment, application_name,
18
- tracer, pricing_info, trace_content, metrics, disable_metrics):
18
+ tracer, pricing_info, capture_message_content, metrics, disable_metrics):
19
19
  """
20
20
  Generates a telemetry wrapper for chat completions to collect metrics.
21
21
 
@@ -26,7 +26,7 @@ def multion_wrap(gen_ai_endpoint, version, environment, application_name,
26
26
  application_name: Name of the application using the multion Agent.
27
27
  tracer: OpenTelemetry tracer for creating spans.
28
28
  pricing_info: Information used for calculating the cost of multion usage.
29
- trace_content: Flag indicating whether to trace the actual content.
29
+ capture_message_content: Flag indicating whether to trace the actual content.
30
30
 
31
31
  Returns:
32
32
  A function that wraps the chat completions method to add telemetry.
@@ -75,7 +75,7 @@ def multion_wrap(gen_ai_endpoint, version, environment, application_name,
75
75
  span.set_attribute(SemanticConvetion.GEN_AI_AGENT_RESPONSE_TIME,
76
76
  response.metadata.processing_time)
77
77
 
78
- if trace_content:
78
+ if capture_message_content:
79
79
  span.add_event(
80
80
  name=SemanticConvetion.GEN_AI_CONTENT_PROMPT_EVENT,
81
81
  attributes={
@@ -92,7 +92,7 @@ def multion_wrap(gen_ai_endpoint, version, environment, application_name,
92
92
  span.set_attribute(SemanticConvetion.GEN_AI_AGENT_BROWSE_URL,
93
93
  kwargs.get("url", ""))
94
94
 
95
- if trace_content:
95
+ if capture_message_content:
96
96
  span.add_event(
97
97
  name=SemanticConvetion.GEN_AI_CONTENT_PROMPT_EVENT,
98
98
  attributes={
@@ -110,7 +110,7 @@ def multion_wrap(gen_ai_endpoint, version, environment, application_name,
110
110
  span.set_attribute(SemanticConvetion.GEN_AI_AGENT_BROWSE_URL,
111
111
  kwargs.get("url", ""))
112
112
 
113
- if trace_content:
113
+ if capture_message_content:
114
114
  span.add_event(
115
115
  name=SemanticConvetion.GEN_AI_CONTENT_COMPLETION_EVENT,
116
116
  attributes={
openlit/instrumentation/ollama/__init__.py CHANGED
@@ -1,5 +1,6 @@
1
- # pylint: disable=useless-return, bad-staticmethod-argument, disable=duplicate-code
2
- """Initializer of Auto Instrumentation of Ollama Functions"""
1
+ """
2
+ Initializer of Auto Instrumentation of Ollama Functions
3
+ """
3
4
 
4
5
  from typing import Collection
5
6
  import importlib.metadata
@@ -27,9 +28,10 @@ class OllamaInstrumentor(BaseInstrumentor):
27
28
  application_name = kwargs.get("application_name", "default_application")
28
29
  environment = kwargs.get("environment", "default_environment")
29
30
  tracer = kwargs.get("tracer")
31
+ event_provider = kwargs.get("event_provider")
30
32
  metrics = kwargs.get("metrics_dict")
31
33
  pricing_info = kwargs.get("pricing_info", {})
32
- trace_content = kwargs.get("trace_content", False)
34
+ capture_message_content = kwargs.get("capture_message_content", False)
33
35
  disable_metrics = kwargs.get("disable_metrics")
34
36
  version = importlib.metadata.version("ollama")
35
37
 
@@ -38,13 +40,13 @@ class OllamaInstrumentor(BaseInstrumentor):
38
40
  "ollama",
39
41
  "chat",
40
42
  chat(version, environment, application_name,
41
- tracer, pricing_info, trace_content, metrics, disable_metrics),
43
+ tracer, event_provider, pricing_info, capture_message_content, metrics, disable_metrics),
42
44
  )
43
45
  wrap_function_wrapper(
44
46
  "ollama",
45
47
  "Client.chat",
46
48
  chat(version, environment, application_name,
47
- tracer, pricing_info, trace_content, metrics, disable_metrics),
49
+ tracer, event_provider, pricing_info, capture_message_content, metrics, disable_metrics),
48
50
  )
49
51
 
50
52
  # sync embeddings
@@ -52,13 +54,13 @@ class OllamaInstrumentor(BaseInstrumentor):
52
54
  "ollama",
53
55
  "embeddings",
54
56
  embeddings(version, environment, application_name,
55
- tracer, pricing_info, trace_content, metrics, disable_metrics),
57
+ tracer, event_provider, pricing_info, capture_message_content, metrics, disable_metrics),
56
58
  )
57
59
  wrap_function_wrapper(
58
60
  "ollama",
59
61
  "Client.embeddings",
60
62
  embeddings(version, environment, application_name,
61
- tracer, pricing_info, trace_content, metrics, disable_metrics),
63
+ tracer, event_provider, pricing_info, capture_message_content, metrics, disable_metrics),
62
64
  )
63
65
 
64
66
  # async chat
@@ -66,7 +68,7 @@ class OllamaInstrumentor(BaseInstrumentor):
66
68
  "ollama",
67
69
  "AsyncClient.chat",
68
70
  async_chat(version, environment, application_name,
69
- tracer, pricing_info, trace_content, metrics, disable_metrics),
71
+ tracer, event_provider, pricing_info, capture_message_content, metrics, disable_metrics),
70
72
  )
71
73
 
72
74
  # async embeddings
@@ -74,7 +76,7 @@ class OllamaInstrumentor(BaseInstrumentor):
74
76
  "ollama",
75
77
  "AsyncClient.embeddings",
76
78
  async_embeddings(version, environment, application_name,
77
- tracer, pricing_info, trace_content, metrics, disable_metrics),
79
+ tracer, event_provider, pricing_info, capture_message_content, metrics, disable_metrics),
78
80
  )
79
81
 
80
82
  def _uninstrument(self, **kwargs):