diff --git a/modules/llm.py b/modules/llm.py index 183c2f6..5be12f8 100644 --- a/modules/llm.py +++ b/modules/llm.py @@ -146,6 +146,10 @@ def llm_query(input, nodeID=0, location_name=None): googleResults = [] if not location_name: location_name = "no location provided " + + # remove askai: and ask: from the input + if input.startswith(trap_list_llm): + input = input.split(":", 1)[1].strip() # add the naughty list here to stop the function before we continue # add a list of allowed nodes only to use the function