config.py
1 import sys 2 import os 3 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../..'))) 4 5 from ragaai_catalyst import RagaAICatalyst, init_tracing 6 from ragaai_catalyst.tracers import Tracer 7 8 from dotenv import load_dotenv 9 load_dotenv() 10 11 def initialize_tracing(): 12 catalyst = RagaAICatalyst( 13 access_key=os.getenv("RAGAAI_CATALYST_ACCESS_KEY"), 14 secret_key=os.getenv("RAGAAI_CATALYST_SECRET_KEY"), 15 base_url=os.getenv("RAGAAI_CATALYST_BASE_URL"), 16 ) 17 18 tracer = Tracer( 19 project_name='prompt_metric_dataset',#os.getenv("RAGAAI_PROJECT_NAME"), 20 dataset_name='pytest_dataset',#os.getenv("RAGAAI_DATASET_NAME"), 21 tracer_type="Agentic", 22 ) 23 24 init_tracing(catalyst=catalyst, tracer=tracer) 25 return tracer