andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix import

Browse Source
This commit is contained in:
Lorenze Jay
2024-12-15 22:23:17 -08:00
parent a67ec7e37a
commit ee74ad0d6d
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/crewai/knowledge/source/docling_source.py
View File

@@ -1,14 +1,15 @@
from pathlib import Path from pathlib import Path
from typing import List, Union, Iterator from typing import Iterator, List, Union
from pydantic import Field
from urllib.parse import urlparse from urllib.parse import urlparse
from docling.datamodel.base_models import InputFormat
from docling.document_converter import DocumentConverter
from docling_core.transforms.chunker.hierarchical_chunker import HierarchicalChunker
from docling_core.types.doc.document import DoclingDocument
from pydantic import Field
from crewai.knowledge.source.base_file_knowledge_source import BaseFileKnowledgeSource from crewai.knowledge.source.base_file_knowledge_source import BaseFileKnowledgeSource
from crewai.utilities.constants import KNOWLEDGE_DIRECTORY from crewai.utilities.constants import KNOWLEDGE_DIRECTORY
from docling_core.transforms.chunker.hierarchical_chunker import HierarchicalChunker
from docling.document_converter import DocumentConverter
from docling_core.types.doc.document import DoclingDocument
from docling.datamodel.base_models import InputFormat
class DoclingSource(BaseFileKnowledgeSource): class DoclingSource(BaseFileKnowledgeSource):