@@ -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