From 50a29297eb0eb2e522e7c296c1948752a76bc211 Mon Sep 17 00:00:00 2001
From: yhjun1026 <460342015@qq.com>
Date: Tue, 26 Dec 2023 10:23:55 +0800
Subject: [PATCH] feat(Agent): plugin_assistant_agent

1.add plugin assistant agent, support use outside tool
---
 .../agents/expand/plugin_assistant_agent.py    | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/dbgpt/agent/agents/expand/plugin_assistant_agent.py b/dbgpt/agent/agents/expand/plugin_assistant_agent.py
index 2b5a81122..a6f55d788 100644
--- a/dbgpt/agent/agents/expand/plugin_assistant_agent.py
+++ b/dbgpt/agent/agents/expand/plugin_assistant_agent.py
@@ -30,8 +30,22 @@ class PluginAgent(ConversableAgent):
     """
 
     DEFAULT_SYSTEM_MESSAGE = """
-    
-    
+    You are a useful artificial intelligence tool agent assistant.
+    You have been assigned the following list of tools, please select the most appropriate tool to complete the task based on the current user's goals:
+        {tool_list}
+        
+    *** IMPORTANT REMINDER ***
+    Please read the parameter definition of the tool carefully and extract the specific parameters required to execute the tool from the user gogal.
+    Please output the selected tool name and specific parameter information in json in the following required format, refer to the following example:
+        user: Search for the latest hot financial news
+        assisant: {{
+          "tool_name":"The chart rendering method currently selected by SQL",
+          "args": "{{
+            "query": "latest hot financial news",
+          }}",
+          "thought":"I will use the google-search tool to search for the latest hot financial news."
+        }}
+      
       Please think step by step and return it in the following json format
       {{
           "tool_name":"The chart rendering method currently selected by SQL",