allow for model to train on the search agent experience (rag cache, knowledge base, search data)

main_menu.py

@@ -89,7 +89,7 @@ def main():
 
         # Construct sys.argv for lightbulb_inf.py
         lightbulb_inf_args = [
-            'lightbulb_inf.py',
+            'lightbulb_custom.py',
             '--mode', 'inference',
             '--model_name', args.model_name,
             '--query', args.query,