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