@@ -245,38 +245,59 @@ def prompt_agent(
245245 except Exception as e :
246246 return self .handle_error (e )
247247
248- def instruct (self , agent_name : str , prompt : str ) -> str :
248+ def instruct (self , agent_name : str , user_input : str , conversation : str ) -> str :
249249 return self .prompt_agent (
250250 agent_name = agent_name ,
251251 prompt_name = "instruct" ,
252- prompt_args = {"user_input" : prompt },
252+ prompt_args = {
253+ "user_input" : user_input ,
254+ "disable_memory" : True ,
255+ "conversation_name" : conversation ,
256+ },
253257 )
254258
255- def chat (self , agent_name : str , prompt : str ) -> str :
259+ def chat (
260+ self ,
261+ agent_name : str ,
262+ user_input : str ,
263+ conversation : str ,
264+ context_results : int = 4 ,
265+ ) -> str :
256266 return self .prompt_agent (
257267 agent_name = agent_name ,
258268 prompt_name = "Chat" ,
259- prompt_args = {"user_input" : prompt , "context_reslts" : 4 },
269+ prompt_args = {
270+ "user_input" : user_input ,
271+ "context_results" : context_results ,
272+ "conversation_name" : conversation ,
273+ "disable_memory" : True ,
274+ },
260275 )
261276
262- def smartinstruct (self , agent_name : str , prompt : str ) -> str :
277+ def smartinstruct (self , agent_name : str , user_input : str , conversation : str ) -> str :
263278 return self .run_chain (
264279 chain_name = "Smart Instruct" ,
265- user_input = prompt ,
280+ user_input = user_input ,
266281 agent_name = agent_name ,
267282 all_responses = False ,
268283 from_step = 1 ,
269- chain_args = {},
284+ chain_args = {
285+ "conversation_name" : conversation ,
286+ "disable_memory" : True ,
287+ },
270288 )
271289
272- def smartchat (self , agent_name : str , prompt : str ) -> str :
290+ def smartchat (self , agent_name : str , user_input : str , conversation : str ) -> str :
273291 return self .run_chain (
274292 chain_name = "Smart Chat" ,
275- user_input = prompt ,
293+ user_input = user_input ,
276294 agent_name = agent_name ,
277295 all_responses = False ,
278296 from_step = 1 ,
279- chain_args = {},
297+ chain_args = {
298+ "conversation_name" : conversation ,
299+ "disable_memory" : True ,
300+ },
280301 )
281302
282303 def get_commands (self , agent_name : str ) -> Dict [str , Dict [str , bool ]]:
0 commit comments