methods/Predo/config.py

from yacs.config import CfgNode as CN

cfg = CN()

# cfg.model_name_test_gen = "gpt-4o"
# cfg.model_name_answer_gen = "gpt-4o"

# cfg.model_name_test_gen = "deepseek-r1-0528-ali"
# cfg.model_name_answer_gen = "deepseek-r1-0528-ali"
# cfg.model_name = "qwen3-235b-a22b"
cfg.model_name_test_gen = "qwen3-235b-a22b"
cfg.model_name_answer_gen = "qwen3-235b-a22b"

# cfg.model_name = "deepseek-v3"
# cfg.model_name = "claude-sonnet-4-20250514-thinking"
# cfg.model_name_test_gen = "claude-sonnet-4-20250514"
# cfg.model_name_answer_gen = "claude-sonnet-4-20250514"
cfg.model_name = "claude-sonnet-4-20250514"
# 阶跃
cfg.api_base = "https://models-proxy.stepfun-inc.com/v1"
cfg.api_key = "ak-58d7efgh23i4jkl67mno89pqrs01tuv6k5"

# 生成几组解来投票
cfg.predo_count = 10

cfg.dataset_name = "ours"
prefix = "/home/i-luoxianzhen/data/TestCase-Gen/save_tests_qwen3-nothink"
cfg.tests_path = prefix + '/predo/tests-{}.jsonl'
cfg.pass_rate_file = prefix + "/predo/test_pass_rate-{}.json"
cfg.log_file = "./log_dir/log-predo-{}.txt"


cfg.response_file = "/home/i-luoxianzhen/data/TestCase-Gen/data/response-orginal/orginal_response_{}_{}.jsonl"