mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-05-01 23:32:39 +00:00
Incorporate @lorenzejay feedback
This commit is contained in:
@@ -10,6 +10,7 @@ from chromadb.api import ClientAPI
|
|||||||
|
|
||||||
from crewai.memory.storage.base_rag_storage import BaseRAGStorage
|
from crewai.memory.storage.base_rag_storage import BaseRAGStorage
|
||||||
from crewai.utilities import EmbeddingConfigurator
|
from crewai.utilities import EmbeddingConfigurator
|
||||||
|
from crewai.utilities.constants import MAX_FILE_NAME_LENGTH
|
||||||
from crewai.utilities.paths import db_storage_path
|
from crewai.utilities.paths import db_storage_path
|
||||||
|
|
||||||
|
|
||||||
@@ -91,20 +92,11 @@ class RAGStorage(BaseRAGStorage):
|
|||||||
"""
|
"""
|
||||||
base_path = f"{db_storage_path()}/{type}"
|
base_path = f"{db_storage_path()}/{type}"
|
||||||
|
|
||||||
try:
|
if len(file_name) > MAX_FILE_NAME_LENGTH:
|
||||||
# Returns platform-dependent max length for a file name
|
|
||||||
max_length = os.pathconf(base_path, "PC_NAME_MAX")
|
|
||||||
except (OSError, AttributeError) as e:
|
|
||||||
logging.error(f"Error accessing path configuration: {e}")
|
|
||||||
# Fallback to a reasonable default if necessary
|
|
||||||
max_length = 255
|
|
||||||
|
|
||||||
# Trim if necessary
|
|
||||||
if len(file_name) > max_length:
|
|
||||||
logging.warning(
|
logging.warning(
|
||||||
f"Trimming file name from {len(file_name)} to {max_length} characters."
|
f"Trimming file name from {len(file_name)} to {MAX_FILE_NAME_LENGTH} characters."
|
||||||
)
|
)
|
||||||
file_name = file_name[:max_length]
|
file_name = file_name[:MAX_FILE_NAME_LENGTH]
|
||||||
|
|
||||||
return f"{base_path}/{file_name}"
|
return f"{base_path}/{file_name}"
|
||||||
|
|
||||||
|
|||||||
@@ -2,3 +2,4 @@ TRAINING_DATA_FILE = "training_data.pkl"
|
|||||||
TRAINED_AGENTS_DATA_FILE = "trained_agents_data.pkl"
|
TRAINED_AGENTS_DATA_FILE = "trained_agents_data.pkl"
|
||||||
DEFAULT_SCORE_THRESHOLD = 0.35
|
DEFAULT_SCORE_THRESHOLD = 0.35
|
||||||
KNOWLEDGE_DIRECTORY = "knowledge"
|
KNOWLEDGE_DIRECTORY = "knowledge"
|
||||||
|
MAX_FILE_NAME_LENGTH = 255
|
||||||
|
|||||||
Reference in New Issue
Block a user