from pathlib import Path
from dingo.config import InputArgs
from dingo.exec import Executor
# 获取项目根目录
PROJECT_ROOT = Path(__file__).parent.parent.parent
def local_plaintext():
input_data = {
"input_path": str(PROJECT_ROOT / "test/data/test_local_plaintext.txt"),
"dataset": {
"source": "local",
"format": "plaintext",
},
"evaluator": [
{
"fields": {"content": "content"},
"evals": [
{"name": "RuleColonEnd"}
]
}
]
}
input_args = InputArgs(**input_data)
executor = Executor.exec_map["local"](input_args)
result = executor.execute()
print(result)
def local_json():
input_data = {
"input_path": str(PROJECT_ROOT / "test/data/test_local_json.json"),
"dataset": {
"source": "local",
"format": "json",
},
"evaluator": [
{
"fields": {"content": "prediction"},
"evals": [
{"name": "RuleColonEnd"}
]
}
]
}
input_args = InputArgs(**input_data)
executor = Executor.exec_map["local"](input_args)
result = executor.execute()
print(result)
def local_jsonl():
input_data = {
"input_path": str(PROJECT_ROOT / "test/data/test_local_jsonl.jsonl"),
"dataset": {
"source": "local",
"format": "jsonl",
},
"evaluator": [
{
"fields": {"content": "content"},
"evals": [
{"name": "RuleColonEnd"}
]
}
]
}
input_args = InputArgs(**input_data)
executor = Executor.exec_map["local"](input_args)
result = executor.execute()
print(result)
def local_listjson():
input_data = {
"input_path": str(PROJECT_ROOT / "test/data/test_local_listjson.json"),
"dataset": {
"source": "local",
"format": "listjson",
},
"evaluator": [
{
"fields": {"content": "output"},
"evals": [
{"name": "RuleColonEnd"}
]
}
]
}
input_args = InputArgs(**input_data)
executor = Executor.exec_map["local"](input_args)
result = executor.execute()
print(result)
if __name__ == '__main__':
local_plaintext()
local_json()
local_jsonl()
local_listjson()
