set agent system prompt in agent config (#85)

Author: TataKKKL

examples/atomic/IOAgent.py

@@ -21,8 +21,9 @@ def main():
         "agent_class": "FunctionCallingAgent",
         "meta_data":{},
         "tool_names": ["read_file", "write_to_file", "generate_and_download_image"],
-        "agent_description": "Read or write content from/to a file, or generate and save an image using text input",
-        "parameter_description": "The task description detailing what to read, write, or generate. This can include file operations or image generation requests."
+        "system_prompt": "You are an ai agent, Read or write content from/to a file, or generate and save an image using text input",
+        "agent_description": None,
+        "parameter_description": None,
     }
 
     system_config = {

examples/composite/MasterAgent.py

@@ -26,7 +26,8 @@ def main():
             "model_name": "gpt-4o-mini",
             "tool_choice": "auto"
         },
-        "tool_names": ["read_file", "write_to_file", "generate_and_download_image"],  # Changed from "write_file" to "write_to_file"
+        "tool_names": ["read_file", "write_to_file", "generate_and_download_image"],
+        "system_prompt": "You are an AI agent. Read or write content from/to a file, or generate and save an image using text input",
         "agent_description": "Read or write content from/to a file, or generate and save an image using text input",
         "parameter_description": "The task description detailing what to read, write, or generate. This can include file operations or image generation requests."
     }
@@ -36,7 +37,8 @@ def main():
         "type": "atomic",
         "agent_class": "FunctionCallingAgent",
         "meta_data": {},
-        "tool_names": ["retrieve_db"], # Changed from "write_file" to "write_to_file"
+        "tool_names": ["retrieve_db"],
+        "system_prompt": "You are an AI agent. Use a database retrieval agent to fetch information based on a given query.",
         "agent_description": "Use a database retrieval agent to fetch information based on a given query.",
         "parameter_description": "The query to be processed by the database retrieval agent."
     }
@@ -46,7 +48,8 @@ def main():
         "type": "atomic",
         "agent_class": "FunctionCallingAgent",
         "meta_data": {},
-        "tool_names": ["retrieve_file"],  # Changed from "write_file" to "write_to_file"
+        "tool_names": ["retrieve_file"],
+        "system_prompt": "You are an AI agent. Retrieve information from local documents to answer questions or perform tasks.",
         "agent_description": "Retrieve information from local documents to answer questions or perform tasks.",
         "parameter_description": "The task description specifying the local file and the question to be answered. specify this in natural language"
     }
@@ -56,7 +59,8 @@ def main():
         "type": "atomic",
         "agent_class": "FunctionCallingAgent",
         "meta_data": {},
-        "tool_names": ["bing_search", "scrape"],  # Changed from "write_file" to "write_to_file"
+        "tool_names": ["bing_search", "scrape"],
+        "system_prompt": "You are an AI agent. Perform a search using API and return the searched results.",
         "agent_description": "Perform a search using API and return the searched results.",
         "parameter_description": "The task description describing what to read or write."
     }
@@ -66,7 +70,8 @@ def main():
         "type": "atomic",
         "agent_class": "FunctionCallingAgent",
         "meta_data": {},
-        "tool_names": ["execute_shell_command", "run_python_script"],  # Changed from "write_file" to "write_to_file"
+        "tool_names": ["execute_shell_command", "run_python_script"],
+        "system_prompt": "You are an AI agent. Execute some script in a subprocess, either run a bash script, or run a python script ",
         "agent_description": "Execute some script in a subprocess, either run a bash script, or run a python script ",
         "parameter_description": "The task description describing what to execute in the subprocess."
     }
@@ -82,6 +87,7 @@ def main():
             file_retrieval_agent_config,
             db_retrieval_agent_config
         ],
+        "system_prompt": "You are an AI agent, Use a smart research assistant to look up information using multiple sources including web search, database retrieval, and local file retrieval.",
         "agent_description": "Use a smart research assistant to look up information using multiple sources including web search, database retrieval, and local file retrieval.",
         "parameter_description": "The task description specifying the information source (web search, database, local file) and the question to be answered. specify this in natural language"
     }
@@ -97,6 +103,7 @@ def main():
             exec_agent_config,
             io_agent_config,
         ],
+        "system_prompt": "You are an AI agent that leverage tools and sub agents",
         "agent_description": None,
         "parameter_description": None
     }

examples/new_tool/add_function_example.py

@@ -51,8 +51,9 @@ def calculate(operation, num1, num2):
         "meta_data": {},
         "tool_names": ["read_file", "write_to_file", "generate_and_download_image"],
         "self_defined_tools": [new_tool],
