ai-workflows 0.0.1 → 0.0.2

package/README.md

@@ -10,5 +10,9 @@ workflow.on('ticket.created', ticket => {
   workflow.send('ticket.summarized', summary)
 })
 
+workflow.on('ticket.summarized', summary => ai.sentiment(summary).then('update.ticket'))
+
+workflow.every('30 minutes during business hours', context => ai.reviewKPIs(context).then('slack.post'))
+
 export default workflow
 ```

package/index.test.ts

@@ -0,0 +1,9 @@
+import { describe, it } from 'vitest'
+
+// An entry will be shown in the report for this suite
+describe.todo('unimplemented suite')
+
+// An entry will be shown in the report for this test
+describe('suite', () => {
+  it.todo('unimplemented test')
+})

package/index.ts

@@ -4,9 +4,12 @@ import { AutoRouter } from 'itty-router'
 export const ai: Record<string, any> = {}
 
 export type WorkflowOptions = {
-  queue: string
+  queue?: string
+  dql?: string
 }
 
+// export let send: (event: string | WorkflowEvent) => Promise<any>
+
 // TODO: Add generic type for the event data
 export const Workflow = (options: WorkflowOptions) => {
 
@@ -17,8 +20,11 @@ export const Workflow = (options: WorkflowOptions) => {
 
   const events: Record<string, any> = {}
 
-  const send = (env: any) => async (event: WorkflowEvent) => {
+  const send = (env: any) => async (event: string | WorkflowEvent) => {
     if (env[queue] && env[queue].send) {
+      if (typeof event === 'string') {
+        return (data: any) => (env[queue] as Queue<WorkflowEvent>).send({ event, data })
+      }
       return await (env[queue] as Queue<WorkflowEvent>).send(event)
     }
     throw new Error('Queue not found')
@@ -39,7 +45,23 @@ export const Workflow = (options: WorkflowOptions) => {
     fetch: api.fetch,
     queue: (batch: MessageBatch<WorkflowEvent>, env: any, ctx: ExecutionContext) => {
       batch.messages.map(async (message) => {
-
+        const { event, data } = message.body
+        if (events[event]) {
+          try {
+            const results = await events[event](message.body, { ai, db: env.db, api: env.api, data, message, send: send(env) })
+            console.log(results)
+            message.ack()
+            // TODO: Try to support `workflow.on('ticket.summarized', summary => ai.sentiment(summary).then('update.ticket'))`
+            // if (typeof results === 'string') {
+            //   const resultData = await results
+            //   return send(env)({ event: results, data })
+            // }
+          } catch (error) {
+            console.error(error)
+            message.retry()
+          }
+        }
+        throw new Error('Event not found')
       })
     },
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-workflows",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "description": "Framework for Developing AI-Native Workflows",
   "module": "dist/index.js",
   "scripts": {