Skip to content

Commit

Permalink
remove extraneous stuff with verifiable inference
Browse files Browse the repository at this point in the history
  • Loading branch information
@lalalune
lalalune committed Jan 28, 2025
1 parent 81095fe commit ccf370e
Show file tree
Hide file tree
Showing 95 changed files with 198 additions and 1,626 deletions.
215 changes: 0 additions & 215 deletions agent/src/index.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,6 @@ import {
type ICacheManager,
type IDatabaseAdapter,
type IDatabaseCacheAdapter,
ModelProviderName,
parseBooleanFromText,
settings,
stringToUuid,
Expand Down Expand Up @@ -518,165 +517,6 @@ async function handlePluginImporting(plugins: string[]) {
}
}

export function getTokenForProvider(
provider: ModelProviderName,
character: Character
): string | undefined {
switch (provider) {
// no key needed for llama_local or gaianet
case ModelProviderName.LLAMALOCAL:
return "";
case ModelProviderName.OLLAMA:
return "";
case ModelProviderName.GAIANET:
return "";
case ModelProviderName.BEDROCK:
return "";
case ModelProviderName.OPENAI:
return (
character.settings?.secrets?.OPENAI_API_KEY ||
settings.OPENAI_API_KEY
);
case ModelProviderName.ETERNALAI:
return (
character.settings?.secrets?.ETERNALAI_API_KEY ||
settings.ETERNALAI_API_KEY
);
case ModelProviderName.NINETEEN_AI:
return (
character.settings?.secrets?.NINETEEN_AI_API_KEY ||
settings.NINETEEN_AI_API_KEY
);
case ModelProviderName.LLAMACLOUD:
case ModelProviderName.TOGETHER:
return (
character.settings?.secrets?.LLAMACLOUD_API_KEY ||
settings.LLAMACLOUD_API_KEY ||
character.settings?.secrets?.TOGETHER_API_KEY ||
settings.TOGETHER_API_KEY ||
character.settings?.secrets?.OPENAI_API_KEY ||
settings.OPENAI_API_KEY
);
case ModelProviderName.CLAUDE_VERTEX:
case ModelProviderName.ANTHROPIC:
return (
character.settings?.secrets?.ANTHROPIC_API_KEY ||
character.settings?.secrets?.CLAUDE_API_KEY ||
settings.ANTHROPIC_API_KEY ||
settings.CLAUDE_API_KEY
);
case ModelProviderName.REDPILL:
return (
character.settings?.secrets?.REDPILL_API_KEY ||
settings.REDPILL_API_KEY
);
case ModelProviderName.OPENROUTER:
return (
character.settings?.secrets?.OPENROUTER_API_KEY ||
settings.OPENROUTER_API_KEY
);
case ModelProviderName.GROK:
return (
character.settings?.secrets?.GROK_API_KEY ||
settings.GROK_API_KEY
);
case ModelProviderName.HEURIST:
return (
character.settings?.secrets?.HEURIST_API_KEY ||
settings.HEURIST_API_KEY
);
case ModelProviderName.GROQ:
return (
character.settings?.secrets?.GROQ_API_KEY ||
settings.GROQ_API_KEY
);
case ModelProviderName.GALADRIEL:
return (
character.settings?.secrets?.GALADRIEL_API_KEY ||
settings.GALADRIEL_API_KEY
);
case ModelProviderName.FAL:
return (
character.settings?.secrets?.FAL_API_KEY || settings.FAL_API_KEY
);
case ModelProviderName.ALI_BAILIAN:
return (
character.settings?.secrets?.ALI_BAILIAN_API_KEY ||
settings.ALI_BAILIAN_API_KEY
);
case ModelProviderName.VOLENGINE:
return (
character.settings?.secrets?.VOLENGINE_API_KEY ||
settings.VOLENGINE_API_KEY
);
case ModelProviderName.NANOGPT:
return (
character.settings?.secrets?.NANOGPT_API_KEY ||
settings.NANOGPT_API_KEY
);
case ModelProviderName.HYPERBOLIC:
return (
character.settings?.secrets?.HYPERBOLIC_API_KEY ||
settings.HYPERBOLIC_API_KEY
);

case ModelProviderName.VENICE:
return (
character.settings?.secrets?.VENICE_API_KEY ||
settings.VENICE_API_KEY
);
case ModelProviderName.ATOMA:
return (
character.settings?.secrets?.ATOMASDK_BEARER_AUTH ||
settings.ATOMASDK_BEARER_AUTH
);
case ModelProviderName.NVIDIA:
return (
character.settings?.secrets?.NVIDIA_API_KEY ||
settings.NVIDIA_API_KEY
);
case ModelProviderName.AKASH_CHAT_API:
return (
character.settings?.secrets?.AKASH_CHAT_API_KEY ||
settings.AKASH_CHAT_API_KEY
);
case ModelProviderName.GOOGLE:
return (
character.settings?.secrets?.GOOGLE_GENERATIVE_AI_API_KEY ||
settings.GOOGLE_GENERATIVE_AI_API_KEY
);
case ModelProviderName.MISTRAL:
return (
character.settings?.secrets?.MISTRAL_API_KEY ||
settings.MISTRAL_API_KEY
);
case ModelProviderName.LETZAI:
return (
character.settings?.secrets?.LETZAI_API_KEY ||
settings.LETZAI_API_KEY
);
case ModelProviderName.INFERA:
return (
character.settings?.secrets?.INFERA_API_KEY ||
settings.INFERA_API_KEY
);
case ModelProviderName.DEEPSEEK:
return (
character.settings?.secrets?.DEEPSEEK_API_KEY ||
settings.DEEPSEEK_API_KEY
);
case ModelProviderName.LIVEPEER:
return (
character.settings?.secrets?.LIVEPEER_GATEWAY_URL ||
settings.LIVEPEER_GATEWAY_URL
);
default:
const errorMessage = `Failed to get token - unsupported model provider: ${provider}`;
elizaLogger.error(errorMessage);
throw new Error(errorMessage);
}
}

