ragaai-catalyst 2.1b5__tar.gz → 2.1.1__tar.gz

{ragaai_catalyst-2.1b5 → ragaai_catalyst-2.1.1}/PKG-INFO

@@ -1,9 +1,9 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1b5
+Version: 2.1.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>
-Requires-Python: >=3.9
+Requires-Python: <3.13,>=3.9
 Description-Content-Type: text/markdown
 Requires-Dist: aiohttp>=3.10.2
 Requires-Dist: opentelemetry-api==1.25.0
@@ -19,7 +19,7 @@ Requires-Dist: opentelemetry-instrumentation-openai~=0.24.0
 Requires-Dist: langchain-core>=0.2.11
 Requires-Dist: langchain>=0.2.11
 Requires-Dist: openai>=1.57.0
-Requires-Dist: pandas>=2.1.1
+Requires-Dist: pandas
 Requires-Dist: groq>=0.11.0
 Requires-Dist: PyPDF2>=3.0.1
 Requires-Dist: google-generativeai>=0.8.2
@@ -27,7 +27,7 @@ Requires-Dist: Markdown>=3.7
 Requires-Dist: litellm==1.51.1
 Requires-Dist: tenacity==8.3.0
 Requires-Dist: tqdm>=4.66.5
-Requires-Dist: llama-index>=0.10.0
+Requires-Dist: llama-index<0.11.0,>=0.10.0
 Requires-Dist: pyopenssl>=24.2.1
 Requires-Dist: psutil~=6.0.0
 Requires-Dist: py-cpuinfo~=9.0.0

{ragaai_catalyst-2.1b5 → ragaai_catalyst-2.1.1}/ai_travel_agent.py

