Skip to content

Commit 482d1a4

Browse files
authored
Merge pull request #157 from mindsdb/create_mind-fix
Use Lower Case Naming for File Upload & KB Creation
2 parents 4a6f725 + 07a3388 commit 482d1a4

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

mindsdb_sdk/agents.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -290,7 +290,7 @@ def add_files(self, name: str, file_paths: List[str], description: str, knowledg
290290
filename_no_extension = ''
291291
all_filenames = []
292292
for file_path in file_paths:
293-
filename = file_path.split('/')[-1]
293+
filename = file_path.split('/')[-1].lower()
294294
filename_no_extension = filename.split('.')[0]
295295
all_filenames.append(filename_no_extension)
296296
try:
@@ -306,7 +306,7 @@ 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}_{filename_no_extension}_{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.
@@ -361,7 +361,7 @@ def add_webpages(
361361
if knowledge_base is not None:
362362
kb = self.knowledge_bases.get(knowledge_base)
363363
else:
364-
kb_name = f'{name}_web_{uuid4().hex}_kb'
364+
kb_name = f'{name.lower()}_web_{uuid4().hex}_kb'
365365
kb = self._create_default_knowledge_base(agent, kb_name)
366366

367367
# Insert crawled webpage.

mindsdb_sdk/utils/mind.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,8 @@ def create_mind(
9595

9696
url = f"{base_url.rstrip('/')}/minds"
9797
headers = {"Authorization": f"Bearer {api_key}"}
98+
if data_source_configs is None:
99+
data_source_configs = []
98100
payload = {
99101
"name": name,
100102
"data_source_configs": [d.model_dump() for d in data_source_configs],

0 commit comments

Comments
 (0)