修改引用逻辑,修改长期记忆bug
This commit is contained in:
@@ -8,24 +8,21 @@ import asyncio
|
||||
import logging
|
||||
from dataclasses import dataclass, field
|
||||
from pathlib import Path
|
||||
from typing import List, Union, Optional, Any, Dict, Tuple
|
||||
from typing import List, Union, Optional, Any, Dict
|
||||
|
||||
from httpx import RemoteProtocolError
|
||||
from langchain_core.documents import Document
|
||||
from langchain_core.embeddings import Embeddings
|
||||
from langchain_core.stores import BaseStore
|
||||
from langchain_text_splitters import RecursiveCharacterTextSplitter, TextSplitter
|
||||
from langchain_classic.retrievers import ParentDocumentRetriever
|
||||
from qdrant_client.http.exceptions import ResponseHandlingException
|
||||
|
||||
from .loaders import DocumentLoader
|
||||
from .splitters import SplitterType, get_splitter, SemanticChunkerAdapter
|
||||
from rag_indexer.loaders import DocumentLoader
|
||||
from rag_indexer.splitters import SplitterType, get_splitter
|
||||
from rag_core import LlamaCppEmbedder, QdrantVectorStore, create_docstore, create_parent_retriever
|
||||
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
# ---------- 配置数据类 ----------
|
||||
@dataclass
|
||||
class DocstoreConfig:
|
||||
@@ -36,7 +33,6 @@ class DocstoreConfig:
|
||||
# 若要从外部注入已创建好的 docstore,可直接设置此字段
|
||||
instance: Optional[BaseStore] = None
|
||||
|
||||
|
||||
@dataclass
|
||||
class IndexBuilderConfig:
|
||||
"""索引构建器配置。"""
|
||||
@@ -60,7 +56,6 @@ class IndexBuilderConfig:
|
||||
# 其他切分器参数(当 splitter_type 非父子块时使用)
|
||||
extra_splitter_kwargs: Dict[str, Any] = field(default_factory=dict)
|
||||
|
||||
|
||||
# ---------- 索引构建器 ----------
|
||||
class IndexBuilder:
|
||||
"""RAG 索引构建主流水线,支持单块切分与父子块切分。"""
|
||||
|
||||
Reference in New Issue
Block a user