ouyangyouzhang/QueryRewrite
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1、修改api文件位置

Browse Source 
2、意图识别继承langfuse
This commit is contained in:
ouyangyouzhang
2025-08-27 11:22:54 +08:00
parent 53ac47f4a5
commit c9c7f13060
12 changed files with 1385 additions and 1321 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rag2_0/dify/AnswerType.py → rag2_0/api/AnswerType_api.py
View File
rag2_0/dify/DifyQueryRetrieval_api.py → rag2_0/api/DifyQueryRetrieval_api.py
+5 -3
View File
@@ -44,6 +44,7 @@ class RetrieveRequest(BaseModel):
data_set_list: str
query_expand_dict: dict | str = Field(default="{}")
topk: int = Field(default=4)
metadata_filtering_conditions : dict = Field(default={})
# 创建FastAPI应用
app = FastAPI(
@@ -103,7 +104,8 @@ async def retrieve(request: RetrieveRequest):
query_list,
data_set_list,
query_expand_dict=query_expand_dict,
top_k=request.topk
top_k=request.topk,
metadata_filtering_conditions=request.metadata_filtering_conditions
)
end_time = time.time()
@@ -116,11 +118,11 @@ async def retrieve(request: RetrieveRequest):
if __name__ == "__main__":
# 使用Uvicorn运行FastAPI应用
import uvicorn
uvicorn.run("rag2_0.dify.DifyQueryRetrieval_api:app", host="0.0.0.0", port=8002, reload=False, workers=1, log_level="info")
uvicorn.run("rag2_0.api.DifyQueryRetrieval_api:app", host="0.0.0.0", port=9002, reload=False, workers=1, log_level="info")
# # 使用uvicorn启动服务
# import uvicorn
# uvicorn.run(
# "rag2_0.dify.intent_recognition_api:app",
# "rag2_0.api.DifyQueryRetrieval_api:app",
# host="0.0.0.0",
# port=8001,
# reload=False, # 开发环境启用热重载
rag2_0/dify/intent_recognition_api.py → rag2_0/api/intent_recognition_api.py
View File
rag2_0/dify/query_dinge_qingdan_api.py → rag2_0/api/query_dinge_qingdan_api.py
View File
rag2_0/dify/DifyQueryRetrieval.py
+14 -6
View File
@@ -23,12 +23,13 @@ class DifyQueryRetrieval:
datasets_json = datasets.json()
return {dataset["name"]:dataset for dataset in datasets_json["data"]}
def retrieve_by_dataset(self, query: str, dataset_name: str) -> Dict[str, Any]:
def retrieve_by_dataset(self, query: str, dataset_name: str, metadata_filtering_conditions:dict = {}) -> Dict[str, Any]:
try:
dataset_id = self._datasets_list[dataset_name]["id"]
retrieval_model = self._datasets_list[dataset_name]["retrieval_model_dict"]
knowledge_base_client = KnowledgeBaseClient(api_key=self._dify_dataset_key, base_url=self._dify_base_url, dataset_id=dataset_id)
if len(metadata_filtering_conditions) !=0:
retrieval_model["metadata_filtering_conditions"]=metadata_filtering_conditions
documents = knowledge_base_client.retrieve(query, retrieval_model=retrieval_model, timeout=300)
retrieved_documents = documents.json().get("records", [])
@@ -51,7 +52,7 @@ class DifyQueryRetrieval:
"documents": []
}
async def retrieve_by_dataset_async(self, query: str, dataset_name: str) -> Dict[str, Any]:
async def retrieve_by_dataset_async(self, query: str, dataset_name: str, metadata_filtering_conditions:dict = {}) -> Dict[str, Any]:
"""
异步版本的retrieve_by_dataset方法
@@ -67,7 +68,8 @@ class DifyQueryRetrieval:
return await asyncio.to_thread(
self.retrieve_by_dataset,
query,
dataset_name
dataset_name,
metadata_filtering_conditions
)
except Exception as e:
logging.error(f"异步检索数据集 {dataset_name} 时出错: {str(e)}", exc_info=True)
@@ -77,7 +79,13 @@ class DifyQueryRetrieval:
"documents": []
}
async def retrieve_api_async(self, original_query: str, query_list: List[str], data_set_list: List[str], query_expand_dict: dict, top_k: int = 5)->Dict[str, Any]:
async def retrieve_api_async(self,
original_query: str,
query_list: List[str],
data_set_list: List[str],
query_expand_dict: dict,
top_k: int = 5,
metadata_filtering_conditions:dict = {})->Dict[str, Any]:
"""
异步版本的retrieve_api方法，使用asyncio代替线程池
@@ -105,7 +113,7 @@ class DifyQueryRetrieval:
continue
# 创建异步任务
task = self.retrieve_by_dataset_async(query, dataset)
task = self.retrieve_by_dataset_async(query, dataset, metadata_filtering_conditions)
tasks.append(task)
# 并发执行所有异步任务
rag2_0/intent_recognition/IntentRecognition.py
+76 -25
View File
@@ -19,6 +19,8 @@ import jieba
import time
import threading
from langchain_openai import ChatOpenAI
from .PromptTemplates import (classification_prompt, query_rewrite_prompt_pro,
extract_nouns_prompt, classification_info,
slot_filling_prompt, step_back_prompt,
@@ -32,7 +34,10 @@ from .DataModels import (
StepBackPrompt, HypotheticalDocument
)
from .ProfessionalNounVector import ProfessionalNounRetriever, AsyncProfessionalNounRetriever
from rag2_0.tool.ModelTool import XinferenceReRankerModel, OpenAiLLM
from rag2_0.tool.APIKeyManager import APIKeyManager
TEMPERATURE = 0.4
TOP_P = 0.7
class AsyncIntentRecognizer:
SOFT_WIKI_PATH = "data/wiki_data"
@@ -59,21 +64,6 @@ class AsyncIntentRecognizer:
model_name: 要使用的模型名称
vector_index_dir: 向量索引目录，如果为None则使用默认目录
"""
api_key = os.getenv("OPENAI_API_KEY")
base_url = os.getenv("OPENAI_API_BASE")
model_name = os.getenv("MODEL_NAME", "gpt-3.5-turbo")
# 初始化LLM
llm_params = {
"temperature": 0.4, # 降低随机性，使结果更确定
"top_p": 0.7,
"model": model_name,
"api_key": api_key,
"base_url": base_url
}
self._llm = OpenAiLLM(**llm_params)
llm_params["model"] = os.getenv("MINI_MODEL_NAME", "gpt-3.5-turbo")
self._llm_mini = OpenAiLLM(**llm_params)
# 加载suffix关键词
self._suffix_keywords = self._load_suffix_keywords()
@@ -199,7 +189,15 @@ class AsyncIntentRecognizer:
# 解析输出
try:
# 异步调用LLM
response = await self._llm.invoke_async(formatted_prompt, False)
llm = ChatOpenAI(
api_key=APIKeyManager.get_api_key(),
openai_api_base=os.getenv("OPENAI_API_BASE"),
model_name=os.getenv("MODEL_NAME"),
temperature=TEMPERATURE,
top_p=TOP_P
)
llm.with_structured_output(Classification)
response = await llm.ainvoke(formatted_prompt)
# 尝试直接解析JSON响应
response.content = response.content.strip()
@@ -266,8 +264,17 @@ class AsyncIntentRecognizer:
terms_list_parser = PydanticOutputParser(pydantic_object=TermList)
formatted_prompt = formatted_prompt.replace("{output_format}", terms_list_parser.get_format_instructions())
llm = ChatOpenAI(
api_key=APIKeyManager.get_api_key(),
openai_api_base=os.getenv("OPENAI_API_BASE"),
model_name=os.getenv("MODEL_NAME"),
temperature=TEMPERATURE,
top_p=TOP_P
)
llm.with_structured_output(TermList)
# 异步调用LLM
response = await self._llm.invoke_async(formatted_prompt, False)
response = await llm.ainvoke(formatted_prompt)
# 尝试使用Pydantic解析器解析TermList
response.content = response.content.strip()
@@ -349,7 +356,16 @@ class AsyncIntentRecognizer:
"""
try:
response = await self._llm.invoke_async(prompt, False, response_format={"type": "json_object"})
llm = ChatOpenAI(
api_key=APIKeyManager.get_api_key(),
openai_api_base=os.getenv("OPENAI_API_BASE"),
model_name=os.getenv("MODEL_NAME"),
temperature=TEMPERATURE,
top_p=TOP_P
)
response = await llm.ainvoke(prompt, response_format={"type": "json_object"})
response.content = response.content.strip()
clean_output = re.sub(r'<think>.*?</think>', '', response.content, flags=re.DOTALL)
parsed_output = JsonOutputParser().parse(clean_output)
@@ -389,8 +405,17 @@ class AsyncIntentRecognizer:
context=context)
# 解析输出
try:
llm = ChatOpenAI(
api_key=APIKeyManager.get_api_key(),
openai_api_base=os.getenv("OPENAI_API_BASE"),
model_name=os.getenv("MODEL_NAME"),
temperature=TEMPERATURE,
top_p=TOP_P
)
llm.with_structured_output(QueryRewrite)
# 异步调用LLM
response = await self._llm.invoke_async(formatted_prompt, False)
response = await llm.ainvoke(formatted_prompt)
response.content = response.content.strip()
clean_output = re.sub(r'<think>.*?</think>', '', response.content, flags=re.DOTALL)
parsed_output = query_rewrite_parser.parse(clean_output)
@@ -634,8 +659,18 @@ class AsyncIntentRecognizer:
previous_slots=json.dumps(previous_slots,ensure_ascii=False),
)
try:
llm = ChatOpenAI(
api_key=APIKeyManager.get_api_key(),
openai_api_base=os.getenv("OPENAI_API_BASE"),
model_name=os.getenv("MODEL_NAME"),
temperature=TEMPERATURE,
top_p=TOP_P
)
llm.with_structured_output(slot_model_class)
# 异步调用LLM
response = await self._llm.invoke_async(formatted_prompt, False)
response = await llm.ainvoke(formatted_prompt)
response.content = response.content.strip()
clean_output = re.sub(r'<think>.*?</think>', '', response.content, flags=re.DOTALL)
# 尝试解析LLM响应
@@ -669,10 +704,17 @@ class AsyncIntentRecognizer:
)
try:
llm = ChatOpenAI(
api_key=APIKeyManager.get_api_key(),
openai_api_base=os.getenv("OPENAI_API_BASE"),
model_name=os.getenv("MODEL_NAME"),
temperature=TEMPERATURE,
top_p=TOP_P
)
llm.with_structured_output(StepBackPrompt)
# 异步调用LLM
response = await self._llm.invoke_async(formatted_prompt, False)
# 解析输出
response = await llm.ainvoke(formatted_prompt)
response.content = response.content.strip()
clean_output = re.sub(r'<think>.*?</think>', '', response.content, flags=re.DOTALL)
parsed_output = step_back_parser.parse(clean_output)
@@ -728,9 +770,18 @@ class AsyncIntentRecognizer:
"""
try:
llm = ChatOpenAI(
api_key=APIKeyManager.get_api_key(),
openai_api_base=os.getenv("OPENAI_API_BASE"),
model_name=os.getenv("MODEL_NAME"),
temperature=TEMPERATURE,
top_p=TOP_P,
)
# 异步调用LLM
start_time = time.time()
response = await self._llm.invoke_async(prompt, False, response_format={"type": "json_object"})
response = await llm.ainvoke(prompt, response_format={"type": "json_object"})
end_time = time.time()
# 解析JSON响应