脚本

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)