ragaai-catalyst 2.1.1b6__py3-none-any.whl → 2.1.2b0__py3-none-any.whl

ragaai_catalyst/dataset.py

@@ -16,11 +16,7 @@ class Dataset:
     def __init__(self, project_name):
         self.project_name = project_name
         self.num_projects = 99999
-        Dataset.BASE_URL = (
-            os.getenv("RAGAAI_CATALYST_BASE_URL")
-            if os.getenv("RAGAAI_CATALYST_BASE_URL")
-            else "https://catalyst.raga.ai/api"
-        )
+        Dataset.BASE_URL = RagaAICatalyst.BASE_URL
         headers = {
             "Authorization": f'Bearer {os.getenv("RAGAAI_CATALYST_TOKEN")}',
         }

ragaai_catalyst/experiment.py

@@ -31,11 +31,7 @@ class Experiment:
         Returns:
             None
         """
-        Experiment.BASE_URL = (
-            os.getenv("RAGAAI_CATALYST_BASE_URL")
-            if os.getenv("RAGAAI_CATALYST_BASE_URL")
-            else "https://llm-platform.prod5.ragaai.ai/api"
-        )
+        Experiment.BASE_URL = RagaAICatalyst.BASE_URL
         self.project_name = project_name
         self.experiment_name = experiment_name
         self.experiment_description = experiment_description

ragaai_catalyst/tracers/agentic_tracing/tracers/base.py

@@ -9,7 +9,7 @@ from typing import List
 import uuid
 import sys
 import tempfile
-
+from ....ragaai_catalyst import RagaAICatalyst
 from ..data.data_structure import (
     Trace, Metadata, SystemInfo, OSInfo, EnvironmentInfo,
     Resources, CPUResource, MemoryResource, DiskResource, NetworkResource,
@@ -203,7 +203,7 @@ class BaseTracer:
             project_id = self.project_id 
             dataset_name = self.dataset_name
             user_detail = self.user_details
-            base_url = os.getenv('RAGAAI_CATALYST_BASE_URL')
+            base_url = RagaAICatalyst.BASE_URL
             upload_traces = UploadAgenticTraces(
                 json_file_path=json_file_path,
                 project_name=project_name,

ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py

@@ -3,6 +3,7 @@ import requests
 import json
 import os
 import logging
+from ragaai_catalyst.ragaai_catalyst import RagaAICatalyst
 logger = logging.getLogger(__name__)
 
 def upload_code(hash_id, zip_path, project_name, dataset_name):
@@ -26,7 +27,7 @@ def _fetch_dataset_code_hashes(project_name, dataset_name):
 
     try:
         response = requests.request("GET", 
-                                    f"{os.getenv('RAGAAI_CATALYST_BASE_URL')}/v2/llm/dataset/code?datasetName={dataset_name}", 
+                                    f"{RagaAICatalyst.BASE_URL}/v2/llm/dataset/code?datasetName={dataset_name}", 
                                     headers=headers, 
                                     data=payload,
                                     timeout=99999)
@@ -54,7 +55,7 @@ def _fetch_presigned_url(project_name, dataset_name):
 
     try:
         response = requests.request("GET", 
-                                    f"{os.getenv('RAGAAI_CATALYST_BASE_URL')}/v1/llm/presigned-url", 
+                                    f"{RagaAICatalyst.BASE_URL}/v1/llm/presigned-url", 
                                     headers=headers, 
                                     data=payload,
                                     timeout=99999)
@@ -102,7 +103,7 @@ def _insert_code(dataset_name, hash_id, presigned_url, project_name):
     
     try:
         response = requests.request("POST", 
-                                    f"{os.getenv('RAGAAI_CATALYST_BASE_URL')}/v2/llm/dataset/code", 
+                                    f"{RagaAICatalyst.BASE_URL}/v2/llm/dataset/code", 
                                     headers=headers, 
                                     data=payload,
                                     timeout=99999)

ragaai_catalyst/tracers/exporters/raga_exporter.py

@@ -55,11 +55,7 @@ class RagaExporter:
         """
         self.project_name = project_name
         self.dataset_name = dataset_name
