import sys

import openai
import time
import re

model = "deepseek-ai/DeepSeek-V2.5"
max_tokens = 500

def getDim(content, dim):
    client = openai.OpenAI(base_url='https://api.siliconflow.cn/v1',
                           api_key="xxx")
    time.sleep(3)
    messages = []
    messages.append({
        "role": "system",
        "content": "你是一位评分老师，请根据提供给你的评分标准对文本进行评分，结果仅仅输出分数，不需要任何解释"
    })
    messages.append({
        "role": "user",
        "content": f"待评分内容：{content}"
                    f"评分标准：{dim}"
    })
    response = client.chat.completions.create(
        model=model,
        messages=messages,
        max_tokens=max_tokens,
        temperature=0.1
    )

    match = re.search(r'\d+', response.choices[0].message.content)
    if match:
        score = match.group()
    else:
        score = response.choices[0].message.content
    return score

if __name__ == "__main__":
    content = sys.argv[1]
    dim = sys.argv[2]
    score = getDim(content, dim)
    print(score)