diff --git a/core/embedding/base_embedding_model.py b/core/embedding/base_embedding_model.py index 95eed71..d5d3712 100644 --- a/core/embedding/base_embedding_model.py +++ b/core/embedding/base_embedding_model.py @@ -1,7 +1,7 @@ from abc import ABC, abstractmethod from typing import List, Union -from core.models.documents import Chunk +from core.models.chunk import Chunk class BaseEmbeddingModel(ABC): diff --git a/core/embedding/ollama_embedding_model.py b/core/embedding/ollama_embedding_model.py index a8118df..8ccec46 100644 --- a/core/embedding/ollama_embedding_model.py +++ b/core/embedding/ollama_embedding_model.py @@ -1,7 +1,7 @@ from typing import List, Union from ollama import AsyncClient from core.embedding.base_embedding_model import BaseEmbeddingModel -from core.models.documents import Chunk +from core.models.chunk import Chunk class OllamaEmbeddingModel(BaseEmbeddingModel): diff --git a/core/embedding/openai_embedding_model.py b/core/embedding/openai_embedding_model.py index 606ba73..5b7c6ea 100644 --- a/core/embedding/openai_embedding_model.py +++ b/core/embedding/openai_embedding_model.py @@ -1,7 +1,7 @@ from typing import List, Union from openai import OpenAI -from core.models.documents import Chunk +from core.models.chunk import Chunk from core.embedding.base_embedding_model import BaseEmbeddingModel diff --git a/core/parser/base_parser.py b/core/parser/base_parser.py index d5fcfe3..9318e6a 100644 --- a/core/parser/base_parser.py +++ b/core/parser/base_parser.py @@ -1,6 +1,6 @@ from abc import ABC, abstractmethod from typing import Any, Dict, List, Tuple -from core.models.documents import Chunk +from core.models.chunk import Chunk class BaseParser(ABC): diff --git a/core/parser/combined_parser.py b/core/parser/combined_parser.py index 055a186..9b3f29f 100644 --- a/core/parser/combined_parser.py +++ b/core/parser/combined_parser.py @@ -3,7 +3,7 @@ import logging import os import tempfile import magic -from core.models.documents import Chunk +from core.models.chunk import Chunk from core.parser.base_parser import BaseParser from core.parser.unstructured_parser import UnstructuredAPIParser diff --git a/core/parser/unstructured_parser.py b/core/parser/unstructured_parser.py index 91061c6..540ac12 100644 --- a/core/parser/unstructured_parser.py +++ b/core/parser/unstructured_parser.py @@ -2,7 +2,7 @@ from typing import Any, Dict, List, Tuple import io from langchain.text_splitter import RecursiveCharacterTextSplitter from langchain_unstructured import UnstructuredLoader -from core.models.documents import Chunk +from core.models.chunk import Chunk import logging from .base_parser import BaseParser diff --git a/core/services/document_service.py b/core/services/document_service.py index 4f21611..13c7bd7 100644 --- a/core/services/document_service.py +++ b/core/services/document_service.py @@ -3,10 +3,9 @@ from typing import Dict, Any, List, Optional from fastapi import UploadFile from core.models.request import IngestTextRequest +from core.models.chunk import Chunk, DocumentChunk from core.models.documents import ( - Chunk, Document, - DocumentChunk, ChunkResult, DocumentContent, DocumentResult, diff --git a/core/vector_store/base_vector_store.py b/core/vector_store/base_vector_store.py index 39a2314..3b32fdc 100644 --- a/core/vector_store/base_vector_store.py +++ b/core/vector_store/base_vector_store.py @@ -1,6 +1,6 @@ from abc import ABC, abstractmethod from typing import List, Optional, Tuple -from core.models.documents import DocumentChunk +from core.models.chunk import DocumentChunk class BaseVectorStore(ABC): diff --git a/core/vector_store/mongo_vector_store.py b/core/vector_store/mongo_vector_store.py index f840329..70fb43a 100644 --- a/core/vector_store/mongo_vector_store.py +++ b/core/vector_store/mongo_vector_store.py @@ -4,7 +4,7 @@ from motor.motor_asyncio import AsyncIOMotorClient from pymongo.errors import PyMongoError from .base_vector_store import BaseVectorStore -from core.models.documents import DocumentChunk +from core.models.chunk import DocumentChunk logger = logging.getLogger(__name__)