feat: Add token retrieval for model provider in runtime configuration

Author: tercel

agent/src/index.ts

@@ -1556,6 +1556,7 @@ const startAgents = async () => {
 
     directClient.loadCharacterTryPath = loadCharacterTryPath;
     directClient.jsonToCharacter = jsonToCharacter;
+    directClient.getTokenForProvider = getTokenForProvider;
     directClient.db = db;
     directClient.plugins = await getPlugins();
     directClient.clients = await getClients();

packages/client-direct/src/index.ts

@@ -163,6 +163,7 @@ export class DirectClient {
     public db: TypeDatabaseAdapter;
     public plugins: any[];
     public clients: any[];
+    public getTokenForProvider: Function; // store getTokenForProvider functor
 
     constructor() {
         elizaLogger.log("DirectClient constructor");

packages/client-direct/src/manage-api.ts

@@ -748,23 +748,6 @@ export function createManageApiRouter(
                 return;
             }
 
-            // if(runtime.character.modelProvider !== modelProvider) {
-            //     runtime.character.modelProvider = modelProvider;
-            //     runtime.modelProvider = modelProvider;
-            //     if(secrets) {
-            //         Object.assign(runtime.character.settings, secrets);
-            //     }
-            //     elizaLogger.log(`runtime model provider changed to ${modelProvider}`);
-            // }
-
-            await runtime.ensureConnection(
-                userId,
-                roomId,
-                runtime.character.username,
-                runtime.character.name,
-                "direct"
-            );
-
             const tpl = await directClient.loadCharacterTryPath('characters/lpmanager.character.json');
             if(!tpl) {
                 res.status(500).send({ error: "Failed to load template" });
@@ -776,6 +759,21 @@ export function createManageApiRouter(
                 tpl.settings.secrets = secrets;
             }
 
+            if(runtime.character.modelProvider !== modelProvider) {
+                runtime.character.modelProvider = modelProvider;
+                runtime.modelProvider = modelProvider;
+                runtime.token = directClient.getTokenForProvider(modelProvider, tpl);
+                elizaLogger.log(`runtime model provider changed to ${modelProvider}`);
+            }
+
+            await runtime.ensureConnection(
+                userId,
+                roomId,
+                runtime.character.username,
+                runtime.character.name,
+                "direct"
+            );
+
             const text = `According to the user-provided [description] in accordance with the provided json format [template] to generate the user's json content.
             [description]: ${description}. 
             [template]: ${JSON.stringify(tpl)}`;