yuyr
bb7393bcb3
map网站使用socks5连接到官网openstreetmap； 模型改成gpt-4o; llm fuzzy math也使用4o； 增加generate_test_data.py脚本
28 lines
850 B
Python
28 lines
850 B
Python
"""Replace the website placeholders with website domains from env_config
|
|
Generate the test data"""
|
|
import json
|
|
|
|
from browser_env.env_config import *
|
|
|
|
|
|
def main() -> None:
|
|
with open("config_files/test.raw.json", "r") as f:
|
|
raw = f.read()
|
|
raw = raw.replace("__GITLAB__", GITLAB)
|
|
raw = raw.replace("__REDDIT__", REDDIT)
|
|
raw = raw.replace("__SHOPPING__", SHOPPING)
|
|
raw = raw.replace("__SHOPPING_ADMIN__", SHOPPING_ADMIN)
|
|
raw = raw.replace("__WIKIPEDIA__", WIKIPEDIA)
|
|
raw = raw.replace("__MAP__", MAP)
|
|
with open("config_files/tasks/test.json", "w") as f:
|
|
f.write(raw)
|
|
# split to multiple files
|
|
data = json.loads(raw)
|
|
for idx, item in enumerate(data):
|
|
with open(f"config_files/tasks/{idx}.json", "w") as f:
|
|
json.dump(item, f, indent=2)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|