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

Fix missing rag datatype

Browse Source
This commit is contained in:
Brandon Hancock
2024-07-29 15:09:59 -04:00
parent 53e91a7c78
commit 3d5bbf5da9
2 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/crewai/memory/storage/rag_storage.py
View File

@@ -5,12 +5,12 @@ import os
import shutil import shutil
from typing import Any, Dict, List, Optional from typing import Any, Dict, List, Optional
from embedchain import App
from embedchain.llm.base import BaseLlm
from embedchain.vectordb.chroma import InvalidDimensionException
from crewai.memory.storage.interface import Storage from crewai.memory.storage.interface import Storage
from crewai.utilities.paths import db_storage_path from crewai.utilities.paths import db_storage_path
from embedchain import App
from embedchain.llm.base import BaseLlm
from embedchain.models.data_type import DataType
from embedchain.vectordb.chroma import InvalidDimensionException
@contextlib.contextmanager @contextlib.contextmanager
@@ -101,8 +101,7 @@ class RAGStorage(Storage):
return [r for r in results if r["metadata"]["score"] >= score_threshold] return [r for r in results if r["metadata"]["score"] >= score_threshold]
def _generate_embedding(self, text: str, metadata: Dict[str, Any]) -> Any: def _generate_embedding(self, text: str, metadata: Dict[str, Any]) -> Any:
with suppress_logging(): self.app.add(text, data_type=DataType.TEXT, metadata=metadata)
self.app.add(text, data_type="text", metadata=metadata)
def reset(self) -> None: def reset(self) -> None:
try: try:

2
src/crewai/routers/pipeline_router.py
View File

@@ -1,4 +1,4 @@
from typing import Any, Callable, Dict, List, Tuple, Union from typing import Any, Callable, Dict, Tuple, Union
from pydantic import BaseModel, Field from pydantic import BaseModel, Field