上传问题改写、意图识别模块代码

rag2_0/intent_recognition/DataModels.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+File: DataModels.py
+Author: oyyz
+Date: 2025-05-13
+Description: 提取和分类的数据模型
+"""
+
+from pydantic import BaseModel, Field
+from typing import List, Optional
+
+
+# 定义输出模型
+class Term(BaseModel):
+    name: str = Field(description="专业名词")
+    synonymous: List[str] = Field(description="同义词列表")
+    description: str = Field(description="描述信息", default="")
+    
+    def __hash__(self):
+        return hash(self.name)
+    
+    def __eq__(self, other):
+        if isinstance(other, Term):
+            return self.name == other.name
+        return False
+
+class TermList(BaseModel):
+    terms: List[Term] = Field(description="专业名词列表")
+
+class Classification(BaseModel):
+    vertical_classification:str = Field(description="垂直领域一级分类")
+    sub_classification:str = Field(description="一级分类下的二级分类")
+
+class QueryRewrite(BaseModel):
+    rewrite:str = Field(description="问题改写")