ragaai-catalyst 2.1.7b1__tar.gz → 2.1.7.1__tar.gz

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/.github/workflows/ci.yml

@@ -59,11 +59,13 @@ jobs:
           if [ "$RUNNER_OS" == "Windows" ]; then
             python -m pip install --upgrade pip
             pip install pytest pytest-cov
+            pip install -r tests_requirements.txt
             pip install -e ".[dev]"
           else
             curl -LsSf https://astral.sh/uv/install.sh | sh
             export PATH="$HOME/.cargo/bin:$PATH"
             uv pip install --system pytest pytest-cov
+            uv pip install --system -r tests_requirements.txt
             uv pip install --system -e ".[dev]"
           fi
       - name: Test with pytest

{ragaai_catalyst-2.1.7b1/ragaai_catalyst.egg-info → ragaai_catalyst-2.1.7.1}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.1.7b1
+Version: 2.1.7.1
 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>
-Requires-Python: <3.13,>=3.10
+Requires-Python: <=3.13.2,>=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: aiohttp>=3.10.2
@@ -45,7 +45,7 @@ Requires-Dist: opentelemetry-sdk
 Requires-Dist: opentelemetry-exporter-otlp
 Requires-Dist: opentelemetry-proto>=1.12.0
 Provides-Extra: dev
-Requires-Dist: pytest; extra == "dev"
+Requires-Dist: pytest>=8.3.5; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"
 Requires-Dist: black; extra == "dev"
 Requires-Dist: isort; extra == "dev"

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/examples/crewai/scifi_writer/scifi_writer.py

@@ -10,14 +10,14 @@ from ragaai_catalyst.tracers import Tracer
 load_dotenv()
 
 catalyst = RagaAICatalyst(
-    access_key=os.getenv('CATALYST_ACCESS_KEY'), 
-    secret_key=os.getenv('CATALYST_SECRET_KEY'), 
-    base_url=os.getenv('CATALYST_BASE_URL')
+    access_key=os.getenv('RAGAAI_CATALYST_ACCESS_KEY'), 
+    secret_key=os.getenv('RAGAAI_CATALYST_SECRET_KEY'), 
+    base_url=os.getenv('RAGAAI_CATALYST_BASE_URL')
 )
 
 tracer = Tracer(
-    project_name=os.getenv('PROJECT_NAME'),
-    dataset_name=os.getenv('DATASET_NAME'),
+    project_name=os.getenv('RAGAAI_PROJECT_NAME'),
+    dataset_name=os.getenv('RAGAAI_DATASET_NAME'),
     tracer_type="agentic/crewai",
 )
 init_tracing(catalyst=catalyst, tracer=tracer)

{ragaai_catalyst-2.1.7b1/tests → ragaai_catalyst-2.1.7.1}/examples/custom_agents/travel_agent/config.py

@@ -1,7 +1,8 @@
 import sys 
 import os
 from dotenv import load_dotenv
-sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../../../..')))
+load_dotenv()
+
 from ragaai_catalyst import RagaAICatalyst, init_tracing
 from ragaai_catalyst.tracers import Tracer
 import uuid

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/examples/haystack/news_fetching/news_fetching.py

@@ -21,14 +21,14 @@ load_dotenv()
 
 # Setup Raga AI Catalyst for enhanced monitoring and tracing
 catalyst = RagaAICatalyst(
-    access_key=os.getenv('CATALYST_ACCESS_KEY'),
-    secret_key=os.getenv('CATALYST_SECRET_KEY'),
-    base_url=os.getenv('CATALYST_BASE_URL')
+    access_key=os.getenv('RAGAAI_CATALYST_ACCESS_KEY'),
+    secret_key=os.getenv('RAGAAI_CATALYST_SECRET_KEY'),
+    base_url=os.getenv('RAGAAI_CATALYST_BASE_URL')
 )
 
 tracer = Tracer(
-    project_name=os.getenv('PROJECT_NAME'),
-    dataset_name=os.getenv('DATASET_NAME'),
+    project_name=os.getenv('RAGAAI_PROJECT_NAME'),
+    dataset_name=os.getenv('RAGAAI_DATASET_NAME'),
     tracer_type="agentic/haystack",
 )
 

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/examples/langchain/medical_rag/diagnosis_agent.py

