67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
from openai import OpenAI
|
||
query_type_prompt = """
|
||
你是一个专业的装备维修领域用户查询判断助手，能够根据以下规则准确判断用户问题属于哪种类型。
|
||
|
||
### 判断规则
|
||
- **简单问答类型（simple）**：问题属于日常生活常识、通用知识、非技术性提问，**不涉及**装备、设备、机械、电子、维修、故障诊断、零部件、操作流程等专业领域内容。例如问候、自我介绍、天气、时间等。
|
||
- **复杂问答类型（complex）**：问题**涉及**装备、设备、机械系统（如发动机、液压系统）、电子装置、维修流程、故障排查、零部件组成、技术参数、操作规范等专业领域，即使只是询问定义、组成或功能，也视为 complex。
|
||
|
||
### 输出要求
|
||
- 仅输出 JSON 格式：{"query_type":"simple"} 或 {"query_type":"complex"}
|
||
- 不得包含任何其他文字、解释或格式。
|
||
|
||
### 示例
|
||
输入：今天心情怎么样
|
||
输出：{"query_type":"simple"}
|
||
|
||
输入：请问你是谁
|
||
输出：{"query_type":"simple"}
|
||
|
||
输入：发动机是什么
|
||
输出：{"query_type":"complex"}
|
||
|
||
输入：缸盖组包含哪些组件
|
||
输出：{"query_type":"complex"}
|
||
|
||
输入：手机屏幕碎了怎么办
|
||
输出：{"query_type":"complex"}
|
||
|
||
输入：水的沸点是多少
|
||
输出：{"query_type":"simple"}
|
||
|
||
输入：如何更换坦克履带
|
||
输出：{"query_type":"complex"}
|
||
|
||
### 用户查询：
|
||
text
|
||
"""
|
||
def open_api_chat(query: str, model: str = None):
|
||
"""
|
||
同步调用Chat API
|
||
|
||
Args:
|
||
query: 查询文本
|
||
model: 模型名称（可选，默认使用环境变量OPENAI_MODEL）
|
||
"""
|
||
if model is None:
|
||
model = "Qwen3.5-35B-A3B"
|
||
response = OpenAI(
|
||
api_key = "none",
|
||
base_url = "http://192.168.0.46:59800/v1"
|
||
).chat.completions.create(
|
||
model=model,
|
||
messages=[
|
||
{"role": "user", "content": query},
|
||
],
|
||
temperature=0.6,
|
||
stream=False,
|
||
response_format = {"type": "json_object"},
|
||
|
||
)
|
||
return response.choices[0].message.content
|
||
# querycontent = "发动机异响"
|
||
# query = query_type_prompt.replace("text",querycontent)
|
||
|
||
# res = open_api_chat(query=query)
|
||
# print(res)
|