init

workers/site/workflows/preprocessing/scrapeOperator.ts

@@ -0,0 +1,112 @@
+import { WorkflowOperator } from "manifold-workflow-engine";
+import { zodResponseFormat } from "openai/helpers/zod";
+import { z } from "zod";
+
+const UrlActionSchema = z.object({
+  url: z.string(),
+  query: z.string(),
+  action: z.enum(["read", "scrape", "crawl", ""]),
+});
+
+export function createScrapeWebhookOperator({
+  eventHost,
+  streamId,
+  openai,
+  messages,
+}) {
+  return new WorkflowOperator("web-scrape", async (state: any) => {
+    const { latestUserMessage } = state;
+
+    const webscrapeWebhookEndpoint = "/api/webhooks";
+
+    const resource = "web-scrape";
+    const context = await getQueryFromContext({
+      openai,
+      messages,
+      latestUserMessage,
+    });
+
+    const input = {
+      url: context?.url,
+      action: context?.action,
+      query: context.query,
+    };
+
+    const eventSource = new URL(eventHost);
+    const url = `${eventSource}api/webhooks`;
+
+    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: "web-scrape",
+        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" as const,
+    content:
+      `You are modeling a structured output containing a single question, a URL, and an action, all relative to a single input.
+
+Return the result as a JSON object in the following structure:
+{
+  "url": "Full URL in the conversation that references the URL being interacted with. No trailing slash!",
+  "query": "Implied question about the resources at the URL.",
+  "action": "read | scrape | crawl"
+}
+
+- The input being modeled is conversational data from a different conversation than this one.
+- Intent should represent a next likely action the system might take to satisfy or enhance the user's request.
+
+Instructions:
+1. Analyze the provided context and declare the url, action, and question implied by the latest message.
+
+Output the JSON object. Do not include any additional explanations or text.`.trim(),
+  };
+
+  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(UrlActionSchema, "UrlActionSchema"),
+  });
+
+  const { query, action, url } = completion.choices[0].message.parsed;
+
+  return { query, action, url };
+}