@@ -18,20 +18,21 @@ from dotenv import load_dotenv
 load_dotenv()
 
 catalyst = RagaAICatalyst(
-    access_key=os.getenv('CATALYST_ACCESS_KEY'), 
-    secret_key=os.getenv('CATALYST_SECRET_KEY'), 
-    base_url=os.getenv('CATALYST_BASE_URL')
+    access_key=os.getenv('RAGAAI_CATALYST_ACCESS_KEY'), 
+    secret_key=os.getenv('RAGAAI_CATALYST_SECRET_KEY'), 
+    base_url=os.getenv('RAGAAI_CATALYST_BASE_URL')
 )
 tracer = Tracer(
-    project_name=os.environ['PROJECT_NAME'],
-    dataset_name=os.environ['DATASET_NAME'],
+    project_name=os.environ['RAGAAI_PROJECT_NAME'],
+    dataset_name=os.environ['RAGAAI_DATASET_NAME'],
     tracer_type="agentic/langchain",
 )
 
 init_tracing(catalyst=catalyst, tracer=tracer)
 
-MEDICAL_TEXTS_DIR = "data/medical_texts"
-SYMPTOM_MAP_CSV = "data/symptom_disease_map.csv"
+DIR_PATH = os.path.dirname(os.path.abspath(__file__))
+MEDICAL_TEXTS_DIR = os.path.join(DIR_PATH, "data", "medical_texts")
+SYMPTOM_MAP_CSV = os.path.join(DIR_PATH, "data", "symptom_disease_map.csv")
 EMBEDDINGS_MODEL = "all-MiniLM-L6-v2"
 MODEL_TYPE = "openai"
 

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/examples/langgraph/personal_research_assistant/research_assistant.py

@@ -20,14 +20,14 @@ from ragaai_catalyst.tracers import Tracer
 def initialize_catalyst():
     """Initialize RagaAI Catalyst using environment credentials."""
     catalyst = RagaAICatalyst(
-        access_key=os.getenv('CATALYST_ACCESS_KEY'), 
-        secret_key=os.getenv('CATALYST_SECRET_KEY'), 
-        base_url=os.getenv('CATALYST_BASE_URL')
+        access_key=os.getenv('RAGAAI_CATALYST_ACCESS_KEY'), 
+        secret_key=os.getenv('RAGAAI_CATALYST_SECRET_KEY'), 
+        base_url=os.getenv('RAGAAI_CATALYST_BASE_URL')
     )
     
     tracer = Tracer(
-        project_name=os.environ['PROJECT_NAME'],
-        dataset_name=os.environ['DATASET_NAME'],
+        project_name=os.environ['RAGAAI_PROJECT_NAME'],
+        dataset_name=os.environ['RAGAAI_DATASET_NAME'],
         tracer_type="agentic/langgraph",
     )
     

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/examples/llamaindex_examples/legal_research_rag/legal_rag.py

@@ -17,14 +17,14 @@ load_dotenv()
 
 
 catalyst = RagaAICatalyst(
-    access_key=os.getenv('CATALYST_ACCESS_KEY'), 
-    secret_key=os.getenv('CATALYST_SECRET_KEY'), 
-    base_url=os.getenv('CATALYST_BASE_URL')
+    access_key=os.getenv('RAGAAI_CATALYST_ACCESS_KEY'), 
+    secret_key=os.getenv('RAGAAI_CATALYST_SECRET_KEY'), 
+    base_url=os.getenv('RAGAAI_CATALYST_BASE_URL')
 )
 # Initialize tracer
 tracer = Tracer(
-    project_name=os.getenv('PROJECT_NAME'),
-    dataset_name=os.getenv('DATASET_NAME'),
+    project_name=os.getenv('RAGAAI_PROJECT_NAME'),
+    dataset_name=os.getenv('RAGAAI_DATASET_NAME'),
     tracer_type="agentic/llamaindex",
 )
 

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/examples/smolagents/most_upvoted_paper/most_upvoted_paper.py

