Merge pull request #198 from HashWarlock/add-redpill

Add RedPill API Support
elizaOS · Nov 5, 2024 · cae6a5c · cae6a5c
2 parents 5fe429a + 9e7a3da
commit cae6a5c
Show file tree

Hide file tree

Showing 6 changed files with 10,854 additions and 13,459 deletions.
diff --git a/core/.env.example b/core/.env.example
@@ -2,6 +2,7 @@
 DISCORD_APPLICATION_ID=
 DISCORD_API_TOKEN= # Bot token
 OPENAI_API_KEY=sk-* # OpenAI API key, starting with sk-
+REDPILL_API_KEY= # REDPILL API Key
 GROQ_API_KEY=gsk_*
 
 ELEVENLABS_XI_API_KEY= # API key from elevenlabs

diff --git a/core/src/cli/index.ts b/core/src/cli/index.ts
@@ -103,6 +103,11 @@ export function getTokenForProvider(
                 character.settings?.secrets?.CLAUDE_API_KEY ||
                 settings.CLAUDE_API_KEY
             );
+        case ModelProvider.REDPILL:
+            return (
+                character.settings?.secrets?.REDPILL_API_KEY ||
+                settings.REDPILL_API_KEY
+            );
     }
 }
 export function initializeDatabase() {

diff --git a/core/src/core/generation.ts b/core/src/core/generation.ts
@@ -134,7 +134,7 @@ export async function generateText({
             case ModelProvider.GROQ: {
                 console.log("Initializing Groq model.");
                 const groq = createGroq({ apiKey });
-    
+
                 const { text: groqResponse } = await aiGenerateText({
                     model: groq.languageModel(model),
                     prompt: context,
@@ -143,26 +143,51 @@ export async function generateText({
                     frequencyPenalty: frequency_penalty,
                     presencePenalty: presence_penalty,
                 });
-    
+
                 response = groqResponse;
                 console.log("Received response from Groq model.");
                 break;
             }
 
-            case ModelProvider.LLAMALOCAL:
+            case ModelProvider.LLAMALOCAL: {
                 prettyConsole.log(
-                    "Using local Llama model for text completion."
+                  "Using local Llama model for text completion."
                 );
                 response = await runtime.llamaService.queueTextCompletion(
-                    context,
-                    temperature,
-                    _stop,
-                    frequency_penalty,
-                    presence_penalty,
-                    max_response_length
+                  context,
+                  temperature,
+                  _stop,
+                  frequency_penalty,
+                  presence_penalty,
+                  max_response_length
                 );
                 prettyConsole.log("Received response from local Llama model.");
                 break;
+            }
+
+            case ModelProvider.REDPILL: {
+                prettyConsole.log("Initializing RedPill model.");
+                const serverUrl = models[provider].endpoint;
+                const openai = createOpenAI({ apiKey, baseURL: serverUrl });
+
+                console.log('****** MODEL\n', model)
+                console.log('****** CONTEXT\n', context)
+
+                const { text: openaiResponse } = await aiGenerateText({
+                    model: openai.languageModel(model),
+                    prompt: context,
+                    temperature: temperature,
+                    maxTokens: max_response_length,
+                    frequencyPenalty: frequency_penalty,
+                    presencePenalty: presence_penalty,
+                });
+
+                console.log("****** RESPONSE\n", openaiResponse);
+
+                response = openaiResponse;
+                prettyConsole.log("Received response from OpenAI model.");
+                break;
+            }
 
             default: {
                 const errorMessage = `Unsupported provider: ${provider}`;

diff --git a/core/src/core/models.ts b/core/src/core/models.ts
@@ -9,6 +9,7 @@ type Models = {
     [ModelProvider.LLAMALOCAL]: Model;
     [ModelProvider.GOOGLE]: Model;
     [ModelProvider.CLAUDE_VERTEX]: Model;
+    [ModelProvider.REDPILL]: Model;
     // TODO: add OpenRouter - feel free to do this :)
 };
 
@@ -149,6 +150,25 @@ const models: Models = {
             [ModelClass.EMBEDDING]: "text-embedding-004",
         },
     },
+    [ModelProvider.REDPILL]: {
+        endpoint: "https://api.red-pill.ai/v1",
+        settings: {
+            stop: [],
+            maxInputTokens: 128000,
+            maxOutputTokens: 8192,
+            frequency_penalty: 0.0,
+            presence_penalty: 0.0,
+            temperature: 0.6,
+        },
+        // Available models: https://docs.red-pill.ai/get-started/supported-models
+        // To test other models, change the models below
+        model: {
+            [ModelClass.SMALL]: "gpt-4o-mini", // [ModelClass.SMALL]: "claude-3-5-sonnet-20241022",
+            [ModelClass.MEDIUM]: "gpt-4o", // [ModelClass.MEDIUM]: "claude-3-5-sonnet-20241022",
+            [ModelClass.LARGE]: "gpt-4o", // [ModelClass.LARGE]: "claude-3-opus-20240229",
+            [ModelClass.EMBEDDING]: "text-embedding-3-small",
+        },
+    },
 };
 
 export function getModel(provider: ModelProvider, type: ModelClass) {

diff --git a/core/src/core/types.ts b/core/src/core/types.ts
@@ -107,6 +107,7 @@ export enum ModelProvider {
     LLAMALOCAL = "llama_local",
     GOOGLE = "google",
     CLAUDE_VERTEX = "claude_vertex",
+    REDPILL = "redpill"
 }
 
 /**