Skip to content

Commit 127aa0f

Browse files
updated add_webpages to avoid using skills
1 parent 430ec4d commit 127aa0f

File tree

1 file changed

+25
-25
lines changed

1 file changed

+25
-25
lines changed

mindsdb_sdk/agents.py

Lines changed: 25 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -372,15 +372,15 @@ def add_file(self, name: str, file_path: str, description: str, knowledge_base:
372372
self.add_files(name, [file_path], description, knowledge_base)
373373

374374
def add_webpages(
375-
self,
376-
name: str,
377-
urls: List[str],
378-
description: str,
379-
knowledge_base: str = None,
380-
crawl_depth: int = 1,
381-
limit: int = None,
382-
filters: List[str] = None
383-
):
375+
self,
376+
name: str,
377+
urls: List[str],
378+
description: str,
379+
knowledge_base: str = None,
380+
crawl_depth: int = 1,
381+
limit: int = None,
382+
filters: List[str] = None
383+
):
384384
"""
385385
Add a list of webpages to the agent for retrieval.
386386
@@ -407,25 +407,25 @@ def add_webpages(
407407
# Insert crawled webpage.
408408
kb.insert_webpages(urls, crawl_depth=crawl_depth, filters=filters, limit=limit)
409409

410-
# Make sure skill name is unique.
411-
skill_name = f'web_retrieval_skill_{uuid4().hex}'
412-
retrieval_params = {
413-
'source': kb.name,
414-
'description': description,
415-
}
416-
webpage_retrieval_skill = self.skills.create(skill_name, 'retrieval', retrieval_params)
417-
agent.skills.append(webpage_retrieval_skill)
410+
# Add knowledge base to agent's data if it hasn't been added yet.
411+
if 'knowledge_bases' not in agent.data or kb.name not in agent.data['knowledge_bases']:
412+
agent.data.setdefault('knowledge_bases', []).append(kb.name)
413+
414+
# Add the description provided to the agent's prompt template.
415+
agent.prompt_template = (agent.prompt_template or '') + f'\n{description}'
416+
418417
self.update(agent.name, agent)
419418

420419
def add_webpage(
421-
self,
422-
name: str,
423-
url: str,
424-
description: str,
425-
knowledge_base: str = None,
426-
crawl_depth: int = 1,
427-
limit: int = None,
428-
filters: List[str] = None):
420+
self,
421+
name: str,
422+
url: str,
423+
description: str,
424+
knowledge_base: str = None,
425+
crawl_depth: int = 1,
426+
limit: int = None,
427+
filters: List[str] = None
428+
):
429429
"""
430430
Add a webpage to the agent for retrieval.
431431

0 commit comments

Comments
 (0)