jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add current time tool in universal chat agent (#659)

Browse source
This commit is contained in:
John Wang 2023-07-27 17:39:36 +08:00 • committed by GitHub
commit b5825142d1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 40 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

15
api/controllers/console/universal_chat/chat.py
View file

@ -43,6 +43,21 @@ class UniversalChatApi(UniversalChatResource):
args['model_config']['model']['provider'] = llm_constant.models[args['model']]
args['model_config']['agent_mode']['tools'] = args['tools']
if not args['model_config']['agent_mode']['tools']:
args['model_config']['agent_mode']['tools'] = [
{
"current_datetime": {
"enabled": True
}
}
]
else:
args['model_config']['agent_mode']['tools'].append({
"current_datetime": {
"enabled": True
}
})
args['inputs'] = {}
del args['model']