lts
/
auto_score
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

This commit is contained in:
musteven 2025-05-05 12:36:40 +08:00
parent 378f47e8a1 2ef40a7f53
commit 5fd14d4684
1 changed files with 18 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
original.py
View File

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