@@ -26,14 +26,14 @@ sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../.
 from ragaai_catalyst import RagaAICatalyst, Tracer, init_tracing
 
 catalyst = RagaAICatalyst(
-    access_key=os.getenv('CATALYST_ACCESS_KEY'),
-    secret_key=os.getenv('CATALYST_SECRET_KEY'),
-    base_url=os.getenv('CATALYST_BASE_URL'),
+    access_key=os.getenv('RAGAAI_CATALYST_ACCESS_KEY'),
+    secret_key=os.getenv('RAGAAI_CATALYST_SECRET_KEY'),
+    base_url=os.getenv('RAGAAI_CATALYST_BASE_URL'),
 )
 
 tracer = Tracer(
-    project_name=os.getenv('PROJECT_NAME'),
-    dataset_name=os.getenv('DATASET_NAME'),
+    project_name=os.getenv('RAGAAI_PROJECT_NAME'),
+    dataset_name=os.getenv('RAGAAI_DATASET_NAME'),
     tracer_type="agentic/smolagents",
 )
 

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/pyproject.toml

@@ -6,9 +6,9 @@ build-backend = "setuptools.build_meta"
 name = "ragaai_catalyst"
 description = "RAGA AI CATALYST"
 readme = "README.md"
-requires-python = ">=3.10,<3.13"
+requires-python = ">=3.10,<=3.13.2"
 # license = {file = "LICENSE"}
-version = "2.1.7.beta.1"
+version = "2.1.7.1"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},
@@ -61,7 +61,14 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
-dev = ["pytest", "pytest-cov", "black", "isort", "mypy", "flake8"]
+dev = [
+    "pytest>=8.3.5",
+    "pytest-cov", 
+    "black", 
+    "isort", 
+    "mypy", 
+    "flake8"
+]
 
 [tool.setuptools]
 packages = ["ragaai_catalyst"]

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/ragaai_catalyst/tracers/exporters/dynamic_trace_exporter.py

@@ -14,7 +14,7 @@ class DynamicTraceExporter(SpanExporter):
     certain properties to be updated dynamically during execution.
     """
     
-    def __init__(self, tracer_type, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost, timeout=120, post_processor = None, max_upload_workers = 30):
+    def __init__(self, tracer_type, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost, timeout=120, post_processor = None, max_upload_workers = 30,user_context = None, external_id=None):
         """
         Initialize the DynamicTraceExporter.
         
@@ -40,7 +40,9 @@ class DynamicTraceExporter(SpanExporter):
             custom_model_cost=custom_model_cost,
             timeout=timeout,
             post_processor= post_processor,
-            max_upload_workers = max_upload_workers
+            max_upload_workers = max_upload_workers,
+            user_context = user_context,
+            external_id=external_id
         )
         
         # Store the initial values
@@ -53,6 +55,7 @@ class DynamicTraceExporter(SpanExporter):
         self._custom_model_cost = custom_model_cost
         self._post_processor = post_processor
         self._max_upload_workers = max_upload_workers
+        self._user_context = user_context
 
     
     def export(self, spans):
@@ -111,6 +114,7 @@ class DynamicTraceExporter(SpanExporter):
         self._exporter.custom_model_cost = self._custom_model_cost
         self._exporter.post_processor = self._post_processor
         self._exporter.max_upload_workers = self._max_upload_workers
+        self._exporter.user_context = self._user_context
     
     # Getter and setter methods for dynamic properties
     
@@ -177,3 +181,12 @@ class DynamicTraceExporter(SpanExporter):
     @max_upload_workers.setter
     def max_upload_workers(self, value):
         self._max_upload_workers = value
+
+    @property
+    def user_context(self):
+        return self._user_context
+    
+    @user_context.setter
+    def user_context(self, value):
+        self._user_context = value
+

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/ragaai_catalyst/tracers/exporters/ragaai_trace_exporter.py

