parent
276c923864
commit
108da13f3d
@ -1,15 +0,0 @@
|
||||
from swarms.tools.tool import BaseTool
|
||||
|
||||
|
||||
class EdgeGPTTool(BaseTool):
|
||||
def __init__(
|
||||
self,
|
||||
model,
|
||||
name="EdgeGPTTool",
|
||||
description="Tool that uses EdgeGPTModel to generate responses",
|
||||
):
|
||||
super().__init__(name=name, description=description)
|
||||
self.model = model
|
||||
|
||||
def _run(self, prompt):
|
||||
return self.model.__call__(prompt)
|
Loading…
Reference in new issue