function initializeDatabase(dataDir: string) {
if (process.env.MONGODB_CONNECTION_STRING) {
elizaLogger.log("Initializing database on MongoDB Atlas");
Expand Down Expand Up @@ -955,62 +795,8 @@ export async function createAgent(
);
}

// Initialize Reclaim adapter if environment variables are present
// let verifiableInferenceAdapter;
// if (
// process.env.RECLAIM_APP_ID &&
// process.env.RECLAIM_APP_SECRET &&
// process.env.VERIFIABLE_INFERENCE_ENABLED === "true"
// ) {
// verifiableInferenceAdapter = new ReclaimAdapter({
// appId: process.env.RECLAIM_APP_ID,
// appSecret: process.env.RECLAIM_APP_SECRET,
// modelProvider: character.modelProvider,
// token,
// });
// elizaLogger.log("Verifiable inference adapter initialized");
// }
// Initialize Opacity adapter if environment variables are present
let verifiableInferenceAdapter;
if (
process.env.OPACITY_TEAM_ID &&
process.env.OPACITY_CLOUDFLARE_NAME &&
process.env.OPACITY_PROVER_URL &&
process.env.VERIFIABLE_INFERENCE_ENABLED === "true"
) {
verifiableInferenceAdapter = new OpacityAdapter({
teamId: process.env.OPACITY_TEAM_ID,
teamName: process.env.OPACITY_CLOUDFLARE_NAME,
opacityProverUrl: process.env.OPACITY_PROVER_URL,
modelProvider: character.modelProvider,
token: token,
});
elizaLogger.log("Verifiable inference adapter initialized");
elizaLogger.log("teamId", process.env.OPACITY_TEAM_ID);
elizaLogger.log("teamName", process.env.OPACITY_CLOUDFLARE_NAME);
elizaLogger.log("opacityProverUrl", process.env.OPACITY_PROVER_URL);
elizaLogger.log("modelProvider", character.modelProvider);
elizaLogger.log("token", token);
}
if (
process.env.PRIMUS_APP_ID &&
process.env.PRIMUS_APP_SECRET &&
process.env.VERIFIABLE_INFERENCE_ENABLED === "true"
) {
verifiableInferenceAdapter = new PrimusAdapter({
appId: process.env.PRIMUS_APP_ID,
appSecret: process.env.PRIMUS_APP_SECRET,
attMode: "proxytls",
modelProvider: character.modelProvider,
token,
});
elizaLogger.log("Verifiable inference primus adapter initialized");
}

return new AgentRuntime({
databaseAdapter: db,
token,
modelProvider: character.modelProvider,
evaluators: [],
character,
// character.plugins are handled when clients are added
Expand Down Expand Up @@ -1301,7 +1087,6 @@ export async function createAgent(
managers: [],
cacheManager: cache,
fetch: logFetch,
verifiableInferenceAdapter,
});
}

Expand Down
Loading

0 comments on commit ccf370e

Please sign in to comment.