@@ -26,7 +26,7 @@ logging_level = (
 
 
 class RAGATraceExporter(SpanExporter):
-    def __init__(self, tracer_type, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost, timeout=120, post_processor = None, max_upload_workers = 30):
+    def __init__(self, tracer_type, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost, timeout=120, post_processor = None, max_upload_workers = 30,user_context = None, external_id=None):
         self.trace_spans = dict()
         self.tmp_dir = tempfile.gettempdir()
         self.tracer_type = tracer_type
@@ -41,6 +41,8 @@ class RAGATraceExporter(SpanExporter):
         self.timeout = timeout
         self.post_processor = post_processor
         self.max_upload_workers = max_upload_workers
+        self.user_context = user_context
+        self.external_id = external_id
 
     def export(self, spans):
         for span in spans:
@@ -86,12 +88,33 @@ class RAGATraceExporter(SpanExporter):
                 ragaai_trace_details = self.prepare_trace(spans, trace_id)
         except Exception as e:
             print(f"Error converting trace {trace_id}: {e}")
+            return  # Exit early if conversion fails
+            
+        # Check if trace details are None (conversion failed)
+        if ragaai_trace_details is None:
+            logger.error(f"Cannot upload trace {trace_id}: conversion failed and returned None")
+            return  # Exit early if conversion failed
+            
         # Upload the trace if upload_trace function is provided
         try:
             if self.post_processor!=None:
                 ragaai_trace_details['trace_file_path'] = self.post_processor(ragaai_trace_details['trace_file_path'])
             if self.tracer_type == "langchain":
-                asyncio.run(self.upload_rag_trace(ragaai_trace_details, additional_metadata, trace_id))
+                # Check if we're already in an event loop
+                try:
+                    loop = asyncio.get_event_loop()
+                    if loop.is_running():
+                        # We're in a running event loop (like in Colab/Jupyter)
+                        # Create a future and run the coroutine
+                        future = asyncio.ensure_future(self.upload_rag_trace(ragaai_trace_details, additional_metadata, trace_id))
+                        # We don't wait for it to complete as this would block the event loop
+                        logger.info(f"Scheduled async upload for trace {trace_id} in existing event loop")
+                    else:
+                        # No running event loop, use asyncio.run()
+                        asyncio.run(self.upload_rag_trace(ragaai_trace_details, additional_metadata, trace_id))
+                except RuntimeError:
+                    # No event loop exists, create one
+                    asyncio.run(self.upload_rag_trace(ragaai_trace_details, additional_metadata, trace_id))
             else:
                 self.upload_trace(ragaai_trace_details, trace_id)
         except Exception as e:
@@ -185,10 +208,12 @@ class RAGATraceExporter(SpanExporter):
     
     async def upload_rag_trace(self, ragaai_trace, additional_metadata, trace_id):
         try:
+            ragaai_trace[0]['external_id'] = self.external_id
             trace_file_path = os.path.join(self.tmp_dir, f"{trace_id}.json")
             with open(trace_file_path, 'w') as f:
                 json.dump(ragaai_trace, f, indent=2)
-            
+            logger.info(f"Trace file saved at {trace_file_path}")
+
             # Create a ThreadPoolExecutor with max_workers=30
             with concurrent.futures.ThreadPoolExecutor(max_workers=self.max_upload_workers) as executor:
                 # Create a partial function with all the necessary arguments
@@ -233,7 +258,7 @@ class RAGATraceExporter(SpanExporter):
     
     def prepare_rag_trace(self, spans, trace_id):
         try:            
-            ragaai_trace, additional_metadata = rag_trace_json_converter(spans, self.custom_model_cost, trace_id, self.user_details, self.tracer_type)
+            ragaai_trace, additional_metadata = rag_trace_json_converter(spans, self.custom_model_cost, trace_id, self.user_details, self.tracer_type,self.user_context)
             ragaai_trace["metadata"]["recorded_on"] = datetime.datetime.now().astimezone().isoformat()
             ragaai_trace["metadata"]["log_source"] = "langchain_tracer"
 

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/ragaai_catalyst/tracers/tracer.py

@@ -24,6 +24,7 @@ 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 ragaai_catalyst import RagaAICatalyst
 from ragaai_catalyst.tracers.agentic_tracing import AgenticTracing
 from ragaai_catalyst.tracers.agentic_tracing.tracers.llm_tracer import LLMTracerMixin
@@ -59,7 +60,8 @@ class Tracer(AgenticTracing):
         },
         interval_time=2,
         # auto_instrumentation=True/False  # to control automatic instrumentation of everything
