testcase-data/method/ALGO/config.py

from yacs.config import CfgNode as CN

cfg = CN()

# 模型
# cfg.model_name = "gpt-4o"
# cfg.model_name = "claude-sonnet-4-20250514"
# cfg.model_name = "deepseek-r1-0528-volce"
# cfg.model_name = "qwen3-235b-a22b"
# cfg.model_name = "claude-sonnet-4-20250514"
cfg.model_name = "deepseek-r1-0528-volce"
# cfg.model_name = "deepseek-v3"
# cfg.model_name = "claude-sonnet-4-20250514-thinking"
cfg.api_base = "https://models-proxy.stepfun-inc.com/v1"
# cfg.api_key = "ak-8f3d147b2c9a5e6m0n4p8x2v7y1k3l9"
cfg.api_key = "ak-58d7efgh23i4jkl67mno89pqrs01tuv6k5"
# 每道题生成的数量
cfg.gen_test_nums = 200
# 保存
cfg.dataset_name = "ours"
prefix = "/home/i-luoxianzhen/data/TestCase-Gen/save_tests_qwen3-nothink"
cfg.tests_path = prefix + '/algo/tests-{}.jsonl'
cfg.pass_rate_file = prefix + "/algo/test_pass_rate-{}.jsonl"
cfg.log_file = "./log_dir/log-algo-{}.txt"
cfg.response_file = "/home/i-luoxianzhen/data/TestCase-Gen/data/response-orginal/orginal_response_{}_{}.jsonl"