@@ -16,15 +16,15 @@ import openai
 from openai import AsyncOpenAI
 
 catalyst = RagaAICatalyst(
-    access_key="saLy6KmMVlfAzunuQGS9",
-    secret_key="lm39fd4KXffM6gzLjnY9G7QReffhH4RGZPursp3A",
-    base_url="http://52.172.168.127/api"
+    access_key="access_key",
+    secret_key="secret_key",
+    base_url="base_url"
 )
 # Initialize tracer
 tracer = Tracer(
-    project_name="prompt_metric_dataset",
-    dataset_name="ChatOpenAI_tan",
-    tracer_type="anything",
+    project_name="project_name",
+    dataset_name="dataset_name",
+    tracer_type="tracer_type",
     metadata={
         "model": "gpt-3.5-turbo",
         "environment": "production"
@@ -114,7 +114,7 @@ class ItineraryAgent:
     def __init__(self, persona="Itinerary Agent"):
         self.persona = persona
 
-    @tracer.trace_llm(name="plan_itinerary")
+    @tracer.trace_agent(name="plan_itinerary")
     def plan_itinerary(self, user_preferences, duration=3):
         # Get weather information
         weather = weather_tool(user_preferences["destination"])

{ragaai_catalyst-2.1b5 → ragaai_catalyst-2.1.1}/catalyst_v2.ipynb

@@ -49,25 +49,11 @@
       "source": [
         "from ragaai_catalyst import RagaAICatalyst\n",
         "\n",
-        "# catalyst = RagaAICatalyst(\n",
-        "#     access_key=\"dn5OuL29uA0K6G2046ZG\",\n",
-        "#     secret_key=\"Je5Kyg787f4Tdth9c75VBAen198L8KOihMTrhdYB\",\n",
-        "#     base_url='http://15.206.202.3/api'\n",
-        "#     # api_keys=OPENAI_API_KEY\n",
-        "# )\n",
-        "# os.environ[\"RAGAAI_CATALYST_BASE_URL\"] = \"http://15.206.202.3/api\"\n",
-        "\n",
         "catalyst = RagaAICatalyst(\n",
-        "    access_key=\"GOJqDkYz9WHOsJrdnOZq\",\n",
-        "    secret_key=\"UkZdlUU733CXoCFXjVrRKisp3OlDjvgevxLU3pWc\",\n",
-        "    base_url=\"https://llm-dev5.ragaai.ai/api\"\n",
-        "    # api_keys={\"OPENAI_API_KEY\": OPENAI_API_KEY}\n",
-        ")\n",
-        "\n",
-        "# catalyst = RagaAICatalyst(\n",
-        "#     access_key=\"jqa9FHN8B313H5mQS14X\",\n",
-        "#     secret_key=\"BTzSSKKzFbX8YKoHmIMNpcegsKXF9yRhnYFvyMAF\",\n",
-        "# )"
+        "    access_key=\"access_key\",\n",
+        "    secret_key=\"secret_key\",\n",
+        "    base_url=\"base_url\"\n",
+        ")"
       ]
     },
     {

{ragaai_catalyst-2.1b5 → ragaai_catalyst-2.1.1}/finance_agent_catalyst.ipynb

@@ -36,25 +36,11 @@
    "source": [
     "from ragaai_catalyst import RagaAICatalyst\n",
     "\n",
-    "# catalyst = RagaAICatalyst(\n",
-    "#     access_key=\"dn5OuL29uA0K6G2046ZG\",\n",
-    "#     secret_key=\"Je5Kyg787f4Tdth9c75VBAen198L8KOihMTrhdYB\",\n",
-    "#     base_url='http://15.206.202.3/api'\n",
-    "#     # api_keys=OPENAI_API_KEY\n",
-    "# )\n",
-    "# os.environ[\"RAGAAI_CATALYST_BASE_URL\"] = \"http://15.206.202.3/api\"\n",
-    "\n",
     "catalyst = RagaAICatalyst(\n",
-    "    access_key=\"GOJqDkYz9WHOsJrdnOZq\",\n",
-    "    secret_key=\"UkZdlUU733CXoCFXjVrRKisp3OlDjvgevxLU3pWc\",\n",
-    "    base_url=\"https://llm-dev5.ragaai.ai/api\"\n",
-    "    # api_keys={\"OPENAI_API_KEY\": OPENAI_API_KEY}\n",
-    ")\n",
-    "\n",
-    "# catalyst = RagaAICatalyst(\n",
-    "#     access_key=\"jqa9FHN8B313H5mQS14X\",\n",
-    "#     secret_key=\"BTzSSKKzFbX8YKoHmIMNpcegsKXF9yRhnYFvyMAF\",\n",
-    "# )"
+    "    access_key=\"access_key\",\n",
+    "    secret_key=\"secret_key\",\n",
+    "    base_url=\"base_url\"\n",
+    ")"
    ]
   },
   {

{ragaai_catalyst-2.1b5 → ragaai_catalyst-2.1.1}/financial_analysis_agent.py

@@ -14,9 +14,9 @@ load_dotenv()
 # Initialize OpenAI API
 openai.api_key = os.getenv("OPENAI_API_KEY")
 catalyst = RagaAICatalyst(
-    access_key="saLy6KmMVlfAzunuQGS9",
-    secret_key="lm39fd4KXffM6gzLjnY9G7QReffhH4RGZPursp3A",
-    base_url="http://52.172.168.127/api"
+    access_key="access_key",
+    secret_key="secret_key",
+    base_url="base_url"
 )
 
 # Start tracing

ragaai_catalyst-2.1.1/nested_agents.py

@@ -0,0 +1,131 @@
+import os
+import requests
+from dotenv import load_dotenv
+from litellm import completion
+from openai import OpenAI
+
+from ragaai_catalyst.tracers import Tracer
+from ragaai_catalyst import RagaAICatalyst
+import asyncio
+import os
+import requests
+from dotenv import load_dotenv
+load_dotenv()
+from litellm import completion
+import openai
+from openai import AsyncOpenAI
+
+catalyst = RagaAICatalyst(
+    access_key="access_key",
+    secret_key="secret_key",
+    base_url="base_url"
+)
+# Initialize tracer
+tracer = Tracer(
+    project_name="project_name",
+    dataset_name="dataset_name",
+    tracer_type="tracer_type",
+    metadata={
+        "model": "gpt-3.5-turbo",
+        "environment": "production"
+    },
+    pipeline={
+        "llm_model": "gpt-3.5-turbo",
+        "vector_store": "faiss",
+        "embed_model": "text-embedding-ada-002",
+    }
+)
+load_dotenv()
+
+
+tracer.start()
+
+@tracer.trace_llm(name="llm_call")
+def llm_call(prompt, max_tokens=512, model="gpt-4o-mini"):
+    client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
+    response = client.chat.completions.create(
+        model=model,
+        messages=[{"role": "user", "content": prompt}],
+        max_tokens=max_tokens,
+        temperature=0.7,
+    )
+    response_data = response.choices[0].message.content.strip()
+    print('response_data: ', response_data)
+    return response_data
+
+@tracer.trace_agent(name="summary_agent")
+class SummaryAgent:
+    def __init__(self, persona="Summary Agent"):
+        self.persona = persona
+
+    def summarize(self, text):
+        # Make an LLM call
+        prompt = f"Please summarize this text concisely: {text}"
+        summary = llm_call(prompt)
+        return summary
+        
+@tracer.trace_agent(name="analysis_agent")
+class AnalysisAgent:
+    def __init__(self, persona="Analysis Agent"):
+        self.persona = persona
+        self.summary_agent = SummaryAgent()
+
+    def analyze(self, text):
+        # First use the summary agent
+        summary = self.summary_agent.summarize(text)
+        
+        # Then make our own LLM call for analysis
+        prompt = f"Given this summary: {summary}\nProvide a brief analysis of the main points."
+        analysis = llm_call(prompt)
+        
+        return {
+            "summary": summary,
+            "analysis": analysis
+        }
+
+@tracer.trace_agent(name="recommendation_agent")
+class RecommendationAgent:
+    def __init__(self, persona="Recommendation Agent"):
+        self.persona = persona
+        self.analysis_agent = AnalysisAgent()
+
+    def recommend(self, text):
+        # First get analysis from analysis agent (which internally uses summary agent)
+        analysis_result = self.analysis_agent.analyze(text)
+        
+        # Then make our own LLM call for recommendations
+        prompt = f"""Given this summary: {analysis_result['summary']}
+        And this analysis: {analysis_result['analysis']}
+        Provide 2-3 actionable recommendations."""
+        
+        recommendations = llm_call(prompt)
+        
+        return {
+            "summary": analysis_result["summary"],
+            "analysis": analysis_result["analysis"],
+            "recommendations": recommendations
+        }
+
+def main():
+    # Sample text to analyze
+    text = """
+    Artificial Intelligence has transformed various industries in recent years.
+    From healthcare to finance, AI applications are becoming increasingly prevalent.
+    Machine learning models are being used to predict market trends, diagnose diseases,
+    and automate routine tasks. The impact of AI on society continues to grow,
+    raising both opportunities and challenges for the future.
+    """
+    
+    # Create and use the recommendation agent
+    recommendation_agent = RecommendationAgent()
+    result = recommendation_agent.recommend(text)
+    
+    print("\nResults:")
+    print("Summary:", result["summary"])
+    print("\nAnalysis:", result["analysis"])
+    print("\nRecommendations:", result["recommendations"])
+    
+
+if __name__ == "__main__":
+    main()  
+    tracer.stop()  

{ragaai_catalyst-2.1b5 → ragaai_catalyst-2.1.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"
+requires-python = ">=3.9,<3.13"
 # license = {file = "LICENSE"}
-version = "2.1.beta.5"
+version = "2.1.1"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},
@@ -33,7 +33,7 @@ dependencies = [
     "langchain-core>=0.2.11",
     "langchain>=0.2.11",
     "openai>=1.57.0",
-    "pandas>=2.1.1",
+    "pandas",
     "groq>=0.11.0",
     "PyPDF2>=3.0.1",
     "google-generativeai>=0.8.2",
@@ -41,7 +41,7 @@ dependencies = [
     "litellm==1.51.1",  
     "tenacity==8.3.0",
     "tqdm>=4.66.5",
-    "llama-index>=0.10.0",
+    "llama-index>=0.10.0,<0.11.0",
     "pyopenssl>=24.2.1",
     "psutil~=6.0.0",
     "py-cpuinfo~=9.0.0",

ragaai_catalyst-2.1.1/ragaai_catalyst/tracers/agentic_tracing/README.md

@@ -0,0 +1,97 @@
+# Agentic Tracing
+
+This module provides tracing functionality for agentic AI systems, helping track and analyze various aspects of AI agent behavior including LLM interactions, tool usage, and network activities.
+
+## Directory Structure
+
+```
+agentic_tracing/
+├── tracers/                  # Core tracing implementations
+│   ├── main_tracer.py       # Main tracing functionality
+│   ├── agent_tracer.py      # Agent behavior tracing
+│   ├── base.py              # Base tracing classes
+│   ├── llm_tracer.py        # Language model interaction tracing
+│   ├── network_tracer.py    # Network activity tracing
+│   ├── tool_tracer.py       # Tool usage tracing
+│   ├── user_interaction_tracer.py # User interaction tracing
+│   └── __init__.py          # Tracer module initialization
+├── data/                     # Data structures and classes
+│   ├── data_classes.py      # Data class definitions
+│   └── __init__.py          # Data module initialization
+├── utils/                    # Utility functions and helpers
+│   ├── api_utils.py         # API-related utilities
+│   ├── file_name_tracker.py # Tracks file names and paths
+│   ├── generic.py           # Generic utility functions
+│   ├── llm_utils.py         # LLM-specific utilities
+│   ├── model_costs.json     # Model cost configurations
+│   ├── trace_utils.py       # General tracing utilities
+│   ├── unique_decorator.py  # Unique ID generation
+│   ├── zip_list_of_unique_files.py # File handling utilities
+│   └── __init__.py          # Utils module initialization
+├── tests/                    # Test suites and examples
+│   ├── ai_travel_agent.py   # Travel agent test implementation
+│   ├── unique_decorator_test.py # Tests for unique decorator
+│   ├── TravelPlanner.ipynb  # Travel planner example notebook
+│   ├── FinancialAnalysisSystem.ipynb # Financial analysis example
+│   ├── GameActivityEventPlanner.ipynb # Game event planner example
+│   └── __init__.py          # Tests module initialization
+├── upload/                   # Upload functionality
+│   ├── upload_code.py       # Code upload utilities
+│   └── __init__.py          # Upload module initialization
+└── __init__.py              # Package initialization
+```
+
+## Components
+
+### Tracers
+Different types of tracers for various aspects of agent behavior:
+- Main Tracer: Core tracing functionality for managing and coordinating different trace types
+- Agent Tracer: Tracks agent behavior, decisions, and state changes
+- Base Tracer: Provides base classes and common functionality for all tracers
+- LLM Tracer: Monitors language model interactions, including:
+  - Token usage tracking
+  - Cost calculation
+  - Input/output monitoring
+  - Model parameter tracking
+- Network Tracer: Tracks network activities and API calls
+- Tool Tracer: Monitors tool usage and execution
+- User Interaction Tracer: Tracks user interactions and feedback
+
+### Data
+Core data structures and classes:
+- Data Classes: Defines structured data types for:
+  - LLM calls
+  - Network requests
+  - Tool executions
+  - Trace components
+  - Agent states
+  - User interactions
+
+### Utils
+Helper functions and utilities:
+- API Utils: Handles API-related operations and configurations
+- LLM Utils: Utilities for handling LLM-specific operations:
+  - Model name extraction
+  - Token usage calculation
+  - Cost computation
+  - Parameter sanitization
+- Generic Utils: Common utility functions used across modules
+- Trace Utils: General tracing utilities
+- File Name Tracker: Manages file paths and names
+- Unique Decorator: Generates unique identifiers for trace components
+- Model Costs: Configuration for different model pricing
+- Zip List of Unique Files: Handles file compression and unique file management
+
+### Tests
+Test suites and example implementations:
+- AI Travel Agent: Test implementation of a travel planning agent
+- Unique Decorator Tests: Unit tests for unique ID generation
+- Example Notebooks:
+  - Travel Planner: Example of travel planning implementation
+  - Financial Analysis: Example of financial system analysis
+  - Game Event Planner: Example of game activity planning
+
+### Upload
+Components for uploading and managing trace data:
+- Code Upload: Handles uploading of traced code and execution data
+- Supports various data formats and trace types

ragaai_catalyst-2.1.1/ragaai_catalyst/tracers/agentic_tracing/__init__.py

@@ -0,0 +1,5 @@
+from .tracers.main_tracer import AgenticTracing
+from .utils.file_name_tracker import TrackName
+from .utils.unique_decorator import generate_unique_hash_simple, mydecorator
+
+__all__ = ['AgenticTracing', 'TrackName', 'generate_unique_hash_simple', 'mydecorator']

{ragaai_catalyst-2.1b5/ragaai_catalyst/tracers/agentic_tracing → ragaai_catalyst-2.1.1/ragaai_catalyst/tracers/agentic_tracing/data}/data_structure.py

@@ -1,4 +1,4 @@
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import List, Dict, Optional, Any, Union
 from datetime import datetime
 import uuid
@@ -159,6 +159,19 @@ class ToolInfo:
     version: str
     memory_used: int
 
+@dataclass
+class LLMCall:
+    name: str
+    model_name: str
+    input_prompt: str
+    output_response: str
+    tool_call: Dict
+    token_usage: Dict[str, int]
+    cost: Dict[str, float]
+    start_time: float = field(default=0)
+    end_time: float = field(default=0)
+    duration: float = field(default=0)
+
 class Component:
     def __init__(self, id: str, hash_id: str, type: str, name: str, start_time: str, end_time: str, parent_id: int, info: Dict[str, Any], data: Dict[str, Any], network_calls: Optional[List[NetworkCall]] = None, interactions: Optional[List[Union[Interaction, Dict]]] = None, error: Optional[Dict[str, Any]] = None):
         self.id = id

{ragaai_catalyst-2.1b5/ragaai_catalyst/tracers/agentic_tracing → ragaai_catalyst-2.1.1/ragaai_catalyst/tracers/agentic_tracing/tracers}/agent_tracer.py

@@ -3,12 +3,10 @@ import uuid
 from datetime import datetime
 import psutil
 from typing import Optional, Any, Dict, List
-from .unique_decorator import mydecorator
-from .unique_decorator import generate_unique_hash_simple
-
+from ..utils.unique_decorator import mydecorator, generate_unique_hash_simple
 import contextvars
 import asyncio
-from .file_name_tracker import TrackName
+from ..utils.file_name_tracker import TrackName
 
 
 class AgentTracerMixin:
@@ -20,16 +18,23 @@ class AgentTracerMixin:
         self.agent_children = contextvars.ContextVar("agent_children", default=[])
         self.component_network_calls = contextvars.ContextVar("component_network_calls", default={})
         self.component_user_interaction = contextvars.ContextVar("component_user_interaction", default={})
+        self.version = contextvars.ContextVar("version", default="1.0.0")
+        self.agent_type = contextvars.ContextVar("agent_type", default="generic")
+        self.capabilities = contextvars.ContextVar("capabilities", default=[])
+        self.start_time = contextvars.ContextVar("start_time", default=None)
+        self.input_data = contextvars.ContextVar("input_data", default=None)
         self.gt = None
 
 
-    def trace_agent(self, name: str, agent_type: str = "generic", version: str = "1.0.0", capabilities: List[str] = None):
+    def trace_agent(self, name: str, agent_type: str = None, version: str = None, capabilities: List[str] = None):
         def decorator(target):
             # Check if target is a class
             is_class = isinstance(target, type)
             tracer = self  # Store reference to tracer instance
             top_level_hash_id = generate_unique_hash_simple(target)   # Generate hash based on the decorated target code
-
+            self.version.set(version)
+            self.agent_type.set(agent_type)
+            self.capabilities.set(capabilities)
             
             if is_class:
                 # Store original __init__
@@ -160,6 +165,8 @@ class AgentTracerMixin:
             return func(*args, **kwargs)
 
         start_time = datetime.now()
+        self.start_time = start_time
+        self.input_data = self._sanitize_input(args, kwargs)
         start_memory = psutil.Process().memory_info().rss
         component_id = str(uuid.uuid4())
 
@@ -206,12 +213,11 @@ class AgentTracerMixin:
                 start_time=start_time,
                 end_time=end_time,
                 memory_used=memory_used,
-                input_data=self._sanitize_input(args, kwargs),
+                input_data=self.input_data,
                 output_data=self._sanitize_output(result),
                 children=children,
                 parent_id=parent_agent_id
             )
-
             # Add ground truth to component data if present
             if ground_truth is not None:
                 agent_component["data"]["gt"] = ground_truth
@@ -253,13 +259,12 @@ class AgentTracerMixin:
                 start_time=start_time,
                 end_time=datetime.now(),
                 memory_used=0,
-                input_data=self._sanitize_input(args, kwargs),
+                input_data=self.input_data,
                 output_data=None,
                 error=error_component,
                 children=children,
                 parent_id=parent_agent_id  # Add parent ID if exists
             )
-            
             # If this is a nested agent, add it to parent's children
             if parent_agent_id:
                 parent_component = self._agent_components.get(parent_agent_id)

{ragaai_catalyst-2.1b5/ragaai_catalyst/tracers/agentic_tracing → ragaai_catalyst-2.1.1/ragaai_catalyst/tracers/agentic_tracing/tracers}/base.py

@@ -10,18 +10,17 @@ import uuid
 import sys
 import tempfile
 
-from .data_structure import (
+from ..data.data_structure import (
     Trace, Metadata, SystemInfo, OSInfo, EnvironmentInfo,
     Resources, CPUResource, MemoryResource, DiskResource, NetworkResource,
     ResourceInfo, MemoryInfo, DiskInfo, NetworkInfo,
     Component, 
 )
 
-from .upload_agentic_traces import UploadAgenticTraces
-from .upload_code import upload_code
-
-from .file_name_tracker import TrackName
-from .zip_list_of_unique_files import zip_list_of_unique_files
+from ..upload.upload_agentic_traces import UploadAgenticTraces
+from ..upload.upload_code import upload_code
+from ..utils.file_name_tracker import TrackName
+from ..utils.zip_list_of_unique_files import zip_list_of_unique_files
 
 class TracerJSONEncoder(json.JSONEncoder):
     def default(self, obj):