-        max_upload_workers=30
+        max_upload_workers=30,
+        external_id=None
 
     ):
         """
@@ -143,6 +145,10 @@ class Tracer(AgenticTracing):
         self.file_tracker = TrackName()
         self.post_processor = None
         self.max_upload_workers = max_upload_workers
+        self.user_details = self._pass_user_data()
+        self.update_llm_cost = update_llm_cost
+        self.auto_instrumentation = auto_instrumentation
+        self.external_id = external_id
         
         try:
             response = requests.get(
@@ -460,6 +466,30 @@ class Tracer(AgenticTracing):
             self.dynamic_exporter._exporter.post_processor = post_processor_func
             self.dynamic_exporter._post_processor = post_processor_func
         logger.info("Registered post process as: "+str(post_processor_func))
+
+    
+    def set_external_id(self, external_id):
+        current_params = {
+            'project_name': self.project_name,
+            'dataset_name': self.dataset_name,
+            'trace_name': self.trace_name,
+            'tracer_type': self.tracer_type,
+            'pipeline': self.pipeline,
+            'metadata': self.metadata,
+            'description': self.description,
+            'timeout': self.timeout,
+            'update_llm_cost': self.update_llm_cost,
+            'auto_instrumentation': self.auto_instrumentation,
+            'interval_time': self.interval_time,
+            'max_upload_workers': self.max_upload_workers
+        }
+
+        # Reinitialize self with new external_id and stored parameters
+        self.__init__(
+            external_id=external_id,
+            **current_params
+        )
+
     
 
     def set_dataset_name(self, dataset_name):
@@ -485,15 +515,20 @@ class Tracer(AgenticTracing):
             # Also update the user_details in the dynamic exporter
             self.dynamic_exporter.user_details = self.user_details
         else:
-            # Store current parameters
             current_params = {
-                'project_name': self.project_name,
-                'tracer_type': self.tracer_type,
-                'pipeline': self.pipeline,
-                'metadata': self.metadata,
-                'description': self.description,
-                'timeout': self.timeout
-            }
+            'project_name': self.project_name,
+            'trace_name': self.trace_name,
+            'tracer_type': self.tracer_type,
+            'pipeline': self.pipeline,
+            'metadata': self.metadata,
+            'description': self.description,
+            'timeout': self.timeout,
+            'update_llm_cost': self.update_llm_cost,
+            'auto_instrumentation': self.auto_instrumentation,
+            'interval_time': self.interval_time,
+            'max_upload_workers': self.max_upload_workers,
+            'external_id': self.external_id
+        }
             
             # Reinitialize self with new dataset_name and stored parameters
             self.__init__(
@@ -503,7 +538,7 @@ class Tracer(AgenticTracing):
 
     def _improve_metadata(self, metadata, tracer_type):
         if metadata is None:
-            metadata = {}
+            metadata = {"metadata": {}}
         metadata.setdefault("log_source", f"{tracer_type}_tracer")
         metadata.setdefault("recorded_on", str(datetime.datetime.now()))
         return metadata
@@ -766,7 +801,9 @@ class Tracer(AgenticTracing):
             custom_model_cost=self.model_custom_cost,
             timeout = self.timeout,
             post_processor= self.post_processor,
-            max_upload_workers = self.max_upload_workers
+            max_upload_workers = self.max_upload_workers,
+            user_context = self.user_context,
+            external_id=self.external_id
         )
         
         # Set up tracer provider
@@ -775,7 +812,15 @@ class Tracer(AgenticTracing):
         
         # Instrument all specified instrumentors
         for instrumentor_class, args in instrumentors:
-            instrumentor_class().instrument(tracer_provider=tracer_provider, *args)
+            # Create an instance of the instrumentor
+            instrumentor = instrumentor_class()
+            
+            # Uninstrument only if it is already instrumented
+            if isinstance(instrumentor, LangChainInstrumentor) and instrumentor._is_instrumented_by_opentelemetry:
+                instrumentor.uninstrument()
+            
+            # Instrument with the provided tracer provider and arguments
+            instrumentor.instrument(tracer_provider=tracer_provider, *args)
             
     def update_file_list(self):
         """
@@ -810,6 +855,7 @@ class Tracer(AgenticTracing):
         
         # Convert string context to string if needed
         if isinstance(context, str):
+            self.dynamic_exporter.user_context = context
             self.user_context = context
         else:
             raise TypeError("context must be a string")

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/ragaai_catalyst/tracers/utils/rag_trace_json_converter.py

