ragaai-catalyst 2.1.6b2__tar.gz → 2.1.6.1__tar.gz

{ragaai_catalyst-2.1.6b2 → ragaai_catalyst-2.1.6.1}/PKG-INFO

@@ -1,9 +1,9 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.1.6b2
+Version: 2.1.6.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.9
+Requires-Python: <3.13,>=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: aiohttp>=3.10.2
@@ -39,7 +39,7 @@ Requires-Dist: openinference-instrumentation-openai
 Requires-Dist: openinference-instrumentation-bedrock
 Requires-Dist: openinference-instrumentation-crewai
 Requires-Dist: openinference-instrumentation-haystack
-Requires-Dist: openinference-instrumentation-autogen
+Requires-Dist: openinference-instrumentation-openai-agents
 Requires-Dist: openinference-instrumentation-smolagents
 Requires-Dist: opentelemetry-sdk
 Requires-Dist: opentelemetry-exporter-otlp
@@ -51,6 +51,7 @@ Requires-Dist: black; extra == "dev"
 Requires-Dist: isort; extra == "dev"
 Requires-Dist: mypy; extra == "dev"
 Requires-Dist: flake8; extra == "dev"
+Dynamic: license-file
 
 # RagaAI Catalyst&nbsp; ![GitHub release (latest by date)](https://img.shields.io/github/v/release/raga-ai-hub/ragaai-catalyst) ![GitHub stars](https://img.shields.io/github/stars/raga-ai-hub/ragaai-catalyst?style=social)  ![Issues](https://img.shields.io/github/issues/raga-ai-hub/ragaai-catalyst) 
 

{ragaai_catalyst-2.1.6b2 → ragaai_catalyst-2.1.6.1}/examples/crewai/scifi_writer/scifi_writer.py

@@ -1,15 +1,11 @@
-import sys
-sys.path.append('.')
-
-from ragaai_catalyst import RagaAICatalyst, init_tracing
-from ragaai_catalyst.tracers import Tracer
-
 import os
 from dotenv import load_dotenv
 from crewai import Agent, Task, Crew, Process
 from crewai.tools import tool
 from typing import Any
 
+from ragaai_catalyst import RagaAICatalyst, init_tracing
+from ragaai_catalyst.tracers import Tracer
 
 load_dotenv()
 
@@ -98,4 +94,4 @@ try:
         print("\nGenerated Story Content:")
         print(file.read())
 except FileNotFoundError:
-    print("Story file not found. Check the writer agent's execution.")
+    print("Story file not found. Check the writer agent's execution.")

ragaai_catalyst-2.1.6.1/examples/openai_agents_sdk/README.md

