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

2023-07-27 17:39:36 +08:00
parent 741e9303d4
commit b5825142d1
6 changed files with 40 additions and 21 deletions
--- a/api/controllers/console/universal_chat/chat.py
+++ b/api/controllers/console/universal_chat/chat.py
@@ -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']