Skip to content

Commit e650671

Browse files
updated error message when auto-creating KB
1 parent e6dce77 commit e650671

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

mindsdb_sdk/agents.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -304,14 +304,13 @@ def completion_stream_v2(self, name, messages: List[dict]) -> Iterable[object]:
304304

305305
def _create_default_knowledge_base(self, agent: Agent, name: str) -> KnowledgeBase:
306306
try:
307-
# TODO: Use the agent's model credentials?
308-
# The model used will not be an embedding model though.
309307
kb = self.knowledge_bases.create(name)
310308
return kb
311309
except Exception as e:
312310
raise ValueError(
313-
f'Failed to automatically create knowledge base for agent {agent.name}. '
314-
"Please set your default embedding model in MindsDB settings or provide an existing knowledge base name."
311+
f"Failed to automatically create knowledge base for agent {agent.name}. "
312+
"Either provide an existing knowledge base name, "
313+
"or set your default embedding model via server.config.set_default_embedding_model(...) or through the MindsDB UI."
315314
)
316315

317316
def add_files(self, name: str, file_paths: List[str], description: str = None):

0 commit comments

Comments
 (0)