@@ -11,9 +11,8 @@ logging_level = (
     logger.setLevel(logging.DEBUG) if os.getenv("DEBUG") == "1" else logging.INFO
 )
 
-def rag_trace_json_converter(input_trace, custom_model_cost, trace_id, user_details, tracer_type):
+def rag_trace_json_converter(input_trace, custom_model_cost, trace_id, user_details, tracer_type,user_context):
     trace_aggregate = {}
-    
     def get_prompt(input_trace):
         if tracer_type == "langchain":
             for span in input_trace:
@@ -46,7 +45,9 @@ def rag_trace_json_converter(input_trace, custom_model_cost, trace_id, user_deta
         return None
     
     def get_context(input_trace):
-        if tracer_type == "langchain":
+        if user_context.strip():
+            return user_context
+        elif tracer_type == "langchain":
             for span in input_trace:
                 if span["name"] == "VectorStoreRetriever":
                     return span["attributes"].get("retrieval.documents.1.document.content")
@@ -82,6 +83,7 @@ def rag_trace_json_converter(input_trace, custom_model_cost, trace_id, user_deta
     else:
         additional_metadata = get_additional_metadata(input_trace, custom_model_cost, model_cost)
     
+    trace_aggregate["metadata"] = user_details.get("trace_user_detail", {}).get("metadata")
     trace_aggregate["metadata"].update(additional_metadata)
     additional_metadata.pop("total_cost")
     additional_metadata.pop("total_latency")

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1/ragaai_catalyst.egg-info}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.1.7b1
+Version: 2.1.7.1
 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>
-Requires-Python: <3.13,>=3.10
+Requires-Python: <=3.13.2,>=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: aiohttp>=3.10.2
@@ -45,7 +45,7 @@ Requires-Dist: opentelemetry-sdk
 Requires-Dist: opentelemetry-exporter-otlp
 Requires-Dist: opentelemetry-proto>=1.12.0
 Provides-Extra: dev
-Requires-Dist: pytest; extra == "dev"
+Requires-Dist: pytest>=8.3.5; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"
 Requires-Dist: black; extra == "dev"
 Requires-Dist: isort; extra == "dev"

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/ragaai_catalyst.egg-info/SOURCES.txt

@@ -5,6 +5,7 @@ README.md
 pyproject.toml
 requirements.txt
 test_report_20250407_183101.txt
+tests_requirements.txt
 .github/PULL_REQUEST_TEMPLATE.md
 .github/ISSUE_TEMPLATE/bug_report.md
 .github/ISSUE_TEMPLATE/feature_request.md
@@ -26,7 +27,7 @@ docs/img/projects_new.png
 docs/img/trace_comp.png
 examples/all_llm_provider/all_llm_provider.py
 examples/all_llm_provider/config.py
-examples/all_llm_provider/test_llm_call_utils.py
+examples/all_llm_provider/run_all_llm_provider.py
 examples/crewai/scifi_writer/README.md
 examples/crewai/scifi_writer/requirements.txt
 examples/crewai/scifi_writer/sample.env
@@ -205,7 +206,6 @@ tests/test_catalyst/test_base_tracer_metrics.py
 tests/test_catalyst/test_dataset.py
 tests/test_catalyst/test_evaluation.py
 tests/test_catalyst/test_evaluation_metrics.py
-tests/test_catalyst/test_openinference_langchain.py
 tests/test_catalyst/test_prompt_manager.py
 tests/test_catalyst/test_synthetic_data_generation.py
 tests/test_catalyst/test_the_configuration.py

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/ragaai_catalyst.egg-info/requires.txt

@@ -38,7 +38,7 @@ opentelemetry-exporter-otlp
 opentelemetry-proto>=1.12.0
 
 [dev]
-pytest
+pytest>=8.3.5
 pytest-cov
 black
 isort

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1/tests}/examples/all_llm_provider/config.py

@@ -1,20 +1,23 @@
-from ragaai_catalyst import RagaAICatalyst, init_tracing
-from ragaai_catalyst.tracers import Tracer
 import sys
 import os
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../..')))
 
