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)
