Rename imports (#9)

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):

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):

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
 
 

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):

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

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

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,

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):

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__)