@@ -0,0 +1,71 @@
+# Email Data Extraction with OpenAI Agents SDK
+
+This example demonstrates how to use the OpenAI Agents SDK with RagaAI Catalyst to extract structured information from emails.
+
+## Overview
+
+The application uses OpenAI's Agents SDK to parse unstructured email text and extract key information such as:
+- Email subject and sender details
+- Main discussion points
+- Meeting information (date, time, location)
+- Action items and tasks with assignees
+- Next steps
+
+The extracted data is structured using Pydantic models for easy manipulation and validation.
+
+## Requirements
+
+- Python 3.8+
+- OpenAI API key
+- RagaAI Catalyst credentials
+
+## Installation
+
+1. Clone the repository
+2. Install the required dependencies:
+```bash
+pip install -r requirements.txt
+```
+3. Copy [sample.env](cci:7://file:///Users/ragaai_user/work/ragaai-catalyst/examples/openai_agents_sdk/sample.env:0:0-0:0) to [.env](cci:7://file:///Users/ragaai_user/work/ragaai-catalyst/examples/openai_agents_sdk/sample.env:0:0-0:0) and fill in your API keys:
+```bash
+cp sample.env .env
+```
+
+## Environment Variables
+
+Configure the following environment variables in your [.env](cci:7://file:///Users/ragaai_user/work/ragaai-catalyst/examples/openai_agents_sdk/sample.env:0:0-0:0) file:
+
+- `OPENAI_API_KEY`: Your OpenAI API key
+- `CATALYST_ACCESS_KEY`: Your RagaAI Catalyst access key
+- `CATALYST_SECRET_KEY`: Your RagaAI Catalyst secret key
+- `CATALYST_BASE_URL`: RagaAI Catalyst base URL
+- `PROJECT_NAME`: Name for your project in RagaAI Catalyst (default: 'email-extraction')
+- `DATASET_NAME`: Name for your dataset in RagaAI Catalyst (default: 'email-data')
+
+## Usage
+
+Run the example script:
+
+```bash
+python data_extraction_email.py
+```
+The script will:
+
+1. Initialize the RagaAI Catalyst client for tracing
+2. Set up an OpenAI Agent with appropriate instructions
+3. Process a sample email to extract structured data
+4. Display the extracted information
+
+## Customization
+
+You can modify the `sample_email` variable in the script to process different emails, or adapt the code to read emails from files or an API.
+
+The Pydantic models (`Person`, `Meeting`, `Task`, `EmailData`) can be extended to capture additional information as needed.
+
+## Integration with RagaAI Catalyst
+
+This example integrates with RagaAI Catalyst for tracing and monitoring agent interactions. The integration helps with:
+
+- Tracking agent performance
+- Debugging complex agent workflows
+- Collecting data for future improvements

ragaai_catalyst-2.1.6.1/examples/openai_agents_sdk/data_extraction_email.py

@@ -0,0 +1,189 @@
+import os
+import time
+from typing import List, Optional, Callable, Any
+from pydantic import BaseModel
+from dotenv import load_dotenv
+
+from agents import Agent, Runner, ModelSettings, set_tracing_export_api_key
+
+from ragaai_catalyst import RagaAICatalyst, init_tracing
+from ragaai_catalyst.tracers import Tracer
+
+load_dotenv()
+set_tracing_export_api_key(os.getenv('OPENAI_API_KEY'))
+
+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')
+    )
+    
+    tracer = Tracer(
+        project_name=os.environ.get('PROJECT_NAME', 'email-extraction'),
+        dataset_name=os.environ.get('DATASET_NAME', 'email-data'),
+        tracer_type="agentic/openai_agents",
+    )
+    
+    init_tracing(catalyst=catalyst, tracer=tracer)
+
+class Person(BaseModel):
+    """Person data model for email sender and recipients."""
+    name: str
+    role: Optional[str] = None
+    contact: Optional[str] = None
+
+class Meeting(BaseModel):
+    """Meeting data model for scheduled meetings in emails."""
+    date: str
+    time: str
+    location: Optional[str] = None
+    duration: Optional[str] = None
+
+class Task(BaseModel):
+    """Task data model for action items in emails."""
+    description: str
+    assignee: Optional[str] = None
+    deadline: Optional[str] = None
+    priority: Optional[str] = None
+
+class EmailData(BaseModel):
+    """Complete email data model with structured information."""
+    subject: str
+    sender: Person
+    recipients: List[Person]
+    main_points: List[str]
+    meetings: List[Meeting]
+    tasks: List[Task]
+    next_steps: Optional[str] = None
+
+def initialize_agent(agent_name: str, agent_instructions: str|Callable, handoff_description: Optional[str]=None, handoffs: List[Agent]=list(), model_name: str='gpt-4o', temperature: float=0.3, max_tokens: int=1000, output_type: Optional[type[Any]]=None):
+    """Initialize the OpenAI agent for email extraction."""
+    # Initialize the agent with appropriate configuration
+    # This could include model selection, temperature settings, etc.
+    model_settings = ModelSettings(
+        temperature=temperature,
+        max_tokens=max_tokens
+    )
+    agent = Agent(
+        name=agent_name,
+        instructions=agent_instructions,
+        handoff_description=handoff_description,
+        handoffs=handoffs,
+        model=model_name,
+        model_settings=model_settings, 
+        output_type=output_type
+    )
+    return agent
+
+email_extractor = initialize_agent(
+    agent_name="Email Extractor",
+    agent_instructions="You are an expert at extracting structured information from emails.",
+    model_name="gpt-4o",
+    temperature=0.2,
+    output_type=EmailData
+)
+
+async def extract_email_data(email_text: str) -> EmailData:
+    """
+    Extract structured data from an email using an OpenAI agent.
+    
+    Args:
+        email_text: The raw email text to process
+        
+    Returns:
+        EmailData object containing structured information from the email
+    """
+    runner = Runner()
+    extraction_prompt = f"Please extract information from this email:\n\n{email_text}"
+    result = await runner.run(
+        email_extractor,
+        extraction_prompt
+    )
+    return result.final_output
+
+sample_email = """
+From: Alex Johnson <alex.j@techcorp.com>
+To: Team Development <team-dev@techcorp.com>
+CC: Sarah Wong <sarah.w@techcorp.com>, Miguel Fernandez <miguel.f@techcorp.com>
+Subject: Project Phoenix Update and Next Steps
+
+Hi team,
+
+I wanted to follow up on yesterday's discussion about Project Phoenix and outline our next steps.
+
+Key points from our discussion:
+- The beta testing phase has shown promising results with 85% positive feedback
+- We're still facing some performance issues on mobile devices
+- The client has requested additional features for the dashboard
+
+Let's schedule a follow-up meeting this Friday, June 15th at 2:00 PM in Conference Room B. The meeting should last about 1.5 hours, and we'll need to prepare the updated project timeline.
+
+Action items:
+1. Sarah to address the mobile performance issues by June 20th (High priority)
+2. Miguel to create mock-ups for the new dashboard features by next Monday
+3. Everyone to review the beta testing feedback document and add comments by EOD tomorrow
+
+If you have any questions before Friday's meeting, feel free to reach out.
+
+Best regards,
+Alex Johnson
+Senior Project Manager
+(555) 123-4567
+"""
+
+def display_email_data(email_data: EmailData):
+    """
+    Display the extracted email data in a formatted way.
+    
+    Args:
+        email_data: The structured EmailData object to display
+    """
+    print(f"Subject: {email_data.subject}")
+    print(f"From: {email_data.sender.name} ({email_data.sender.role})")
+    
+    print("\nMain points:")
+    for point in email_data.main_points:
+        print(f"- {point}")
+    
+    print("\nMeetings:")
+    for meeting in email_data.meetings:
+        print(f"- {meeting.date} at {meeting.time}, Location: {meeting.location}")
+    
+    print("\nTasks:")
+    for task in email_data.tasks:
+        print(f"- {task.description}")
+        print(
+            f"  Assignee: {task.assignee}, Deadline: {task.deadline}, Priority: {task.priority}"
+        )
+    
+    if email_data.next_steps:
+        print(f"\nNext Steps: {email_data.next_steps}")
+
+async def process_email(email_text: str):
+    """
+    Process an email to extract structured data and display the results.
+    
+    Args:
+        email_text: The raw email text to process
+        
+    Returns:
+        The structured EmailData object
+    """
+    if os.getenv('CATALYST_ACCESS_KEY'):
+        initialize_catalyst()
+    
+    start_time = time.time()
+    email_data = await extract_email_data(email_text)
+    duration = time.time() - start_time
+    
+    print(f"Email processing completed in {duration:.2f} seconds")
+    display_email_data(email_data)
+    
+    return email_data
+
+if __name__ == "__main__":
+    import asyncio
+    
+    asyncio.run(process_email(sample_email))

ragaai_catalyst-2.1.6.1/examples/openai_agents_sdk/requirements.txt

@@ -0,0 +1,3 @@
+openai-agents
+python-dotenv
+ragaai_catalyst

ragaai_catalyst-2.1.6.1/examples/openai_agents_sdk/sample.env

@@ -0,0 +1,6 @@
+OPENAI_API_KEY=your_openai_api_key
+CATALYST_ACCESS_KEY=your_catalyst_access_key
+CATALYST_SECRET_KEY=your_catalyst_secret_key
+CATALYST_BASE_URL=your_catalyst_base_url
+PROJECT_NAME=your_project_name
+DATSET_NAME=your_dataset_name

{ragaai_catalyst-2.1.6b2 → ragaai_catalyst-2.1.6.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.9,<3.13"
+requires-python = ">=3.10,<3.13"
 # license = {file = "LICENSE"}
-version = "2.1.6.b2"
+version = "2.1.6.1"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},
@@ -28,8 +28,8 @@ dependencies = [
     "groq>=0.11.0",
     "pypdf>=5.3.1",
     "google-genai>=1.3.0",
-    "Markdown>=3.7",  
-    "litellm>=1.51.1",  
+    "Markdown>=3.7",
+    "litellm>=1.51.1",
     "tenacity==8.3.0",
     "tqdm>=4.66.5",
     "llama-index>=0.10.0",
@@ -53,11 +53,11 @@ dependencies = [
     "openinference-instrumentation-bedrock",
     "openinference-instrumentation-crewai",
     "openinference-instrumentation-haystack",
-    "openinference-instrumentation-autogen",
+    "openinference-instrumentation-openai-agents",
     "openinference-instrumentation-smolagents",
     "opentelemetry-sdk",
     "opentelemetry-exporter-otlp",
-    "opentelemetry-proto>=1.12.0"
+    "opentelemetry-proto>=1.12.0",
 ]
 
 [project.optional-dependencies]

{ragaai_catalyst-2.1.6b2 → ragaai_catalyst-2.1.6.1}/ragaai_catalyst/guard_executor.py

@@ -164,6 +164,7 @@ class GuardExecutor:
         return doc
 
     def execute_input_guardrails(self, prompt, prompt_params):
+        self.current_trace_id =None
         doc = self.set_variables(prompt,prompt_params)
         deployment_response = self.execute_deployment(self.input_deployment_id,doc)
         self.current_trace_id = deployment_response['data']['results'][0]['executionId']

{ragaai_catalyst-2.1.6b2 → ragaai_catalyst-2.1.6.1}/ragaai_catalyst/tracers/agentic_tracing/tracers/base.py

@@ -92,6 +92,7 @@ class BaseTracer:
         self._upload_tasks = []
         self._is_uploading = False
         self._upload_completed_callback = None
+        self.timeout = self.user_details.get("timeout", 120)
         
         ensure_uploader_running()
 
@@ -314,7 +315,8 @@ class BaseTracer:
                 project_id=self.project_id,
                 dataset_name=self.dataset_name,
                 user_details=self.user_details,
-                base_url=self.base_url
+                base_url=self.base_url,
+                timeout=self.timeout
             )
             
             # For backward compatibility

{ragaai_catalyst-2.1.6b2 → ragaai_catalyst-2.1.6.1}/ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py

@@ -48,7 +48,7 @@ from ragaai_catalyst.tracers.upload_traces import UploadTraces
 class AgenticTracing(
     BaseTracer, LLMTracerMixin, ToolTracerMixin, AgentTracerMixin, CustomTracerMixin
 ):
-    def __init__(self, user_detail, auto_instrumentation=None):
+    def __init__(self, user_detail, auto_instrumentation=None, timeout=120):
         # Initialize all parent classes
         self.user_interaction_tracer = UserInteractionTracer()
         LLMTracerMixin.__init__(self)
@@ -60,7 +60,7 @@ class AgenticTracing(
         self.project_id = user_detail["project_id"]
         self.trace_user_detail = user_detail["trace_user_detail"]
         self.base_url = f"{RagaAICatalyst.BASE_URL}"
-        self.timeout = 10
+        self.timeout = timeout
         
         # Add warning flag
         self._warning_shown = False

{ragaai_catalyst-2.1.6b2 → ragaai_catalyst-2.1.6.1}/ragaai_catalyst/tracers/agentic_tracing/upload/trace_uploader.py

@@ -77,7 +77,7 @@ def get_executor():
 
 def process_upload(task_id: str, filepath: str, hash_id: str, zip_path: str, 
                   project_name: str, project_id: str, dataset_name: str, 
-                  user_details: Dict[str, Any], base_url: str) -> Dict[str, Any]:
+                  user_details: Dict[str, Any], base_url: str, timeout=120) -> Dict[str, Any]:
     """
     Process a single upload task
     
@@ -147,7 +147,8 @@ def process_upload(task_id: str, filepath: str, hash_id: str, zip_path: str,
                     json_file_path=filepath,
                     dataset_name=dataset_name,
                     project_name=project_name,
-                    base_url=base_url
+                    base_url=base_url,
+                    timeout=timeout
                 )
                 logger.info(f"Trace metrics uploaded: {response}")
             except Exception as e:
@@ -167,6 +168,7 @@ def process_upload(task_id: str, filepath: str, hash_id: str, zip_path: str,
                     dataset_name=dataset_name,
                     user_detail=user_details,
                     base_url=base_url,   
+                    timeout=timeout
                 )
                 upload_traces.upload_agentic_traces()
                 logger.info("Agentic traces uploaded successfully")
@@ -185,7 +187,8 @@ def process_upload(task_id: str, filepath: str, hash_id: str, zip_path: str,
                     zip_path=zip_path,
                     project_name=project_name,
                     dataset_name=dataset_name,
-                    base_url=base_url
+                    base_url=base_url,
+                    timeout=timeout
                 )
                 logger.info(f"Code hash uploaded: {response}")
             except Exception as e:
@@ -215,7 +218,7 @@ def save_task_status(task_status: Dict[str, Any]):
     with open(status_path, "w") as f:
         json.dump(task_status, f, indent=2)
 
-def submit_upload_task(filepath, hash_id, zip_path, project_name, project_id, dataset_name, user_details, base_url):
+def submit_upload_task(filepath, hash_id, zip_path, project_name, project_id, dataset_name, user_details, base_url, timeout=120):
     """
     Submit a new upload task using futures.
     
@@ -259,7 +262,8 @@ def submit_upload_task(filepath, hash_id, zip_path, project_name, project_id, da
         project_id=project_id,
         dataset_name=dataset_name,
         user_details=user_details,
-        base_url=base_url
+        base_url=base_url,
+        timeout=timeout
     )
     
     # Store the future for later status checks

{ragaai_catalyst-2.1.6b2 → ragaai_catalyst-2.1.6.1}/ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py

@@ -24,7 +24,7 @@ class UploadAgenticTraces:
         self.dataset_name = dataset_name
         self.user_detail = user_detail
         self.base_url = base_url
-        self.timeout = 30
+        self.timeout = 120
 
 
     def _get_presigned_url(self):

{ragaai_catalyst-2.1.6b2 → ragaai_catalyst-2.1.6.1}/ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py

@@ -9,19 +9,19 @@ logger = logging.getLogger(__name__)
 from urllib.parse import urlparse, urlunparse
 import re
 
-def upload_code(hash_id, zip_path, project_name, dataset_name, base_url=None):
-    code_hashes_list = _fetch_dataset_code_hashes(project_name, dataset_name, base_url)
+def upload_code(hash_id, zip_path, project_name, dataset_name, base_url=None, timeout=120):
+    code_hashes_list = _fetch_dataset_code_hashes(project_name, dataset_name, base_url, timeout=timeout)
 
     if hash_id not in code_hashes_list:
-        presigned_url = _fetch_presigned_url(project_name, dataset_name, base_url)
-        _put_zip_presigned_url(project_name, presigned_url, zip_path)
+        presigned_url = _fetch_presigned_url(project_name, dataset_name, base_url, timeout=timeout)
+        _put_zip_presigned_url(project_name, presigned_url, zip_path, timeout=timeout)
 
-        response = _insert_code(dataset_name, hash_id, presigned_url, project_name, base_url)
+        response = _insert_code(dataset_name, hash_id, presigned_url, project_name, base_url, timeout=timeout)
         return response
     else:
         return "Code already exists"
 
-def _fetch_dataset_code_hashes(project_name, dataset_name, base_url=None):
+def _fetch_dataset_code_hashes(project_name, dataset_name, base_url=None, timeout=120):
     payload = {}
     headers = {
         "Authorization": f"Bearer {os.getenv('RAGAAI_CATALYST_TOKEN')}",
@@ -36,7 +36,7 @@ def _fetch_dataset_code_hashes(project_name, dataset_name, base_url=None):
                                     endpoint, 
                                     headers=headers, 
                                     data=payload,
-                                    timeout=99999)
+                                    timeout=timeout)
         elapsed_ms = (time.time() - start_time) * 1000
         logger.debug(
             f"API Call: [GET] {endpoint} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
@@ -66,7 +66,7 @@ def update_presigned_url(presigned_url, base_url):
     return presigned_url
 
 
-def _fetch_presigned_url(project_name, dataset_name, base_url=None):
+def _fetch_presigned_url(project_name, dataset_name, base_url=None, timeout=120):
     payload = json.dumps({
             "datasetName": dataset_name,
             "numFiles": 1,
@@ -87,7 +87,7 @@ def _fetch_presigned_url(project_name, dataset_name, base_url=None):
                                     endpoint, 
                                     headers=headers, 
                                     data=payload,
-                                    timeout=99999)
+                                    timeout=timeout)
         elapsed_ms = (time.time() - start_time) * 1000
         logger.debug(
             f"API Call: [GET] {endpoint} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
@@ -102,7 +102,7 @@ def _fetch_presigned_url(project_name, dataset_name, base_url=None):
         logger.error(f"Failed to list datasets: {e}")
         raise
 
-def _put_zip_presigned_url(project_name, presignedUrl, filename):
+def _put_zip_presigned_url(project_name, presignedUrl, filename, timeout=120):
     headers = {
             "X-Project-Name": project_name,
             "Content-Type": "application/zip",
@@ -119,14 +119,14 @@ def _put_zip_presigned_url(project_name, presignedUrl, filename):
                                 presignedUrl, 
                                 headers=headers, 
                                 data=payload,
-                                timeout=99999)
+                                timeout=timeout)
     elapsed_ms = (time.time() - start_time) * 1000
     logger.debug(
         f"API Call: [PUT] {presignedUrl} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
     if response.status_code != 200 or response.status_code != 201:
         return response, response.status_code
 
-def _insert_code(dataset_name, hash_id, presigned_url, project_name, base_url=None):
+def _insert_code(dataset_name, hash_id, presigned_url, project_name, base_url=None, timeout=120):
     payload = json.dumps({
         "datasetName": dataset_name,
         "codeHash": hash_id,
@@ -147,7 +147,7 @@ def _insert_code(dataset_name, hash_id, presigned_url, project_name, base_url=No
                                     endpoint, 
                                     headers=headers, 
                                     data=payload,
-                                    timeout=99999)
+                                    timeout=timeout)
         elapsed_ms = (time.time() - start_time) * 1000
         logger.debug(
             f"API Call: [POST] {endpoint} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")

{ragaai_catalyst-2.1.6b2 → ragaai_catalyst-2.1.6.1}/ragaai_catalyst/tracers/agentic_tracing/upload/upload_trace_metric.py

@@ -15,7 +15,7 @@ logging_level = (
 )
 
 
-def upload_trace_metric(json_file_path, dataset_name, project_name, base_url=None):
+def upload_trace_metric(json_file_path, dataset_name, project_name, base_url=None, timeout=120):
     try:
         with open(json_file_path, "r") as f:
             traces = json.load(f)
@@ -51,7 +51,7 @@ def upload_trace_metric(json_file_path, dataset_name, project_name, base_url=Non
                                     endpoint,
                                     headers=headers,
                                     data=payload,
-                                    timeout=10)
+                                    timeout=timeout)
         elapsed_ms = (time.time() - start_time) * 1000
         logger.debug(
             f"API Call: [POST] {endpoint} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")

{ragaai_catalyst-2.1.6b2 → ragaai_catalyst-2.1.6.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, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost):
+    def __init__(self, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost, timeout=120):
         """
         Initialize the DynamicTraceExporter.
         
@@ -33,7 +33,8 @@ class DynamicTraceExporter(SpanExporter):
             dataset_name=dataset_name,
             user_details=user_details,
             base_url=base_url,
-            custom_model_cost=custom_model_cost
+            custom_model_cost=custom_model_cost,
+            timeout=timeout
         )
         
         # Store the initial values

{ragaai_catalyst-2.1.6b2 → ragaai_catalyst-2.1.6.1}/ragaai_catalyst/tracers/exporters/ragaai_trace_exporter.py

@@ -19,7 +19,7 @@ logging_level = (
 
 
 class RAGATraceExporter(SpanExporter):
-    def __init__(self, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost):
+    def __init__(self, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost, timeout=120):
         self.trace_spans = dict()
         self.tmp_dir = tempfile.gettempdir()
         self.files_to_zip = files_to_zip
@@ -30,6 +30,7 @@ class RAGATraceExporter(SpanExporter):
         self.base_url = base_url
         self.custom_model_cost = custom_model_cost
         self.system_monitor = SystemMonitor(dataset_name)
+        self.timeout = timeout
 
     def export(self, spans):
         for span in spans:
@@ -122,7 +123,8 @@ class RAGATraceExporter(SpanExporter):
                 project_id=self.project_id,
                 dataset_name=self.dataset_name,
                 user_details=self.user_details,
-                base_url=self.base_url
+                base_url=self.base_url,
+                timeout=self.timeout
             )
 
         logger.info(f"Submitted upload task with ID: {self.upload_task_id}")