Merge pull request #64 from upstash/add-anthropic-model

feat: add anthropic model

package.json

@@ -82,6 +82,7 @@
   },
   "optionalDependencies": {
     "langsmith": "^0.1.41",
-    "@langchain/mistralai": "^0.0.28"
+    "@langchain/mistralai": "^0.0.28",
+    "@langchain/anthropic": "^0.2.15"
   }
 }

src/models.ts

@@ -3,6 +3,7 @@ import { ChatOpenAI } from "@langchain/openai";
 import { Client as LangsmithClient } from "langsmith";
 import type { OLLAMA_MODELS } from "./constants";
 import { ChatMistralAI } from "@langchain/mistralai";
+import { ChatAnthropic } from "@langchain/anthropic";
 
 // Initialize global Langsmith tracer
 // We use a global variable because:
@@ -263,3 +264,10 @@ export const mistralai = (model: string, options?: Omit<ModelOptions, "baseUrl">
     ...options,
   });
 };
+
+export const antrophic = (model: string, options?: Omit<ModelOptions, "baseUrl">) => {
+  return new ChatAnthropic({
+    model,
+    ...options,
+  });
+};

src/types.ts

@@ -5,6 +5,7 @@ import type { Redis } from "@upstash/redis";
 import type { Index } from "@upstash/vector";
 import type { CustomPrompt } from "./rag-chat";
 import type { ChatMistralAI } from "@langchain/mistralai";
+import type { ChatAnthropic } from "@langchain/anthropic";
 
 declare const __brand: unique symbol;
 type Brand<B> = { [__brand]: B };
@@ -96,7 +97,7 @@ export type RAGChatConfig = {
   */
 
   // eslint-disable-next-line @typescript-eslint/no-redundant-type-constituents
-  model?: ChatOpenAI | ChatMistralAI | OpenAIChatLanguageModel;
+  model?: ChatOpenAI | ChatMistralAI | ChatAnthropic | OpenAIChatLanguageModel;
   /**
      * Ratelimit instance
      * @example new Ratelimit({