Merge remote-tracking branch 'origin/develop' into develop

original.py

@@ -1,5 +1,3 @@
-import sys
-
 import openai
 import time
 import re
@@ -7,6 +5,11 @@ import re
 model = "Qwen/Qwen3-8B"
 max_tokens = 500
 
+output_format = """
+score:
+reason:
+"""
+
 def getDim(content, dim):
     client = openai.OpenAI(base_url='https://api.siliconflow.cn/v1',
                            api_key="xxx")
@@ -15,6 +18,9 @@ def getDim(content, dim):
     messages.append({
         "role": "system",
         "content": "你是一位评分老师，请根据提供给你的评分标准对文本进行评分，结果仅仅输出分数，不需要任何解释"
+                    "输出格式："
+                    f"{output_format}"
+
     })
     messages.append({
         "role": "user",
@@ -28,15 +34,17 @@ def getDim(content, dim):
         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
+    # print(response.choices[0].message.content)
+    # 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 response.choices[0].message.content
 
 if __name__ == "__main__":
     content = sys.argv[1]
     dim = sys.argv[2]
-    score = getDim(content, dim)
-    print(score)
+    result = getDim(content, dim)
+    print(result)