Coverage for apps/inners/models/dtos/document_category.py: 90%

1from typing import List 

2 

3from langchain_core.documents import Document 

4 

5from apps.inners.models.base_model import BaseModel 

6 

7 

8class DocumentCategory(BaseModel): 

9 texts: List[Document] 

10 tables: List[Document] 

11 images: List[Document] 

12 id_key: str 

13 

14 def get_all(self) -> List[Document]: 

15 return self.texts + self.tables + self.images