eduzrh commited on
Commit
4547d13
·
verified ·
1 Parent(s): 9486dca

Security: remove API keys

Browse files
Files changed (1) hide show
  1. experiments/run_paper_scale_v5.py +7 -7
experiments/run_paper_scale_v5.py CHANGED
@@ -15,19 +15,19 @@ Fixes over V4:
15
  import os, sys, json, time, csv, random, copy
16
  from collections import defaultdict
17
 
18
- sys.path.insert(0, '/root/autodl-tmp/AdaCoAgentEA')
19
- sys.path.insert(0, '/root/autodl-tmp/AdaCoAgentEA/baselines/evorm_wrappers')
20
 
21
  from openai import OpenAI
22
  from baselines.evorm_wrappers.matchgpt import run_matchgpt, load_er_dataset
23
  from baselines.evorm_wrappers.lela_el import run_lela, run_schema_matching
24
 
25
  # Config
26
- API_BASE = "https://hk.xty.app/v1"
27
- API_KEY = "sk-7a7Ev4VcVyysPLT5hqtqIVD6PybzJ1ZlEIVZddIR3NtZvPgK"
28
- DATA_DIR = '/root/autodl-tmp/AdaCoAgentEA/data'
29
- PERSIST_DIR = '/root/autodl-tmp/AdaCoAgentEA/persist_v5'
30
- RESULTS_DIR = '/root/autodl-tmp/AdaCoAgentEA/results'
31
 
32
  EVAL_N = 300
33
  WARMUP_N = 100
 
15
  import os, sys, json, time, csv, random, copy
16
  from collections import defaultdict
17
 
18
+ sys.path.insert(0, '/root/autodl-tmp/EvoRM')
19
+ sys.path.insert(0, '/root/autodl-tmp/EvoRM/baselines/evorm_wrappers')
20
 
21
  from openai import OpenAI
22
  from baselines.evorm_wrappers.matchgpt import run_matchgpt, load_er_dataset
23
  from baselines.evorm_wrappers.lela_el import run_lela, run_schema_matching
24
 
25
  # Config
26
+ API_BASE = os.environ.get("OPENAI_BASE_URL", "https://api.openai.com/v1")
27
+ # API_KEY set via environment variable
28
+ DATA_DIR = '/root/autodl-tmp/EvoRM/data'
29
+ PERSIST_DIR = '/root/autodl-tmp/EvoRM/persist_v5'
30
+ RESULTS_DIR = '/root/autodl-tmp/EvoRM/results'
31
 
32
  EVAL_N = 300
33
  WARMUP_N = 100