Update app.py

app.py

@@ -47,27 +47,27 @@ def format_prompt(message, history):
 def call_search(purpose, task, history, action_input):
     return_list=[]
     print (action_input)
-    if action_input in query.tasks:
-        print (trying)        
-        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 use a query from the list of tasks provide in the format 'text-generation'"
+    #if action_input in query.tasks:
+    print (trying)        
+    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 from the list of options"
+    #else:
+    #    history = "observation: The search query I used did not return a valid response"
         
     return "MAIN", None, history, task