Harshil Patel commited on
Commit
64a2e26
·
1 Parent(s): 40afc88

Add tool to create AI agent

Browse files

- For now the tool can only create an ollama agent.

Files changed (3) hide show
  1. .gitignore +1 -1
  2. CEO/toolLoader.py +3 -3
  3. tools/agent_creater_tool.py +74 -0
.gitignore CHANGED
@@ -1,4 +1,4 @@
1
- venv/
2
  __pycache__/
3
  *.pyc
4
  .env
 
1
+ *venv/
2
  __pycache__/
3
  *.pyc
4
  .env
CEO/toolLoader.py CHANGED
@@ -5,7 +5,7 @@ import pip
5
 
6
  toolsImported = []
7
 
8
- TOOLS_DIRECTORY = 'D:/Projects/AI/HASHIRU/tools'
9
 
10
  class Tool:
11
  def __init__(self, toolClass):
@@ -62,5 +62,5 @@ class ToolLoader:
62
  toolLoader = ToolLoader()
63
 
64
  # Example usage
65
- # print(toolLoader.getTools())
66
- # print(toolLoader.runTool("WeatherApi", {"location": "Davis"}))
 
5
 
6
  toolsImported = []
7
 
8
+ TOOLS_DIRECTORY = os.path.abspath("../tools")
9
 
10
  class Tool:
11
  def __init__(self, toolClass):
 
62
  toolLoader = ToolLoader()
63
 
64
  # Example usage
65
+ print(toolLoader.getTools())
66
+ print(toolLoader.runTool("AgentCreator", {"agent_name": "Kunla","base_model":"llama3.2","system_prompt": "You love making the indian dish called Kulcha. You declare that in every conversation you have in a witty way." }))
tools/agent_creater_tool.py ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import importlib
2
+
3
+ __all__ = ['AgentCreator']
4
+
5
+ class AgentCreator():
6
+ dependencies = ["ollama==0.4.7",
7
+ "pydantic==2.11.1",
8
+ "pydantic_core==2.33.0"]
9
+
10
+ inputSchema = {
11
+ "name": "AgentCreator",
12
+ "description": "Creates an AI agent for an given AI model with a given system prompt",
13
+ "parameters": {
14
+ "type": "object",
15
+ "properties":{
16
+ "agent_name": {
17
+ "type": "string",
18
+ "description": "Name of the AI agent that is to be created"
19
+ },
20
+ "base_model": {
21
+ "type": "string",
22
+ "description": "A base model from which the new agent mode is to be created"
23
+ },
24
+ "system_prompt": {
25
+ "type": "string",
26
+ "description": "A string containing the system prompt for the AI agent"
27
+ }
28
+ }
29
+ }
30
+ }
31
+
32
+ def __init__(self):
33
+ pass
34
+
35
+ def does_agent_exist(self, agent_name):
36
+ ollama = importlib.import_module("ollama")
37
+ all_agents = [a.model for a in ollama.list().models]
38
+ if agent_name in all_agents or f'{agent_name}:latest' in all_agents:
39
+ return True
40
+
41
+ return False
42
+
43
+ def run(self, **kwargs):
44
+ print("Running Agent Creator")
45
+ agent_name= kwargs.get("agent_name")
46
+ base_model = kwargs.get("base_model")
47
+ system_prompt = kwargs.get("system_prompt")
48
+ ollama = importlib.import_module("ollama")
49
+
50
+ if self.does_agent_exist(agent_name):
51
+ return {
52
+ "status": "error",
53
+ "message": "Agent already exists",
54
+ "output": None
55
+ }
56
+ ollama_response = ollama.create(
57
+ model = agent_name,
58
+ from_ = base_model,
59
+ system = system_prompt,
60
+ stream = False
61
+ )
62
+
63
+ if "success" in ollama_response["status"]:
64
+ return {
65
+ "status": "success",
66
+ "message": "Agent successfully created",
67
+ "output": None
68
+ }
69
+ else:
70
+ return {
71
+ "status": "error",
72
+ "message": "Agent creation failed",
73
+ "output": None
74
+ }