ragaai-catalyst 2.2.1__py3-none-any.whl → 2.2.2__py3-none-any.whl

ragaai_catalyst/tracers/tracer.py

@@ -1,35 +1,26 @@
-import os
-import uuid
+import asyncio
 import datetime
+import json
 import logging
-import asyncio
+import os
+from concurrent.futures import ThreadPoolExecutor
+from contextlib import contextmanager
+from pathlib import Path
+
 import aiohttp
 import requests
 from litellm import model_cost
-from pathlib import Path
-from contextlib import contextmanager
-from concurrent.futures import ThreadPoolExecutor
-from ragaai_catalyst.tracers.langchain_callback import LangchainTracer
-from ragaai_catalyst.tracers.utils.convert_langchain_callbacks_output import convert_langchain_callbacks_output
 
-from ragaai_catalyst.tracers.utils.langchain_tracer_extraction_logic import langchain_tracer_extraction
-from ragaai_catalyst.tracers.upload_traces import UploadTraces
-import tempfile
-import json
-import numpy as np
-from opentelemetry.sdk import trace as trace_sdk
-from opentelemetry.sdk.trace.export import SimpleSpanProcessor
-from ragaai_catalyst.tracers.exporters.file_span_exporter import FileSpanExporter
-from ragaai_catalyst.tracers.exporters.raga_exporter import RagaExporter
-from ragaai_catalyst.tracers.utils import get_unique_key
 # from ragaai_catalyst.tracers.llamaindex_callback import LlamaIndexTracer
-from ragaai_catalyst.tracers.llamaindex_instrumentation import LlamaIndexInstrumentationTracer
 from openinference.instrumentation.langchain import LangChainInstrumentor
+from opentelemetry.sdk import trace as trace_sdk
+from opentelemetry.sdk.trace.export import SimpleSpanProcessor
+
 from ragaai_catalyst import RagaAICatalyst
 from ragaai_catalyst.tracers.agentic_tracing import AgenticTracing
-from ragaai_catalyst.tracers.agentic_tracing.tracers.llm_tracer import LLMTracerMixin
-from ragaai_catalyst.tracers.exporters.ragaai_trace_exporter import RAGATraceExporter
 from ragaai_catalyst.tracers.agentic_tracing.utils.file_name_tracker import TrackName
