Cloudflare Workers AI Adapter

import { MemoFS, RemoteBlobMemoryStore } from "@memofs/core";
import { createWorkersAiExtractor } from "@memofs/adapter-workers-ai";
import { createR2BlobClient } from "@memofs/adapter-r2";
import { createTursoMetadataStore } from "@memofs/adapter-turso";
import { createClient } from "@libsql/client";

export interface Env {
  BLOBS: R2Bucket;
  TURSO_DATABASE_URL: string;
  TURSO_AUTH_TOKEN: string;
  AI: Ai; // Injected Cloudflare Workers AI binding
}

export default {
  async fetch(request: Request, env: Env): Promise<Response> {
    const projectId = "serverless-project";

    // 1. Initialize remote blob memory store (R2 + Turso)
    const store = new RemoteBlobMemoryStore({
      blobClient: createR2BlobClient({ binding: env.BLOBS }),
      metadata: createTursoMetadataStore({
        client: createClient({
          url: env.TURSO_DATABASE_URL,
          authToken: env.TURSO_AUTH_TOKEN,
        }),
        projectId,
      }),
      rootKey: projectId,
    });

    // 2. Initialize MemoFS client with Workers AI extractor
    const memo = new MemoFS({
      store,
      projectId,
      mode: "local",
      extractor: createWorkersAiExtractor({
        ai: env.AI,
        model: "@cf/meta/llama-3.1-8b-instruct",
      }),
    });

    // 3. Writing memories automatically extracts entities and relationships
    const result = await memo.writeMemory({
      title: "Auth Middleware Migration",
      content: "The AuthModule depends on the RedisSessionStore and supersedes the LegacyCookieAuth.",
      kind: "decision",
    });

    return new Response(JSON.stringify(result), {
      headers: { "Content-Type": "application/json" },
    });
  },
};