mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 00:28:31 +00:00
improvements
This commit is contained in:
@@ -30,7 +30,7 @@ class Knowledge(BaseModel):
|
|||||||
Logger(verbose=True).log(
|
Logger(verbose=True).log(
|
||||||
"warning",
|
"warning",
|
||||||
f"Failed to init knowledge: {e}",
|
f"Failed to init knowledge: {e}",
|
||||||
color="red",
|
color="yellow",
|
||||||
)
|
)
|
||||||
|
|
||||||
def query(
|
def query(
|
||||||
|
|||||||
@@ -1,11 +1,10 @@
|
|||||||
from abc import ABC, abstractmethod
|
from abc import ABC, abstractmethod
|
||||||
from typing import List
|
from typing import List, Dict, Any
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
from pydantic import BaseModel, ConfigDict, Field
|
from pydantic import BaseModel, ConfigDict, Field
|
||||||
|
|
||||||
from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage
|
from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage
|
||||||
from typing import Dict, Any
|
|
||||||
|
|
||||||
|
|
||||||
class BaseKnowledgeSource(BaseModel, ABC):
|
class BaseKnowledgeSource(BaseModel, ABC):
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
from typing import List
|
from typing import Dict, List
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
from crewai.knowledge.source.base_file_knowledge_source import BaseFileKnowledgeSource
|
from crewai.knowledge.source.base_file_knowledge_source import BaseFileKnowledgeSource
|
||||||
|
|
||||||
@@ -6,11 +7,11 @@ from crewai.knowledge.source.base_file_knowledge_source import BaseFileKnowledge
|
|||||||
class TextFileKnowledgeSource(BaseFileKnowledgeSource):
|
class TextFileKnowledgeSource(BaseFileKnowledgeSource):
|
||||||
"""A knowledge source that stores and queries text file content using embeddings."""
|
"""A knowledge source that stores and queries text file content using embeddings."""
|
||||||
|
|
||||||
def load_content(self) -> str:
|
def load_content(self) -> Dict[Path, str]:
|
||||||
"""Load and preprocess text file content."""
|
"""Load and preprocess text file content."""
|
||||||
super().load_content() # Validate the file path
|
super().load_content()
|
||||||
with self.file_path.open("r", encoding="utf-8") as f:
|
with self.file_path.open("r", encoding="utf-8") as f:
|
||||||
return f.read()
|
return f.read() # type: ignore
|
||||||
|
|
||||||
def add(self) -> None:
|
def add(self) -> None:
|
||||||
"""
|
"""
|
||||||
|
|||||||
Reference in New Issue
Block a user