Skip to content

Commit 68ef33f

Browse files
committed
Align mta-trigger-test with working capability pattern
1 parent 655fb18 commit 68ef33f

1 file changed

Lines changed: 24 additions & 0 deletions

File tree

community/mta-trigger-test/main.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
from src.agent.capability import MatchingCapability
2+
from src.agent.capability_worker import CapabilityWorker
3+
from src.main import AgentWorker
4+
5+
6+
class MTATriggerTestCapability(MatchingCapability):
7+
worker: AgentWorker = None
8+
capability_worker: CapabilityWorker = None
9+
10+
#{{register capability}}
11+
12+
def call(self, worker: AgentWorker):
13+
self.worker = worker
14+
self.capability_worker = CapabilityWorker(self)
15+
self.worker.session_tasks.create(self.run())
16+
17+
async def run(self):
18+
try:
19+
self.worker.editor_logging_handler.info("MTA ability triggered")
20+
await self.capability_worker.speak(
21+
"MTA ability triggered successfully."
22+
)
23+
finally:
24+
self.capability_worker.resume_normal_flow()

0 commit comments

Comments
 (0)