+from ragaai_catalyst import RagaAICatalyst, init_tracing
+from ragaai_catalyst.tracers import Tracer
+
+from dotenv import load_dotenv
+load_dotenv()
 
 def initialize_tracing():
     catalyst = RagaAICatalyst(
-        access_key=os.getenv("CATALYST_ACCESS_KEY"),
-        secret_key=os.getenv("CATALYST_SECRET_KEY"),
-        base_url=os.getenv("CATALYST_BASE_URL"),
+        access_key=os.getenv("RAGAAI_CATALYST_ACCESS_KEY"),
+        secret_key=os.getenv("RAGAAI_CATALYST_SECRET_KEY"),
+        base_url=os.getenv("RAGAAI_CATALYST_BASE_URL"),
     )
 
     tracer = Tracer(
-        project_name=os.getenv("PROJECT_NAME"),
-        dataset_name=os.getenv("DATASET_NAME"),
+        project_name='prompt_metric_dataset',#os.getenv("RAGAAI_PROJECT_NAME"),
+        dataset_name='pytest_dataset',#os.getenv("RAGAAI_DATASET_NAME"),
         tracer_type="Agentic",
     )
 

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/tests/examples/crewai/scifi_writer/scifi_writer.py

@@ -21,8 +21,8 @@ catalyst = RagaAICatalyst(
 )
 
 tracer = Tracer(
-    project_name=os.getenv('RAGAAI_PROJECT_NAME'),
-    dataset_name=os.getenv('RAGAAI_DATASET_NAME'),
+    project_name='prompt_metric_dataset',#os.getenv("RAGAAI_PROJECT_NAME"),
+    dataset_name='pytest_dataset',#os.getenv("RAGAAI_DATASET_NAME"),
     tracer_type="agentic/crewai",
 )
 init_tracing(catalyst=catalyst, tracer=tracer)

ragaai_catalyst-2.1.7.1/tests/examples/custom_agents/travel_agent/config.py

@@ -0,0 +1,24 @@
+import sys 
+import os
+from dotenv import load_dotenv
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../../../..')))
+from ragaai_catalyst import RagaAICatalyst, init_tracing
+from ragaai_catalyst.tracers import Tracer
+import uuid
+
+
+def initialize_tracing():
+    catalyst = RagaAICatalyst(
+        access_key=os.getenv("RAGAAI_CATALYST_ACCESS_KEY"),
+        secret_key=os.getenv("RAGAAI_CATALYST_SECRET_KEY"),
+        base_url=os.getenv("RAGAAI_CATALYST_BASE_URL"),
+    )
+
+    tracer = Tracer(
+        project_name='prompt_metric_dataset',#os.getenv("RAGAAI_PROJECT_NAME"),
+        dataset_name='pytest_dataset',#os.getenv("RAGAAI_DATASET_NAME"),
+        tracer_type="Agentic",
+    )
+
+    init_tracing(catalyst=catalyst, tracer=tracer)
+    return tracer

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/tests/examples/haystack/news_fetching/news_fetching.py

@@ -28,8 +28,8 @@ catalyst = RagaAICatalyst(
 )
 
 tracer = Tracer(
-    project_name=os.getenv('RAGAAI_PROJECT_NAME'),
-    dataset_name=os.getenv('RAGAAI_DATASET_NAME'),
+    project_name='prompt_metric_dataset',#os.getenv("RAGAAI_PROJECT_NAME"),
+    dataset_name='pytest_dataset',#os.getenv("RAGAAI_DATASET_NAME"),
     tracer_type="agentic/haystack",
 )
 

{ragaai_catalyst-2.1.7b1 → ragaai_catalyst-2.1.7.1}/tests/examples/langchain/medical_rag/diagnosis_agent.py

@@ -27,8 +27,8 @@ catalyst = RagaAICatalyst(
     base_url=os.getenv('RAGAAI_CATALYST_BASE_URL')
 )
 tracer = Tracer(
-    project_name=os.environ['RAGAAI_PROJECT_NAME'],
-    dataset_name=os.environ['RAGAAI_DATASET_NAME'],
+    project_name='prompt_metric_dataset',#os.getenv("RAGAAI_PROJECT_NAME"),
+    dataset_name='pytest_dataset',#os.getenv("RAGAAI_DATASET_NAME"),
     tracer_type="agentic/langchain",
 )