Update app.py

app.py

@@ -46,24 +46,28 @@ def format_prompt(message, history):
 
 def call_search(purpose, task, history, action_input):
     return_list=[]
-    try:
-        if action_input != "" and action_input != None:
-            action_input.strip('""')
-            model_list = api.list_models(task=f"{action_input}")
-            this_obj = list(model_list)
-            print(f'THIS_OBJ :: {this_obj[0]}')
-            for i,eb in enumerate(this_obj):
-                #return_list.append(this_obj[i].id)
-                return_list.append(this_obj[i])
-            #json_obj=json.dumps(return_list, indent=4)
-            print (return_list)
-            history = "observation: the search results are:\n {}\n".format(return_list)
-        else: 
-            history = "observation: I need to trigger a search using the following syntax:\naction: SEARCH action_input SEARCH_QUERY"
-    except Exception as e:
-        print (e)
+    if action_input in tasks:
+        
+        try:
+            if action_input != "" and action_input != None:
+                action_input.strip('""')
+                model_list = api.list_models(task=f"{action_input}")
+                this_obj = list(model_list)
+                print(f'THIS_OBJ :: {this_obj[0]}')
+                for i,eb in enumerate(this_obj):
+                    #return_list.append(this_obj[i].id)
+                    return_list.append(this_obj[i])
+                #json_obj=json.dumps(return_list, indent=4)
+                print (return_list)
+                history = "observation: the search results are:\n {}\n".format(return_list)
+            else: 
+                history = "observation: I need to trigger a search using the following syntax:\naction: SEARCH action_input SEARCH_QUERY"
+        except Exception as e:
+            print (e)
+            history = "observation: The search query I used did not return a valid response, I should try a different query"
+    else:
         history = "observation: The search query I used did not return a valid response, I should try a different query"
-
+        
     return "MAIN", None, history, task