YaTharThShaRma999 commited on
Commit
771209e
1 Parent(s): a0f6bb7

Update llm_modules/llm.py

Browse files
Files changed (1) hide show
  1. llm_modules/llm.py +3 -1
llm_modules/llm.py CHANGED
@@ -82,7 +82,9 @@ Final_response: your final response."""
82
  prompt = self.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
83
  prompt += "Action:"
84
  return str(prompt)
85
- def process_llm(self, prompt, function_executer, functions=self.default_function, personality="kind, happy, nice, perfect", name="Hermes", previous_prompts=None, previous_answers=None):
 
 
86
  functions_list = functions.split("\n\n")
87
  results = self.similar.infer([prompt], functions_list, top=5)
88
  similar_functions = "" ## string variable to store the most similar functions(5)
 
82
  prompt = self.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
83
  prompt += "Action:"
84
  return str(prompt)
85
+ def process_llm(self, prompt, function_executer, functions='default', personality="kind, happy, nice, perfect", name="Hermes", previous_prompts=None, previous_answers=None):
86
+ if functions == 'default':
87
+ functions = self.default_function
88
  functions_list = functions.split("\n\n")
89
  results = self.similar.infer([prompt], functions_list, top=5)
90
  similar_functions = "" ## string variable to store the most similar functions(5)