feat: added phi4, codestral and deepseek r1

xavidop · Jan 30, 2025 · 0c9e135 · 0c9e135
1 parent c3be996
commit 0c9e135
Show file tree

Hide file tree

Showing 2 changed files with 56 additions and 0 deletions.
diff --git a/src/github_llms.ts b/src/github_llms.ts
@@ -384,6 +384,21 @@ export const mistralLarge2411 = modelRef({
   configSchema: GenerationCommonConfigSchema,
 });
 
+export const mistralCodestral2501 = modelRef({
+  name: "github/codestral-2501",
+  info: {
+    versions: ["Codestral-2501"],
+    label: "Mistral - Codestral-2501",
+    supports: {
+      multiturn: true,
+      tools: true,
+      media: false,
+      systemRole: true,
+      output: ["text", "json"],
+    },
+  },
+  configSchema: GenerationCommonConfigSchema,
+});
 
 export const mistralNemo = modelRef({
   name: "github/mistral-nemo",
@@ -593,6 +608,22 @@ export const microsoftPhi35Vision128kInstruct = modelRef({
   configSchema: GenerationCommonConfigSchema,
 });
 
+export const microsoftPhi4 = modelRef({
+  name: "github/phi-4",
+  info: {
+    versions: ["Phi-4"],
+    label: "Microsoft - Phi-4",
+    supports: {
+      multiturn: true,
+      tools: true,
+      media: false,
+      systemRole: true,
+      output: ["text", "json"],
+    },
+  },
+  configSchema: GenerationCommonConfigSchema,
+});
+
 export const jais30bChat = modelRef({
   name: "github/jais-30b-chat",
   info: {
@@ -609,6 +640,22 @@ export const jais30bChat = modelRef({
   configSchema: GenerationCommonConfigSchema,
 });
 
+export const deepseekR1 = modelRef({
+  name: "github/deepseek-r1",
+  info: {
+    versions: ["DeepSeek-R1"],
+    label: "DeepSeek - R1",
+    supports: {
+      multiturn: true,
+      tools: true,
+      media: false,
+      systemRole: true,
+      output: ["text", "json"],
+    },
+  },
+  configSchema: GenerationCommonConfigSchema,
+});
+
 export const SUPPORTED_GITHUB_MODELS: Record<string, any> = {
   "gpt-4o": openAIGpt4o,
   "gpt-4o-mini": openAIGpt4oMini,
@@ -633,6 +680,7 @@ export const SUPPORTED_GITHUB_MODELS: Record<string, any> = {
   "Mistral-large-2411": mistralLarge2411,
   "Mistral-nemo": mistralNemo,
   "Ministral-3B": ministral3B,
+  "Codestral-2501": mistralCodestral2501,
   "ai21-jamba-1.5-mini": ai21Jamba15Mini,
   "ai21-jamba-1.5-large": ai21Jamba15Large,
   "Phi-3-mini-4k-instruct": microsoftPhi3Mini4kInstruct,
@@ -644,7 +692,9 @@ export const SUPPORTED_GITHUB_MODELS: Record<string, any> = {
   "Phi-3.5-mini-instruct": microsoftPhi35Mini128kInstruct,
   "Phi-3.5-moe-instruct": microsoftPhi35MoE128kInstruct,
   "Phi-3.5-vision-instruct": microsoftPhi35Vision128kInstruct,
+  "Phi-4": microsoftPhi4,
   "jais-30b-chat": jais30bChat,
+  "Deepseek-r1": deepseekR1,
 };
 
 function toGithubRole(role: Role): string {

diff --git a/src/index.ts b/src/index.ts
@@ -27,6 +27,7 @@ import {
   mistralLarge2407,
   mistralLarge2411,
   mistralNemo,
+  mistralCodestral2501,
   ministral3B,
   ai21Jamba15Mini,
   ai21Jamba15Large,
@@ -39,7 +40,9 @@ import {
   microsoftPhi35Mini128kInstruct,
   microsoftPhi35MoE128kInstruct,
   microsoftPhi35Vision128kInstruct,
+  microsoftPhi4,
   jais30bChat,
+  deepseekR1,
   SUPPORTED_GITHUB_MODELS,
 } from "./github_llms.js";
 import {
@@ -73,6 +76,7 @@ export {
   mistralLarge,
   mistralLarge2407,
   mistralLarge2411,
+  mistralCodestral2501,
   mistralNemo,
   ministral3B,
   ai21Jamba15Mini,
@@ -86,7 +90,9 @@ export {
   microsoftPhi35Mini128kInstruct,
   microsoftPhi35MoE128kInstruct,
   microsoftPhi35Vision128kInstruct,
+  microsoftPhi4,
   jais30bChat,
+  deepseekR1,
 };
 
 export {