+from ragaai_catalyst.tracers.exporters.file_span_exporter import FileSpanExporter
+from ragaai_catalyst.tracers.utils import get_unique_key
 
 logger = logging.getLogger(__name__)
 logging_level = (
@@ -200,12 +191,16 @@ class Tracer(AgenticTracing):
             # Add LLM Instrumentors
             if tracer_type in ['agentic/crewai']:
                 try:
-                    from openinference.instrumentation.vertexai import VertexAIInstrumentor
+                    from openinference.instrumentation.vertexai import (
+                        VertexAIInstrumentor,
+                    )
                     instrumentors.append((VertexAIInstrumentor, []))
                 except (ImportError, ModuleNotFoundError):
                     logger.debug("VertexAI not available in environment")
                 try:
-                    from openinference.instrumentation.anthropic import AnthropicInstrumentor
+                    from openinference.instrumentation.anthropic import (
+                        AnthropicInstrumentor,
+                    )
                     instrumentors.append((AnthropicInstrumentor, []))
                 except (ImportError, ModuleNotFoundError):
                     logger.debug("Anthropic not available in environment")
@@ -215,12 +210,16 @@ class Tracer(AgenticTracing):
                 except (ImportError, ModuleNotFoundError):
                     logger.debug("Groq not available in environment")
                 try:
-                    from openinference.instrumentation.litellm import LiteLLMInstrumentor
+                    from openinference.instrumentation.litellm import (
+                        LiteLLMInstrumentor,
+                    )
                     instrumentors.append((LiteLLMInstrumentor, []))
                 except (ImportError, ModuleNotFoundError):
                     logger.debug("LiteLLM not available in environment")
                 try:
-                    from openinference.instrumentation.mistralai import MistralAIInstrumentor
+                    from openinference.instrumentation.mistralai import (
+                        MistralAIInstrumentor,
+                    )
                     instrumentors.append((MistralAIInstrumentor, []))
                 except (ImportError, ModuleNotFoundError):
                     logger.debug("MistralAI not available in environment")
@@ -230,7 +229,9 @@ class Tracer(AgenticTracing):
                 except (ImportError, ModuleNotFoundError):
                     logger.debug("OpenAI not available in environment")
                 try:
-                    from openinference.instrumentation.bedrock import BedrockInstrumentor
+                    from openinference.instrumentation.bedrock import (
+                        BedrockInstrumentor,
+                    )
                     instrumentors.append((BedrockInstrumentor, []))
                 except (ImportError, ModuleNotFoundError):
                     logger.debug("Bedrock not available in environment")
@@ -243,7 +244,9 @@ class Tracer(AgenticTracing):
                 try:
                     # LlamaIndex
                     try:
-                        from openinference.instrumentation.llama_index import LlamaIndexInstrumentor
+                        from openinference.instrumentation.llama_index import (
+                            LlamaIndexInstrumentor,
+                        )
                         instrumentors.append((LlamaIndexInstrumentor, []))
                         logger.info("Instrumenting LlamaIndex...")
                     except (ImportError, ModuleNotFoundError):
@@ -251,7 +254,9 @@ class Tracer(AgenticTracing):
                     
                     # LangChain
                     try:
-                        from openinference.instrumentation.langchain import LangChainInstrumentor
+                        from openinference.instrumentation.langchain import (
+                            LangChainInstrumentor,
+                        )
                         instrumentors.append((LangChainInstrumentor, []))
                         logger.info("Instrumenting LangChain...")
                     except (ImportError, ModuleNotFoundError):
@@ -259,7 +264,9 @@ class Tracer(AgenticTracing):
                     
                     # CrewAI
                     try:
-                        from openinference.instrumentation.crewai import CrewAIInstrumentor
+                        from openinference.instrumentation.crewai import (
+                            CrewAIInstrumentor,
+                        )
                         instrumentors.append((CrewAIInstrumentor, []))
                         logger.info("Instrumenting CrewAI...")
                     except (ImportError, ModuleNotFoundError):
@@ -267,7 +274,9 @@ class Tracer(AgenticTracing):
                     
                     # Haystack
                     try:
-                        from openinference.instrumentation.haystack import HaystackInstrumentor
+                        from openinference.instrumentation.haystack import (
+                            HaystackInstrumentor,
+                        )
                         instrumentors.append((HaystackInstrumentor, []))
                         logger.info("Instrumenting Haystack...")
                     except (ImportError, ModuleNotFoundError):
@@ -275,7 +284,9 @@ class Tracer(AgenticTracing):
                     
                     # AutoGen
                     try:
-                        from openinference.instrumentation.autogen import AutogenInstrumentor
+                        from openinference.instrumentation.autogen import (
+                            AutogenInstrumentor,
+                        )
                         instrumentors.append((AutogenInstrumentor, []))
                         logger.info("Instrumenting AutoGen...")
                     except (ImportError, ModuleNotFoundError):
@@ -283,7 +294,9 @@ class Tracer(AgenticTracing):
                     
                     # Smolagents
                     try:
-                        from openinference.instrumentation.smolagents import SmolagentsInstrumentor
+                        from openinference.instrumentation.smolagents import (
+                            SmolagentsInstrumentor,
+                        )
                         instrumentors.append((SmolagentsInstrumentor, []))
                         logger.info("Instrumenting Smolagents...")
                     except (ImportError, ModuleNotFoundError):
@@ -291,7 +304,9 @@ class Tracer(AgenticTracing):
 
                     # OpenAI Agents
                     try:
-                        from openinference.instrumentation.openai_agents import OpenAIAgentsInstrumentor
+                        from openinference.instrumentation.openai_agents import (
+                            OpenAIAgentsInstrumentor,
+                        )
                         instrumentors.append((OpenAIAgentsInstrumentor, []))
                         logger.info("Instrumenting OpenAI Agents...")
                     except (ImportError, ModuleNotFoundError):
@@ -309,16 +324,22 @@ class Tracer(AgenticTracing):
             
             # Handle specific framework instrumentation
             elif tracer_type == "agentic/llamaindex" or tracer_type == "llamaindex":
-                from openinference.instrumentation.llama_index import LlamaIndexInstrumentor
+                from openinference.instrumentation.llama_index import (
+                    LlamaIndexInstrumentor,
+                )
                 instrumentors += [(LlamaIndexInstrumentor, [])] 
 
             elif tracer_type == "agentic/langchain" or tracer_type == "agentic/langgraph" or tracer_type == "langchain":
-                from openinference.instrumentation.langchain import LangChainInstrumentor
+                from openinference.instrumentation.langchain import (
+                    LangChainInstrumentor,
+                )
                 instrumentors += [(LangChainInstrumentor, [])]
             
             elif tracer_type == "agentic/crewai":
                 from openinference.instrumentation.crewai import CrewAIInstrumentor
-                from openinference.instrumentation.langchain import LangChainInstrumentor
+                from openinference.instrumentation.langchain import (
+                    LangChainInstrumentor,
+                )
                 instrumentors += [(CrewAIInstrumentor, []), (LangChainInstrumentor, [])]
             
             elif tracer_type == "agentic/haystack":
@@ -330,11 +351,15 @@ class Tracer(AgenticTracing):
                 instrumentors += [(AutogenInstrumentor, [])]
             
             elif tracer_type == "agentic/smolagents":
-                from openinference.instrumentation.smolagents import SmolagentsInstrumentor
+                from openinference.instrumentation.smolagents import (
+                    SmolagentsInstrumentor,
+                )
                 instrumentors += [(SmolagentsInstrumentor, [])]
 
             elif tracer_type == "agentic/openai_agents":
-                from openinference.instrumentation.openai_agents import OpenAIAgentsInstrumentor
+                from openinference.instrumentation.openai_agents import (
+                    OpenAIAgentsInstrumentor,
+                )
                 instrumentors += [(OpenAIAgentsInstrumentor, [])] 
             
             else:
@@ -800,7 +825,10 @@ class Tracer(AgenticTracing):
         """
         from opentelemetry.sdk import trace as trace_sdk
         from opentelemetry.sdk.trace.export import SimpleSpanProcessor
-        from ragaai_catalyst.tracers.exporters.dynamic_trace_exporter import DynamicTraceExporter
+
+        from ragaai_catalyst.tracers.exporters.dynamic_trace_exporter import (
+            DynamicTraceExporter,
+        )
         
         # Get the code_files
         self.file_tracker.trace_main_file()

{ragaai_catalyst-2.2.1.dist-info → ragaai_catalyst-2.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.2.1
+Version: 2.2.2
 Summary: RAGA AI CATALYST
 Author-email: Kiran Scaria <kiran.scaria@raga.ai>, Kedar Gaikwad <kedar.gaikwad@raga.ai>, Dushyant Mahajan <dushyant.mahajan@raga.ai>, Siddhartha Kosti <siddhartha.kosti@raga.ai>, Ritika Goel <ritika.goel@raga.ai>, Vijay Chaurasia <vijay.chaurasia@raga.ai>, Tushar Kumar <tushar.kumar@raga.ai>, Rishabh Pandey <rishabh.pandey@raga.ai>, Jyotsana C G <jyotsana@raga.ai>
 Requires-Python: <=3.13.2,>=3.10

{ragaai_catalyst-2.2.1.dist-info → ragaai_catalyst-2.2.2.dist-info}/RECORD

@@ -8,7 +8,7 @@ ragaai_catalyst/guardrails_manager.py,sha256=_VrARJ1udmCF8TklNKy7XTQUaM8ATDhTOAG
 ragaai_catalyst/internal_api_completion.py,sha256=DdICI5yfEudiOAIC8L4oxH0Qz7kX-BZCdo9IWsi2gNo,2965
 ragaai_catalyst/prompt_manager.py,sha256=W8ypramzOprrJ7-22d5vkBXIuIQ8v9XAzKDGxKsTK28,16550
 ragaai_catalyst/proxy_call.py,sha256=CHxldeceZUaLU-to_hs_Kf1z_b2vHMssLS_cOBedu78,5499
-ragaai_catalyst/ragaai_catalyst.py,sha256=W_heG_ab650rzQkAhSOoxaNy3JKHAIXNvivxtOPpyeg,24590
+ragaai_catalyst/ragaai_catalyst.py,sha256=H7nhKuPi3pJH3QzO2dWshZ6FXgyZ5YRNOcB7MkXPRes,25413
 ragaai_catalyst/redteaming_old.py,sha256=W2d89Ok8W-C8g7TBM3fDIFLof3q9FuYSr0jcryH2XQo,7097
 ragaai_catalyst/synthetic_data_generation.py,sha256=7lIWa3nwgW2-FlJrDaGxTN6OE4-dbbhLtKNOBQufhho,37952
 ragaai_catalyst/utils.py,sha256=TlhEFwLyRU690HvANbyoRycR3nQ67lxVUQoUOfTPYQ0,3772
@@ -31,7 +31,7 @@ ragaai_catalyst/tracers/distributed.py,sha256=MwlBwIxCAng-OI-7Ove_rkE1mTLeuW4Jw-
 ragaai_catalyst/tracers/langchain_callback.py,sha256=CB75zzG3-DkYTELj0vI1MOHQTY0MuQJfoHIXz9Cl8S8,34568
 ragaai_catalyst/tracers/llamaindex_callback.py,sha256=ZY0BJrrlz-P9Mg2dX-ZkVKG3gSvzwqBtk7JL_05MiYA,14028
 ragaai_catalyst/tracers/llamaindex_instrumentation.py,sha256=Ys_jLkvVqo12bKgXDmkp4TxJu9HkBATrFE8cIcTYxWw,14329
-ragaai_catalyst/tracers/tracer.py,sha256=hdQK3-zV14lBqz8B6gLrMLNtfV34BmJ4-7eiNeLABq8,41931
+ragaai_catalyst/tracers/tracer.py,sha256=ngHe17GDSjHThrZO9uPr8yH-qYvnysPQrS7B0Aq_hJU,42071
 ragaai_catalyst/tracers/upload_traces.py,sha256=w1clGGfdOMpStUJX40NAlxe6dcFdN4pwcezyII0bGYA,6994
 ragaai_catalyst/tracers/agentic_tracing/README.md,sha256=X4QwLb7-Jg7GQMIXj-SerZIgDETfw-7VgYlczOR8ZeQ,4508
 ragaai_catalyst/tracers/agentic_tracing/__init__.py,sha256=yf6SKvOPSpH-9LiKaoLKXwqj5sez8F_5wkOb91yp0oE,260
@@ -54,9 +54,9 @@ ragaai_catalyst/tracers/agentic_tracing/tracers/network_tracer.py,sha256=m8CxYkl
 ragaai_catalyst/tracers/agentic_tracing/tracers/tool_tracer.py,sha256=xxrliKPfdfbIZRZqMnUewsaTD8_Hv0dbuoBivNZGD4U,21674
 ragaai_catalyst/tracers/agentic_tracing/tracers/user_interaction_tracer.py,sha256=bhSUhNQCuJXKjgJAXhjKEYjnHMpYN90FSZdR84fNIKU,4614
 ragaai_catalyst/tracers/agentic_tracing/upload/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ragaai_catalyst/tracers/agentic_tracing/upload/trace_uploader.py,sha256=FqHjvFdq5Pqp8iZcLTwXaxnMmm8ATec3fLdDuzLITFs,13019
-ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py,sha256=sdeUz2zYTHCNh82K6-I1RmQw7jsnGzJVe1VVglKcb_4,8082
-ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py,sha256=eNjKt8xC2IIs_mC9WdLaP_vUxWpYHVi2OXgJyn2oSx8,7443
+ragaai_catalyst/tracers/agentic_tracing/upload/trace_uploader.py,sha256=yA2_4fO0rWP0O4mrqcPGt_aZiB4gluVS5nCk6Q6k1y8,12946
+ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py,sha256=qcii8NMZGJvmv19AokjCH2egGIsALLWK9JqHtgkFB_0,10447
+ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py,sha256=XWRqVU_4kEgpPpx-TKmfgFFI5mxf_uO54yNhD5ogYNk,10170
 ragaai_catalyst/tracers/agentic_tracing/upload/upload_local_metric.py,sha256=m1O8lKpxKwtHofXLW3fTHX5yfqDW5GxoveARlg5cTw4,2571
 ragaai_catalyst/tracers/agentic_tracing/utils/__init__.py,sha256=XdB3X_ufe4RVvGorxSqAiB9dYv4UD7Hvvuw3bsDUppY,60
 ragaai_catalyst/tracers/agentic_tracing/utils/api_utils.py,sha256=JyNCbfpW-w4O9CjtemTqmor2Rh1WGpQwhRaDSRmBxw8,689
@@ -88,8 +88,8 @@ ragaai_catalyst/tracers/utils/rag_extraction_logic_final.py,sha256=3ygkRT__lLDRf
 ragaai_catalyst/tracers/utils/rag_trace_json_converter.py,sha256=54IEZO-YRjUAahV5nw8KClXqTF1LhfDry_TsZ4KGow4,20467
 ragaai_catalyst/tracers/utils/trace_json_converter.py,sha256=YllXhkfyYOuQ0rsX9VqiUjMUeztDMs8TFMudIPkWvrY,10191
 ragaai_catalyst/tracers/utils/utils.py,sha256=L19LQGc8h08FFhyptBtixIHGG_e-VdSPsKs7JNaXnGE,2378
-ragaai_catalyst-2.2.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.2.1.dist-info/METADATA,sha256=GjsoMgBgsiwIVyWzWkz_k8y08gnD2TsdvnGJbTM7its,17677
-ragaai_catalyst-2.2.1.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-ragaai_catalyst-2.2.1.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.2.1.dist-info/RECORD,,
+ragaai_catalyst-2.2.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.2.2.dist-info/METADATA,sha256=2XcrTsdS1a_QyQyy5qa0dnxSBdKudwn62leYQNx0Udo,17677
+ragaai_catalyst-2.2.2.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+ragaai_catalyst-2.2.2.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.2.2.dist-info/RECORD,,

{ragaai_catalyst-2.2.1.dist-info → ragaai_catalyst-2.2.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.7.1)
+Generator: setuptools (80.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any