Update app.py

app.py

@@ -57,8 +57,19 @@ def call_search(purpose, task, history, action_input):
             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)
+                return_list.append({"id":this_obj[i].id,
+                                    "author":this_obj[i].author,
+                                    "created_at":this_obj[i].created_at,
+                                    "last_modified":this_obj[i].last_modified,
+                                    "private":this_obj[i].private,
+                                    "gated":this_obj[i].gated,
+                                    "disabled":this_obj[i].disabled,
+                                    "downloads":this_obj[i].downloads,
+                                    "likes":this_obj[i].likes,
+                                    "library_name":this_obj[i].library_name,
+                                    "tags":this_obj[i].tags,
+                                    "pipeline_tag":this_obj[i].pipeline_tag,
+                                   })
             print (return_list)
             history = "observation: the search results are:\n {}\n".format(return_list)
         else: 
@@ -182,12 +193,16 @@ def run_action(purpose, task, history, action_name, action_input):
         if VERBOSE:
             print("COMPRESSING HISTORY")
         history = compress_history(purpose, task, history)
+    if action_name in NAME_TO_FUNC:
+        
+        assert action_name in NAME_TO_FUNC
 
-    assert action_name in NAME_TO_FUNC
-
-    print("RUN: ", action_name, action_input)
-    return NAME_TO_FUNC[action_name](purpose, task, history, action_input)
+        print("RUN: ", action_name, action_input)
+        return NAME_TO_FUNC[action_name](purpose, task, history, action_input)
+    else:
+        history += "observation: The TOOL I tried to use returned an error, I need to select a tool from: (UPDATE-TASK, SEARCH, COMPLETE)
 
+        return "MAIN", None, history, task
 
 def run(purpose,history):
     task=None