YaTharThShaRma999
commited on
Commit
•
771209e
1
Parent(s):
a0f6bb7
Update llm_modules/llm.py
Browse files- 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=
|
|
|
|
|
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)
|