Former-commit-id: a89f3c72f7e726f2a01f34fe2a66a45c89ceb1e4pull/160/head
parent
15bc3c1cb1
commit
bb127d769a
@ -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