import os
from dingo.io import Data
from dingo.model.llm.llm_html_extract_compare_v2 import LLMHtmlExtractCompareV2

# 初始化评估器
evaluator = LLMHtmlExtractCompareV2()
evaluator.dynamic_config.model = 'gpt-4'
evaluator.dynamic_config.key = os.getenv("OPENAI_KEY")
evaluator.dynamic_config.api_url = 'https://api.openai.com/v1'

# 准备数据
data = Data(
    data_id="test_001",
    prompt="工具A提取的内容...",
    content="工具B提取的文本内容",
    raw_data={
        "language": "zh"
    }
)

# 执行评估
result = evaluator.eval(data)

# 查看结果
print(f"判断: {result.type}")
print(f"推理: {result.reason[0]}")