Skip to content

Commit c308d86

Browse files
committed
Fix file naming for casing
1 parent 34ef437 commit c308d86

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mindsdb_sdk/agents.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -291,7 +291,7 @@ def add_files(self, name: str, file_paths: List[str], description: str, knowledg
291291
all_filenames = []
292292
for file_path in file_paths:
293293
filename = file_path.split('/')[-1]
294-
filename_no_extension = filename.split('.')[0]
294+
filename_no_extension = filename.split('.')[0].lower()
295295
all_filenames.append(filename_no_extension)
296296
try:
297297
_ = self.api.get_file_metadata(filename_no_extension)
@@ -306,14 +306,14 @@ def add_files(self, name: str, file_paths: List[str], description: str, knowledg
306306
if knowledge_base is not None:
307307
kb = self.knowledge_bases.get(knowledge_base)
308308
else:
309-
kb_name = f'{name.lower()}_{filename_no_extension.lower()}_{uuid4().hex}_kb'
309+
kb_name = f'{name.lower()}_{filename_no_extension}_{uuid4().hex}_kb'
310310
kb = self._create_default_knowledge_base(agent, kb_name)
311311

312312
# Insert the entire file.
313313
kb.insert_files(all_filenames)
314314

315315
# Make sure skill name is unique.
316-
skill_name = f'{filename_no_extension.lower()}_retrieval_skill_{uuid4().hex}'
316+
skill_name = f'{filename_no_extension}_retrieval_skill_{uuid4().hex}'
317317
retrieval_params = {
318318
'source': kb.name,
319319
'description': description,

0 commit comments

Comments
 (0)