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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ragaai_catalyst
-Version: 2.1b1
+Version: 2.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.9

{ragaai_catalyst-2.1b1 → ragaai_catalyst-2.1b3}/ai_travel_agent.py

@@ -14,26 +14,23 @@ load_dotenv()
 from litellm import completion
 import openai
 from openai import AsyncOpenAI
-import vertexai
-from vertexai.generative_models import GenerativeModel
-
 
 catalyst = RagaAICatalyst(
-    access_key="mp4NjrgFqjavNti38Uh4",
-    secret_key="9OC0ZIiVZQRA4sYsb4025cfUbRPD3s5RTl3iho2U",
-    base_url="https://llm-dev5.ragaai.ai/api"
+    access_key="saLy6KmMVlfAzunuQGS9",
+    secret_key="lm39fd4KXffM6gzLjnY9G7QReffhH4RGZPursp3A",
+    base_url="http://52.172.168.127/api"
 )
 # Initialize tracer
 tracer = Tracer(
-    project_name="Alteryx_Copilot_DS",
-    dataset_name="AITravelAgent",
-    tracer_type="AgenticTracing",
+    project_name="prompt_metric_dataset",
+    dataset_name="ChatOpenAI_tan",
+    tracer_type="anything",
     metadata={
-        "model": "google/gemini-1.5-flash-002",
+        "model": "gpt-3.5-turbo",
         "environment": "production"
     },
     pipeline={
-        "llm_model": "google/gemini-1.5-flash-002",
+        "llm_model": "gpt-3.5-turbo",
         "vector_store": "faiss",
         "embed_model": "text-embedding-ada-002",
     }
@@ -45,28 +42,17 @@ tracer.start()
 
 
 @tracer.trace_llm(name="llm_call")
-def llm_call(prompt, max_tokens=512, model="gemini-1.5-flash-002", name="default"):
-    import vertexai
-    from vertexai.generative_models import GenerativeModel
-    
-    # Initialize Vertex AI
-    project_id = os.environ.get("GOOGLE_CLOUD_PROJECT")
-    location = os.environ.get("GOOGLE_CLOUD_LOCATION", "us-central1")
-    vertexai.init(project=project_id, location=location)
-    
-    # Create model instance
-    model = GenerativeModel(model)
-    
-    # Generate content
-    response = model.generate_content(
-        prompt,
-        generation_config={
-            "max_output_tokens": max_tokens,
-            "temperature": 0.7,
-        }
+def llm_call(prompt, max_tokens=512, model="gpt-4o-mini", name="default"):
+    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,
     )
     
-    actual_response = response.text.strip()
+    actual_response = response.choices[0].message.content.strip()
     
     return actual_response
 
@@ -170,8 +156,8 @@ def travel_agent():
 
     # Get user input
     # user_input = input("Please describe your ideal vacation: ")
-    # user_input = input("Please describe your ideal vacation: ")
-    user_input = "Antartica, 2 days, $10000, scientific"
+    user_input = input("Please describe your ideal vacation: ")
+    #"karela, 10 days, $100, nature"
 
     # Extract preferences
     preferences_prompt = f"""

{ragaai_catalyst-2.1b1 → ragaai_catalyst-2.1b3}/pyproject.toml

@@ -8,7 +8,7 @@ description = "RAGA AI CATALYST"
 readme = "README.md"
 requires-python = ">=3.9"
 # license = {file = "LICENSE"}
-version = "2.1.beta.1"
+version = "2.1.beta.3"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},

{ragaai_catalyst-2.1b1 → ragaai_catalyst-2.1b3}/ragaai_catalyst/tracers/agentic_tracing/agentic_tracing.py

@@ -112,6 +112,7 @@ class AgenticTracing(BaseTracer, LLMTracerMixin, ToolTracerMixin, AgentTracerMix
             
             # Cleanup
             self.unpatch_llm_calls()
+            self.user_interaction_tracer.interactions = []  # Clear interactions list
             self.is_active = False
 
 

{ragaai_catalyst-2.1b1 → ragaai_catalyst-2.1b3}/ragaai_catalyst/tracers/agentic_tracing/base.py

@@ -1,27 +1,24 @@
 import json
 import os
 import platform
-import re
 import psutil
 import pkg_resources
 from datetime import datetime
 from pathlib import Path
-from typing import Optional, Dict, Any, List
+from typing import List
 import uuid
 import sys
+import tempfile
 
 from .data_structure import (
     Trace, Metadata, SystemInfo, OSInfo, EnvironmentInfo,
     Resources, CPUResource, MemoryResource, DiskResource, NetworkResource,
     ResourceInfo, MemoryInfo, DiskInfo, NetworkInfo,
-    Component, LLMComponent, AgentComponent, ToolComponent,
-    NetworkCall, Interaction, Error
+    Component, 
 )
 
-from ..upload_traces import UploadTraces
 from .upload_agentic_traces import UploadAgenticTraces
 from .upload_code import upload_code
-from ...ragaai_catalyst import RagaAICatalyst
 
 from .file_name_tracker import TrackName
 from .zip_list_of_unique_files import zip_list_of_unique_files
@@ -182,14 +179,13 @@ class BaseTracer:
             self.trace = self._extract_cost_tokens(self.trace)
             
             # Create traces directory if it doesn't exist
-            self.traces_dir = Path("traces")
-            self.traces_dir.mkdir(exist_ok=True)
+            self.traces_dir = tempfile.gettempdir()
             filename = self.trace.id + ".json"
-            filepath = self.traces_dir / filename
+            filepath = f"{self.traces_dir}/{filename}"
 
             #get unique files and zip it. Generate a unique hash ID for the contents of the files
             list_of_unique_files = self.file_tracker.get_unique_files()
-            hash_id, zip_path = zip_list_of_unique_files(list_of_unique_files)
+            hash_id, zip_path = zip_list_of_unique_files(list_of_unique_files, output_dir=self.traces_dir)
 
             #replace source code with zip_path
             self.trace.metadata.system_info.source_code = hash_id
@@ -230,7 +226,7 @@ class BaseTracer:
             
         # Cleanup
         self.components = []
-        self.file_tracker = TrackName()
+        self.file_tracker.reset()
                 
     def add_component(self, component: Component):
         """Add a component to the trace"""

{ragaai_catalyst-2.1b1 → ragaai_catalyst-2.1b3}/ragaai_catalyst/tracers/agentic_tracing/file_name_tracker.py

@@ -8,13 +8,9 @@ class TrackName:
     def trace_decorator(self, func):
         @wraps(func)
         def wrapper(*args, **kwargs):
-            # frame = inspect.stack()[1]
-            # file_name = frame.filename
             file_name = self._get_file_name()
-            # print(f"Called from file: {frame.filename}")
-            # print(f"Called from line: {frame.lineno}")
-            # print(f"Called from function: {frame.function}")
             self.files.add(file_name)
+
             return func(*args, **kwargs)
         return wrapper
 
@@ -43,4 +39,8 @@ class TrackName:
 
         
     def get_unique_files(self):
-        return list(self.files)
+        return list(self.files)
+        
+    def reset(self):
+        """Reset the file tracker by clearing all tracked files."""
+        self.files.clear()

{ragaai_catalyst-2.1b1 → ragaai_catalyst-2.1b3}/ragaai_catalyst/tracers/agentic_tracing/llm_tracer.py

@@ -654,6 +654,10 @@ class LLMTracerMixin:
                 finally:
 
                     llm_component = self.llm_data
+                    llm_component['name'] = name
+
+                    if self.gt:
+                        llm_component["data"]["gt"] = self.gt
 
                     if error_info:
                         llm_component["error"] = error_info["error"]

{ragaai_catalyst-2.1b1 → ragaai_catalyst-2.1b3}/ragaai_catalyst/tracers/agentic_tracing/zip_list_of_unique_files.py

@@ -1,161 +1,3 @@
-# import os
-# import hashlib
-# import zipfile
-# import re
-# import ast
-# import importlib.util
-# import json
-# from pathlib import Path
-
-# class TraceDependencyTracker:
-#     def __init__(self, output_dir=None):
-#         self.tracked_files = set()
-#         self.python_imports = set()
-#         self.output_dir = output_dir or os.getcwd()
-        
-#     def track_file_access(self, filepath):
-#         """Track a file that's been accessed."""
-#         if os.path.exists(filepath):
-#             self.tracked_files.add(os.path.abspath(filepath))
-
-#     def find_config_files(self, content, base_path):
-#         """Find configuration files referenced in the content."""
-#         patterns = [
-#             r'(?:open|read|load|with\s+open)\s*\([\'"]([^\'"]*\.(?:json|yaml|yml|txt|cfg|config|ini))[\'"]',
-#             r'(?:config|cfg|conf|settings|file|path)(?:_file|_path)?\s*=\s*[\'"]([^\'"]*\.(?:json|yaml|yml|txt|cfg|config|ini))[\'"]',
-#             r'[\'"]([^\'"]*\.txt)[\'"]',
-#             r'[\'"]([^\'"]*\.(?:yaml|yml))[\'"]',
-#             r'from\s+(\S+)\s+import',
-#             r'import\s+(\S+)'
-#         ]
-        
-#         for pattern in patterns:
-#             matches = re.finditer(pattern, content)
-#             for match in matches:
-#                 filepath = match.group(1)
-#                 if not os.path.isabs(filepath):
-#                     full_path = os.path.join(os.path.dirname(base_path), filepath)
-#                 else:
-#                     full_path = filepath
-                
-#                 if os.path.exists(full_path):
-#                     self.track_file_access(full_path)
-#                     try:
-#                         with open(full_path, 'r', encoding='utf-8') as f:
-#                             self.find_config_files(f.read(), full_path)
-#                     except (UnicodeDecodeError, IOError):
-#                         pass
-
-#     def analyze_python_imports(self, filepath):
-#         """Analyze Python file for imports and track imported files."""
-#         try:
-#             with open(filepath, 'r', encoding='utf-8') as file:
-#                 tree = ast.parse(file.read(), filename=filepath)
-                
-#             for node in ast.walk(tree):
-#                 if isinstance(node, (ast.Import, ast.ImportFrom)):
-#                     if isinstance(node, ast.ImportFrom) and node.module:
-#                         module_name = node.module
-#                     else:
-#                         for name in node.names:
-#                             module_name = name.name.split('.')[0]
-                            
-#                     try:
-#                         spec = importlib.util.find_spec(module_name)
-#                         if spec and spec.origin and not spec.origin.startswith(os.path.dirname(importlib.__file__)):
-#                             self.python_imports.add(spec.origin)
-#                     except (ImportError, AttributeError):
-#                         pass
-#         except Exception as e:
-#             print(f"Warning: Could not analyze imports in {filepath}: {str(e)}")
-
-#     def create_zip(self, filepaths):
-#         """
-#         Process files and create a single zip with all dependencies.
-        
-#         Args:
-#             filepaths (list): List of file paths to process.
-            
-#         Returns:
-#             tuple: A tuple containing the hash ID (str) and the path to the saved .zip file (str).
-#         """
-#         # Process all files and their dependencies
-#         for filepath in filepaths:
-#             abs_path = os.path.abspath(filepath)
-#             self.track_file_access(abs_path)
-            
-#             try:
-#                 with open(abs_path, 'r', encoding='utf-8') as file:
-#                     content = file.read()
-                
-#                 self.find_config_files(content, abs_path)
-                
-#                 if filepath.endswith('.py'):
-#                     self.analyze_python_imports(abs_path)
-#             except Exception as e:
-#                 print(f"Warning: Could not process {filepath}: {str(e)}")
-        
-#         # Add Python imports to tracked files
-#         self.tracked_files.update(self.python_imports)
-        
-#         # Generate hash from all files
-#         hash_contents = []
-#         for filepath in sorted(self.tracked_files):
-#             # Skip any file paths that contain 'env'
-#             if 'env' in filepath:
-#                 continue  # Skip env folder
-#             try:
-#                 with open(filepath, 'rb') as file:
-#                     content = file.read()
-#                     hash_contents.append(content)
-#             except Exception as e:
-#                 print(f"Warning: Could not read {filepath} for hash calculation: {str(e)}")
-                
-#         combined_content = b''.join(hash_contents)
-#         hash_id = hashlib.sha256(combined_content).hexdigest()
-
-#         # Create zip file
-#         zip_filename = os.path.join(self.output_dir, f'{hash_id}.zip')
-        
-#         # Determine base path excluding 'env' folders
-#         base_path = os.path.commonpath([os.path.abspath(p) for p in self.tracked_files if 'env' not in p])
-        
-#         with zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED) as zipf:
-#             for filepath in sorted(self.tracked_files):
-#                 # Skip any file paths that contain 'env'
-#                 if 'env' in filepath:
-#                     continue  # Skip env folder
-#                 try:
-#                     relative_path = os.path.relpath(filepath, base_path)
-#                     zipf.write(filepath, relative_path)
-#                     print(f"Added to zip: {relative_path}")
-#                 except Exception as e:
-#                     print(f"Warning: Could not add {filepath} to zip: {str(e)}")
-
-#         return hash_id, zip_filename
-
-# def zip_list_of_unique_files(filepaths):
-#     """
-#     Enhanced version of the original function that tracks all dependencies.
-    
-#     Args:
-#         filepaths (list): List of file paths to process.
-        
-#     Returns:
-#         tuple: A tuple containing the hash ID (str) and the path to the saved .zip file (str).
-#     """
-#     tracker = TraceDependencyTracker()
-#     return tracker.create_zip(filepaths)
-
-# if __name__ == "__main__":
-#     filepaths = ["script1.py", "script2.py"]
-#     hash_id, zip_path = zip_list_of_unique_files(filepaths)
-#     print(f"Created zip file: {zip_path}")
-#     print(f"Hash ID: {hash_id}")
-
-
-
-
 import os
 import hashlib
 import zipfile
@@ -165,6 +7,8 @@ import importlib.util
 import json
 import astor
 from pathlib import Path
+import logging
+logger = logging.getLogger(__name__)
 
 # Define the PackageUsageRemover class
 class PackageUsageRemover(ast.NodeTransformer):
@@ -323,15 +167,15 @@ class TraceDependencyTracker:
                 try:
                     relative_path = os.path.relpath(filepath, base_path)
                     zipf.write(filepath, relative_path)
-                    print(f"Added to zip: {relative_path}")
+                    # logger.info(f"Added to zip: {relative_path}")
                 except Exception as e:
                     print(f"Warning: Could not add {filepath} to zip: {str(e)}")
 
         return hash_id, zip_filename
 
 # Main function for creating a zip of unique files
-def zip_list_of_unique_files(filepaths):
-    tracker = TraceDependencyTracker()
+def zip_list_of_unique_files(filepaths, output_dir):
+    tracker = TraceDependencyTracker(output_dir)
     return tracker.create_zip(filepaths)
 
 # Example usage

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ragaai_catalyst
-Version: 2.1b1
+Version: 2.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.9

{ragaai_catalyst-2.1b1 → ragaai_catalyst-2.1b3}/ragaai_catalyst.egg-info/SOURCES.txt

@@ -72,4 +72,11 @@ test/test_catalyst/test_configuration.py
 test/test_catalyst/test_dataset.py
 test/test_catalyst/test_evaluation.py
 test/test_catalyst/test_prompt_manager.py
-test/test_catalyst/test_synthetic_data_generation.py
+test/test_catalyst/test_synthetic_data_generation.py
+traces/2439beda-8f3a-45c3-82de-4f72624c6ce9.json
+traces/cd571931-29fe-4d57-9523-71381c97abb1.json
+traces/d3d5ad2d-6a98-40d6-8b69-8b082eab1bc2.json
+traces/ed050fde-8705-4635-8570-3aa7f6e4b286.json
+traces/f2d41172-f89e-47e4-9523-1c1fe6ce20eb.json
+traces/prompt_metric_dataset_finance.json
+traces/prompt_metric_dataset_travel.json

{ragaai_catalyst-2.1b1 → ragaai_catalyst-2.1b3}/requirements.txt

@@ -20,7 +20,6 @@ google-api-core==2.24.0
 google-api-python-client==2.156.0
 google-auth==2.37.0
 google-auth-httplib2==0.2.0
-google-cloud-aiplatform==1.45.0
 google-generativeai==0.8.3
 googleapis-common-protos==1.66.0
 groq==0.13.1