31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
"""
|
|
向量化
|
|
"""
|
|
from typing import List, Dict, Any
|
|
|
|
from app.chat_qa_query.query_qa_state import QueryQAState
|
|
from app.client.embedding_client_manager import embedding_client
|
|
from app.repository.milvus.intent_classification_repository import IntentClassificationRepository
|
|
from app.repository.milvus.message_qa_repository import QARepository
|
|
from app.core.log import logger
|
|
|
|
|
|
async def search_embed(state: QueryQAState):
|
|
logger.info("search_embed")
|
|
rewritten_query: List[str] = state.get("rewritten_query", "")
|
|
original_query = state.get("original_query", "")
|
|
|
|
#batch_embeddings = await embedding_client.aembed_query(rewritten_query)
|
|
intent_class = IntentClassificationRepository()
|
|
result: List[Dict[str, Any]] = await intent_class.search_by_intent_category(intent_category=rewritten_query, top_k=3)
|
|
logger.info(f"原始问题:{original_query},改写后的问题：{rewritten_query},匹配结果：{result}")
|
|
|
|
qa_pairs = []
|
|
for item in result:
|
|
qa_pairs.append({
|
|
"intent_category": item.get("intent_category", ""),
|
|
"content": item.get("content", "") + "," +item.get("review_content", ""),
|
|
"score": item.get("score", 0)
|
|
})
|
|
|
|
return {"qa_pairs": qa_pairs} |