File size: 2,234 Bytes
e4fe5a9 64a2e26 e4fe5a9 8b2fc7e e4fe5a9 994ba37 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
import importlib
import importlib.util
import os
import pip
toolsImported = []
TOOLS_DIRECTORY = os.path.abspath("../tools")
class Tool:
def __init__(self, toolClass):
self.tool = toolClass()
self.inputSchema = self.tool.inputSchema
self.name = self.inputSchema["name"]
self.description = self.inputSchema["description"]
self.dependencies = self.tool.dependencies
for package in self.tool.dependencies:
try:
__import__(package.split('==')[0])
except ImportError:
print(f"Installing {package}")
if '==' in package:
package = package.split('==')[0]
pip.main(['install', package])
def run(self, query):
return self.tool.run(**query)
class ToolLoader:
def __init__(self):
self.toolsImported = []
self.load_tools()
pass
def load_tools(self):
for filename in os.listdir(TOOLS_DIRECTORY):
if filename.endswith(".py") and filename != "__init__.py":
module_name = filename[:-3]
spec = importlib.util.spec_from_file_location(module_name, f"{TOOLS_DIRECTORY}/{filename}")
foo = importlib.util.module_from_spec(spec)
spec.loader.exec_module(foo)
class_name = foo.__all__[0]
toolClass = getattr(foo, class_name)
toolObj = Tool(toolClass)
self.toolsImported.append(toolObj)
def runTool(self, toolName, query):
for tool in self.toolsImported:
if tool.name == toolName:
return tool.run(query)
return None
def getTools(self):
toolsList = []
for tool in self.toolsImported:
toolsList.append({
"type": "function",
"function": tool.inputSchema
})
return toolsList
toolLoader = ToolLoader()
# Example usage
# print(toolLoader.getTools())
# 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." })) |