-        RagaExporter.BASE_URL = (
-            os.getenv("RAGAAI_CATALYST_BASE_URL")
-            if os.getenv("RAGAAI_CATALYST_BASE_URL")
-            else "https://catalyst.raga.ai/api"
-        )
+        RagaExporter.BASE_URL = RagaAICatalyst.BASE_URL
         self.access_key = os.getenv("RAGAAI_CATALYST_ACCESS_KEY")
         self.secret_key = os.getenv("RAGAAI_CATALYST_SECRET_KEY")
         self.max_urls = 20

{ragaai_catalyst-2.1.1b6.dist-info → ragaai_catalyst-2.1.2b0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.1b6
+Version: 2.1.2b0
 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

{ragaai_catalyst-2.1.1b6.dist-info → ragaai_catalyst-2.1.2b0.dist-info}/RECORD

@@ -1,8 +1,8 @@
 ragaai_catalyst/__init__.py,sha256=BdIJ_UUre0uEnRTsLw_hE0C0muWk6XWNZqdVOel22R4,537
 ragaai_catalyst/_version.py,sha256=JKt9KaVNOMVeGs8ojO6LvIZr7ZkMzNN-gCcvryy4x8E,460
-ragaai_catalyst/dataset.py,sha256=On-iOhD5R7iLusph6dLAGS1dOnNtb1koiKxKjTH90pE,10660
+ragaai_catalyst/dataset.py,sha256=aTRvZicAXmrC0KdmmsoJH_rtEJrxbqYmf1P806c1Wg0,10521
 ragaai_catalyst/evaluation.py,sha256=34H2bYZNSrcu0jMQgDZw1OLVbQU80PaVLo2avju8POM,20311
-ragaai_catalyst/experiment.py,sha256=8KvqgJg5JVnt9ghhGDJvdb4mN7ETBX_E5gNxBT0Nsn8,19010
+ragaai_catalyst/experiment.py,sha256=8yQo1phCHlpnJ-4CqCaIbLXg_1ZlAuLGI9kqGBl-OTE,18859
 ragaai_catalyst/guard_executor.py,sha256=llPbE3DyVtrybojXknzBZj8-dtUrGBQwi9-ZiPJxGRo,3762
 ragaai_catalyst/guardrails_manager.py,sha256=DILMOAASK57FH9BLq_8yC1AQzRJ8McMFLwCXgYwNAd4,11904
 ragaai_catalyst/internal_api_completion.py,sha256=DdICI5yfEudiOAIC8L4oxH0Qz7kX-BZCdo9IWsi2gNo,2965
@@ -27,7 +27,7 @@ ragaai_catalyst/tracers/agentic_tracing/tests/ai_travel_agent.py,sha256=S4rCcKzU
 ragaai_catalyst/tracers/agentic_tracing/tests/unique_decorator_test.py,sha256=Xk1cLzs-2A3dgyBwRRnCWs7Eubki40FVonwd433hPN8,4805
 ragaai_catalyst/tracers/agentic_tracing/tracers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ragaai_catalyst/tracers/agentic_tracing/tracers/agent_tracer.py,sha256=_9yGrHLw9TnKkYaNkWHQaLFQgsi945CClDTpf6JhINk,21372
-ragaai_catalyst/tracers/agentic_tracing/tracers/base.py,sha256=lxr62VIbULd9vqCT30nvuWfokvbm7LLWlsu84m0YQfk,14043
+ragaai_catalyst/tracers/agentic_tracing/tracers/base.py,sha256=liLdiHdqc6BHH4YyvrdmX_fprXoeJchOqBwmzMJPrCc,14075
 ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py,sha256=uAoUm2NyFL27FF6YExtNiXSYVRHTS1YpqmLHu6_eolE,23983
 ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py,sha256=urY3AWI01JAlGe2rjg-K0EV0CFHti4_SUOAXp23ROk4,10246
 ragaai_catalyst/tracers/agentic_tracing/tracers/network_tracer.py,sha256=6FTA15xMnum9omM_0Jd9cMIuWdKu1gR5Tc8fOXAkP8E,10068
@@ -35,7 +35,7 @@ ragaai_catalyst/tracers/agentic_tracing/tracers/tool_tracer.py,sha256=RuxmiOT8DU
 ragaai_catalyst/tracers/agentic_tracing/tracers/user_interaction_tracer.py,sha256=wsCwTK7tM_L3mdNrcg5Mq3D1k07XCHZkhOB26kz_rLY,1472
 ragaai_catalyst/tracers/agentic_tracing/upload/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py,sha256=2Ab8odZXMpYFTh8a7tO53qx9RptF-xxyfgtFN0A6GzI,7690
-ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py,sha256=u9bRWcM5oDezJupEQoHUXrKz7YvZJK9IZf10ejBWoa4,4254
+ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py,sha256=HgpMgI-JTWZrizcM7GGUIaAgaZF4aRT3D0dJXVEkblY,4271
 ragaai_catalyst/tracers/agentic_tracing/utils/__init__.py,sha256=XdB3X_ufe4RVvGorxSqAiB9dYv4UD7Hvvuw3bsDUppY,60
 ragaai_catalyst/tracers/agentic_tracing/utils/api_utils.py,sha256=JyNCbfpW-w4O9CjtemTqmor2Rh1WGpQwhRaDSRmBxw8,689
 ragaai_catalyst/tracers/agentic_tracing/utils/file_name_tracker.py,sha256=515NNDQJTyy3O-2rdlUYUoWL9qSwLIfvV3sMB9BtHp8,1366
@@ -47,14 +47,14 @@ ragaai_catalyst/tracers/agentic_tracing/utils/unique_decorator.py,sha256=DQHjcEu
 ragaai_catalyst/tracers/agentic_tracing/utils/zip_list_of_unique_files.py,sha256=faFat_OAUnVJGnauMVo6yeHhTv-_njgyXGOtUwYJ8kE,7568
 ragaai_catalyst/tracers/exporters/__init__.py,sha256=kVA8zp05h3phu4e-iHSlnznp_PzMRczB7LphSsZgUjg,138
 ragaai_catalyst/tracers/exporters/file_span_exporter.py,sha256=RgGteu-NVGprXKkynvyIO5yOjpbtA41R3W_NzCjnkwE,6445
-ragaai_catalyst/tracers/exporters/raga_exporter.py,sha256=pU9EH6Fn1Z757bpM3VaXPDHuJ5wyUrMObGaACiztvTU,18014
+ragaai_catalyst/tracers/exporters/raga_exporter.py,sha256=6xvjWXyh8XPkHKSLLmAZUQSvwuyY17ov8pv2VdfI0qA,17875
 ragaai_catalyst/tracers/instrumentators/__init__.py,sha256=FgnMQupoRTzmVsG9YKsLQera2Pfs-AluZv8CxwavoyQ,253
 ragaai_catalyst/tracers/instrumentators/langchain.py,sha256=yMN0qVF0pUVk6R5M1vJoUXezDo1ejs4klCFRlE8x4vE,574
 ragaai_catalyst/tracers/instrumentators/llamaindex.py,sha256=SMrRlR4xM7k9HK43hakE8rkrWHxMlmtmWD-AX6TeByc,416
 ragaai_catalyst/tracers/instrumentators/openai.py,sha256=14R4KW9wQCR1xysLfsP_nxS7cqXrTPoD8En4MBAaZUU,379
 ragaai_catalyst/tracers/utils/__init__.py,sha256=KeMaZtYaTojilpLv65qH08QmpYclfpacDA0U3wg6Ybw,64
 ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
-ragaai_catalyst-2.1.1b6.dist-info/METADATA,sha256=r7VEPiL5VXrVHclTP9Ir_r1YOY-sr1W7CB7fXhWlV_0,1804
-ragaai_catalyst-2.1.1b6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ragaai_catalyst-2.1.1b6.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1.1b6.dist-info/RECORD,,
+ragaai_catalyst-2.1.2b0.dist-info/METADATA,sha256=8UeqmLSidxpyuKNSw3nP2x2-H1Bz2QrrJltX5uVRDIM,1804
+ragaai_catalyst-2.1.2b0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ragaai_catalyst-2.1.2b0.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.2b0.dist-info/RECORD,,