mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-11 17:18:29 +00:00
22 lines
595 B
Python
22 lines
595 B
Python
from typing import List, Optional
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from .embedder.base_embedder import BaseEmbedder
|
|
from .embedder.fastembed import FastEmbed
|
|
from .source.base_knowledge_source import BaseKnowledgeSource
|
|
|
|
|
|
class Knowledge(BaseModel):
|
|
sources: Optional[List[BaseKnowledgeSource]] = None
|
|
embedder: BaseEmbedder
|
|
|
|
def __init__(
|
|
self,
|
|
sources: Optional[List[BaseKnowledgeSource]] = None,
|
|
embedder: Optional[BaseEmbedder] = None,
|
|
):
|
|
super().__init__()
|
|
self.sources = sources or []
|
|
self.embedder = embedder or FastEmbed()
|