ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-19 04:37:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

enhanced function calls

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-06-04 01:12:50 +02:00
parent 20c2e96351
commit 00e0c1e9f5
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
lollms/personality.py
View File

@ -3383,6 +3383,20 @@ The AI should respond in this format using data from actions_list:
return results
def transform_functions_to_text(self, functions):
function_texts = []
for func in functions:
function_text = f'Function: {func["function_name"]}\nDescription: {func["function_description"]}\nParameters:\n'
for param in func["function_parameters"]:
param_type = "string" if param["type"] == "str" else param["type"]
param_description = param.get("description", "")
function_text += f' - {param["name"]} ({param_type}): {param_description}\n'
function_texts.append(function_text.strip())
return "\n\n".join(function_texts)
def transform_functions(self, functions):
tools = []
@ -3429,7 +3443,7 @@ The AI should respond in this format using data from actions_list:
separator_template = self.config.separator_template
tools = self.transform_functions(functions)
tools = self.transform_functions_to_text(functions)
import copy
cd = copy.deepcopy(context_details)
function_descriptions = [