feat: 添加清单定额查询API并优化意图识别模块
新增清单定额查询API服务,支持通过名称和编码查询定额及清单信息 在意图识别模块中添加定额清单信息提取功能,并记录各步骤耗时 将SiliconFlowEmbeddings替换为XinferenceEmbeddings并添加sqlite-vss依赖 优化shell脚本的screen会话检测逻辑
This commit is contained in:
@@ -14,7 +14,7 @@ import asyncio
|
||||
from typing import List, Dict, Any, Tuple, Optional
|
||||
from langchain.embeddings.base import Embeddings
|
||||
from langchain_community.vectorstores import FAISS
|
||||
from rag2_0.tool.ModelTool import SiliconFlowEmbeddings
|
||||
from rag2_0.tool.ModelTool import XinferenceEmbeddings
|
||||
import logging
|
||||
import httpx
|
||||
|
||||
@@ -28,7 +28,7 @@ def get_embedding_model(api_key: str = None) -> Embeddings:
|
||||
Returns:
|
||||
嵌入模型实例
|
||||
"""
|
||||
return SiliconFlowEmbeddings(api_key=api_key)
|
||||
return XinferenceEmbeddings(api_key=api_key)
|
||||
|
||||
|
||||
class ProfessionalNounVectorizer:
|
||||
|
||||
Reference in New Issue
Block a user