weiweiw
/
Intention
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

多轮问询优化

This commit is contained in:
weiweiw 2025-03-03 11:03:48 +08:00
parent 62db786d2b
commit 857b3feb34
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/config.py Normal file
View File

@ -0,0 +1,3 @@
api_base_url = "http://36.33.26.201:27861/v1"
api_key = 'EMPTY'
model_name = 'qwen2.5-instruct'

5
api/mian.py
View File

@ -15,6 +15,7 @@ from fuzzywuzzy import process
from utils import CheckResult, StandardType, load_standard_name from utils import CheckResult, StandardType, load_standard_name
from constants import PROJECT_NAME, PROJECT_DEPARTMENT, SIMILARITY_VALUE from constants import PROJECT_NAME, PROJECT_DEPARTMENT, SIMILARITY_VALUE
from config import *
# 常量 # 常量
MODEL_ERNIE_PATH = R"../ernie/output/checkpoint-4160" MODEL_ERNIE_PATH = R"../ernie/output/checkpoint-4160"
MODEL_UIE_PATH = R"../uie/output/checkpoint-4320" MODEL_UIE_PATH = R"../uie/output/checkpoint-4320"
@ -242,10 +243,6 @@ def agent():
def multi_slot_recognizer(intention_id, messages): def multi_slot_recognizer(intention_id, messages):
from openai import OpenAI from openai import OpenAI
final_slot = {}
api_base_url = "http://36.33.26.201:27861/v1"
api_key = 'EMPTY'
model_name = 'qwen2.5-instruct'
client = OpenAI(base_url = api_base_url, api_key = api_key) client = OpenAI(base_url = api_base_url, api_key = api_key)
prompt = f''' prompt = f'''