in evaluate.py [0:0]
def get_o1_mini_model():
"""Get the o1-mini model for generating evaluation reports."""
# Read API key from the environment variables
openai_api_key = os.environ["OPENAI_API_KEY"]
if not openai_api_key:
raise ValueError("OPENAI_API_KEY environment variable is not set")
# Define o1-mini model
model = ChatOpenAI(
model_name="o1-mini",
temperature=1,
api_key=openai_api_key,
)
return model