init

workers/site/workflows/preprocessing/newsOperator.ts

@@ -0,0 +1,101 @@
+import { WorkflowOperator } from "manifold-workflow-engine";
+import { zodResponseFormat } from "openai/helpers/zod";
+import { z } from "zod";
+
+const QuerySchema = z.object({
+  query: z.string(),
+});
+
+export function createNewsWebhookOperator({
+  eventHost,
+  streamId,
+  openai,
+  messages,
+}) {
+  return new WorkflowOperator("news-search", async (state: any) => {
+    const { latestUserMessage } = state;
+    console.log(`Processing user message: ${latestUserMessage}`);
+
+    const resource = "news-search";
+    const input = await getQueryFromContext({
+      openai,
+      messages,
+      latestUserMessage,
+    });
+
+    const eventSource = new URL(eventHost);
+    const url = `${eventSource}api/webhooks`;
+    console.log({ url });
+
+    const stream = {
+      id: crypto.randomUUID(),
+      parent: streamId,
+      resource,
+      payload: input,
+    };
+    const createStreamResponse = await fetch(`${eventSource}api/webhooks`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+      },
+      body: JSON.stringify({
+        id: stream.id,
+        parent: streamId,
+        resource: "news-search",
+        payload: {
+          input,
+        },
+      }),
+    });
+    const raw = await createStreamResponse.text();
+    const { stream_url } = JSON.parse(raw);
+    const surl = eventHost + stream_url;
+    const webhook = { url: surl, id: stream.id, resource };
+
+    return {
+      ...state,
+      webhook,
+      latestUserMessage: "",
+      latestAiMessage: "",
+    };
+  });
+
+  async function getQueryFromContext({ messages, openai, latestUserMessage }) {
+    const systemMessage = {
+      role: "system",
+      content: `Analyze the latest message in a conversation and generate a JSON object with a single implied question for a news search. The JSON should be structured as follows:
+
+{
+  "query": "<question to be answered by a news search>"
+}
+
+## Example
+{
+  "query": "When was the last Buffalo Sabres hockey game?"
+}
+
+Focus on the most recent message to determine the query. Output only the JSON object without any additional text.`,
+    };
+
+    const conversation = messages.map((m) => ({
+      role: m.role,
+      content: m.content,
+    }));
+    conversation.push({ role: "user", content: `${latestUserMessage}` });
+
+    const m = [systemMessage, ...conversation];
+
+    const completion = await openai.beta.chat.completions.parse({
+      model: "gpt-4o-mini",
+      messages: m,
+      temperature: 0,
+      response_format: zodResponseFormat(QuerySchema, "query"),
+    });
+
+    const { query } = completion.choices[0].message.parsed;
+
+    console.log({ newsWebhookQuery: query });
+
+    return query;
+  }
+}