ragaai-catalyst 2.1.1b1__tar.gz → 2.1.1b3__tar.gz

{ragaai_catalyst-2.1.1b1 → ragaai_catalyst-2.1.1b3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ragaai_catalyst
-Version: 2.1.1b1
+Version: 2.1.1b3
 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.13,>=3.9
@@ -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
+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.1.1b3/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.1.1b1 → ragaai_catalyst-2.1.1b3}/pyproject.toml

@@ -8,7 +8,7 @@ description = "RAGA AI CATALYST"
 readme = "README.md"
 requires-python = ">=3.9,<3.13"
 # license = {file = "LICENSE"}
-version = "2.1.1.beta.1"
+version = "2.1.1.beta.3"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},
@@ -41,7 +41,7 @@ dependencies = [
     "litellm==1.51.1",  
     "tenacity==8.3.0",
     "tqdm>=4.66.5",
-    "llama-index~=0.10",
+    "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.1b1 → ragaai_catalyst-2.1.1b3}/ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py

@@ -445,7 +445,7 @@ class LLMTracerMixin:
                 usage=token_usage,
                 parameters=parameters
             )
-            
+            self.llm_data = llm_component
             self.add_component(llm_component)
             return result
 
@@ -482,7 +482,7 @@ class LLMTracerMixin:
                 output_data=None,
                 error=error_component
             )
-    
+            self.llm_data = llm_component
             self.add_component(llm_component, is_error=True)
             raise
 

{ragaai_catalyst-2.1.1b1 → ragaai_catalyst-2.1.1b3}/ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py

@@ -131,6 +131,9 @@ class UploadAgenticTraces:
                                         headers=headers, 
                                         data=payload,
                                         timeout=self.timeout)
+            if response.status_code != 200:
+                print(f"Error inserting traces: {response.json()['message']}")
+                return None
         except requests.exceptions.RequestException as e:
             print(f"Error while inserting traces: {e}")
             return None

{ragaai_catalyst-2.1.1b1 → ragaai_catalyst-2.1.1b3}/ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py

@@ -13,6 +13,10 @@ def upload_code(hash_id, zip_path, project_name, dataset_name):
         _put_zip_presigned_url(project_name, presigned_url, zip_path)
 
         response = _insert_code(dataset_name, hash_id, presigned_url, project_name)
+
+        if response.status_code != 200 or response.status_code != 201:
+            raise Exception(f"Failed to insert code: {response.json()['message']}")
+
         return response
     else:
         return "Code already exists"

{ragaai_catalyst-2.1.1b1 → ragaai_catalyst-2.1.1b3}/ragaai_catalyst.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ragaai_catalyst
-Version: 2.1.1b1
+Version: 2.1.1b3
 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.13,>=3.9
@@ -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
+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.1.1b1 → ragaai_catalyst-2.1.1b3}/ragaai_catalyst.egg-info/SOURCES.txt

@@ -4,6 +4,7 @@ ai_travel_agent.py
 catalyst_v2.ipynb
 finance_agent_catalyst.ipynb
 financial_analysis_agent.py
+nested_agents.py
 pyproject.toml
 requirements.txt
 sample_llm_testing_code.py

{ragaai_catalyst-2.1.1b1 → ragaai_catalyst-2.1.1b3}/ragaai_catalyst.egg-info/requires.txt

@@ -20,7 +20,7 @@ Markdown>=3.7
 litellm==1.51.1
 tenacity==8.3.0
 tqdm>=4.66.5
-llama-index~=0.10
+llama-index<0.11.0,>=0.10.0
 pyopenssl>=24.2.1
 psutil~=6.0.0
 py-cpuinfo~=9.0.0