-        "agent_description": "test ai agent",
-        "parameter_description": "test ai agent"
+        "system_prompt": "You are an ai agent, we want to test using some tools",
+        "agent_description": None,
+        "parameter_description": None
     }
 
     system_config = {

examples/new_tool/add_llm_generated_function_example.py

@@ -66,8 +66,9 @@ def main():
         "meta_data": {},
         "tool_names": ["read_file", "write_to_file", "generate_and_download_image"],
         "self_defined_tools": new_tools,
-        "agent_description": "test ai agent",
-        "parameter_description": "test ai agent"
+        "system_prompt": "You are an ai agent, we want to test using some tools",
+        "agent_description": None,
+        "parameter_description": None
     }
 
     system_config = {

litemultiagent/agents/agent_class/base.py

@@ -51,13 +51,13 @@
         logger.error(f"Failed to initialize Supabase client: {e}")
 
 class BaseAgent:
-    def __init__(self, agent_name: str, agent_description, parameter_description, tools: List[Dict[str, Any]],
+    def __init__(self, agent_name: str, system_prompt, agent_description, parameter_description, tools: List[Dict[str, Any]],
                  available_tools: Dict[str, callable],
                  meta_data):
         self.agent_name = agent_name
         self.tools = tools
         self.available_tools = available_tools
-        self.messages = []
+        self.messages = [{"role": "system", "content": system_prompt}]
         self.model_name = meta_data.get("model_name", None)
         self.tool_choice = meta_data.get("tool_choice", None)
         self.agent_description = agent_description

litemultiagent/agents/agent_type/atomic.py

@@ -4,7 +4,7 @@
 
 
 class AtomicAgent:
-    def __init__(self, agent_name: str, agent_description: str, parameter_description: str,
+    def __init__(self, agent_name: str, system_prompt: str, agent_description: str, parameter_description: str,
                  tool_names: List[str], self_defined_tools, meta_data: Dict[str, Any],
                  agent_class: Type[BaseAgent]):
         tool_registry = ToolRegistry()
@@ -17,7 +17,7 @@ def __init__(self, agent_name: str, agent_description: str, parameter_descriptio
             available_tools[tool_name] = tool_registry.get_tool(tool_name).func
             tools.append(tool_registry.get_tool_description(tool_name))
 
-        self.agent = agent_class(agent_name, agent_description, parameter_description,
+        self.agent = agent_class(agent_name, system_prompt, agent_description, parameter_description,
                                  tools, available_tools, meta_data)
 
     def execute(self, task: str) -> str:

litemultiagent/agents/agent_type/composite.py

@@ -3,7 +3,7 @@
 from litemultiagent.tools.registry import ToolRegistry, Tool
 
 class CompositeAgent:
-    def __init__(self, agent_name: str, agent_description: str, parameter_description: str,
+    def __init__(self, agent_name: str, system_prompt: str, agent_description: str, parameter_description: str,
                  sub_agent_configs: List[Dict[str, Any]], tool_names: List[str], self_defined_tools, meta_data: Dict[str, Any],
                  agent_class: Type[BaseAgent]):
         self.available_tools = {}
@@ -20,7 +20,7 @@ def __init__(self, agent_name: str, agent_description: str, parameter_descriptio
         self.sub_agents = self._build_sub_agents(sub_agent_configs)
         self._register_sub_agents_as_tools()
 
-        self.agent = agent_class(agent_name, agent_description, parameter_description,
+        self.agent = agent_class(agent_name, system_prompt, agent_description, parameter_description,
                                  self.tools, self.available_tools, meta_data)
 
     def set_shared_config(self, shared_config):

litemultiagent/core/agent_factory.py

@@ -10,6 +10,7 @@ def create_agent(config: Dict[str, Any]) -> Union[AtomicAgent, CompositeAgent]:
         agent_type = config["type"]
         agent_name = config["name"]
         agent_class = config["agent_class"]
+        system_prompt = config["system_prompt"]
         agent_description = config["agent_description"]
         parameter_description = config["parameter_description"]
         tool_names = config.get("tool_names", [])
@@ -28,11 +29,11 @@ def create_agent(config: Dict[str, Any]) -> Union[AtomicAgent, CompositeAgent]:
         selected_agent_class = agent_class_map[agent_class]
 
         if agent_type == "atomic":
-            return AtomicAgent(agent_name, agent_description, parameter_description,
+            return AtomicAgent(agent_name, system_prompt, agent_description, parameter_description,
                                tool_names, self_defined_tools, meta_data, selected_agent_class)
         elif agent_type == "composite":
             sub_agent_configs = config.get("sub_agents", [])
-            return CompositeAgent(agent_name, agent_description, parameter_description,
+            return CompositeAgent(agent_name, system_prompt, agent_description, parameter_description,
                                   sub_agent_configs, tool_names, self_defined_tools, meta_data, selected_agent_class)
         else:
             raise ValueError(f"Unknown agent type: {agent_type}")