ai.infer: anthropic types and adapters (#762)

This add types from the Anthropic SDK to support Anthropic models. --------- Co-authored-by: Jack Williams <[email protected]> Co-authored-by: Jack Williams <[email protected]>
inngest · Nov 27, 2024 · 255416c · 255416c
1 parent efc6c79
commit 255416c
Show file tree

Hide file tree

Showing 7 changed files with 734 additions and 2 deletions.
diff --git a/.changeset/twenty-buttons-retire.md b/.changeset/twenty-buttons-retire.md
@@ -0,0 +1,5 @@
+---
+"inngest": minor
+---
+
+Add `anthropic()` model for `step.ai.*()`
diff --git a/packages/inngest/src/components/ai/adapter.ts b/packages/inngest/src/components/ai/adapter.ts
@@ -1,3 +1,4 @@
+import { type AnthropicAiAdapter } from "./adapters/anthropic.js";
 import { type OpenAiAiAdapter } from "./adapters/openai.js";
 
 /**
@@ -91,7 +92,7 @@ export namespace AiAdapter {
   /**
    * Supported I/O formats for AI models.
    */
-  export type Format = "openai-chat"; // | "anthropic" | "gemini" | "bedrock";
+  export type Format = "openai-chat" | "anthropic";
 
   /**
    * A function that creates a model that adheres to an existng AI adapter
@@ -108,6 +109,7 @@ export namespace AiAdapter {
  */
 const adapters = {
   "openai-chat": null as unknown as OpenAiAiAdapter,
+  anthropic: null as unknown as AnthropicAiAdapter,
 } satisfies Record<AiAdapter.Format, AiAdapter>;
 
 /**