42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
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) |