andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 08:08:32 +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 typing import List, Union, Iterator
from pydantic import Field
from typing import Iterator, List, Union
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.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):