feat(vercel-adapter): add Vercel AI adapter

Author: gwardwell

packages/ai-vercel-adapter/src/VercelAIAdapter.ts

@@ -0,0 +1,50 @@
+import { ApolloLink } from "@apollo/client";
+import { type LanguageModel, generateObject } from "ai";
+import { AIAdapter } from "@apollo/client-ai";
+
+namespace VercelAIAdapter {
+  export interface Options extends AIAdapter.Options {
+    model: LanguageModel;
+  }
+}
+
+type RequiredGenerateObjectOptions = {
+  model: LanguageModel;
+  mode: "json";
+  prompt: string;
+  system?: string;
+};
+
+export class VercelAIAdapter extends AIAdapter {
+  public model: LanguageModel;
+
+  constructor(options: VercelAIAdapter.Options) {
+    super(options);
+
+    this.model = options.model;
+  }
+
+  public async generateResponseForOperation(
+    operation: ApolloLink.Operation,
+    prompt: string
+  ): Promise<AIAdapter.Result> {
+    const promptOptions: RequiredGenerateObjectOptions = {
+      mode: "json",
+      model: this.model,
+      prompt: this.createPrompt(operation, prompt),
+      system: this.systemPrompt,
+    };
+
+    return generateObject(promptOptions as any).then(
+      ({ object: result, finishReason, usage, warnings }) => {
+        if (!result || typeof result !== "object") {
+          return { data: null };
+        }
+        if (!Object.prototype.hasOwnProperty.call(result, "data")) {
+          return { data: result };
+        }
+        return result;
+      }
+    );
+  }
+}