diff --git a/original.py b/original.py index 104d7f6..cbe187d 100644 --- a/original.py +++ b/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) \ No newline at end of file + result = getDim(content, dim) + print(result) \ No newline at end of file