diff --git a/original.py b/original.py new file mode 100644 index 0000000..14359c8 --- /dev/null +++ b/original.py @@ -0,0 +1,42 @@ +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 + print(score) + return score + +if __name__ == "__main__": + content = sys.argv[1] + dim = sys.argv[2] + score = getDim(content, dim) \ No newline at end of file