diff --git a/.env.example b/.env.example index 2a0a180d..4fb7a21b 100644 --- a/.env.example +++ b/.env.example @@ -63,6 +63,12 @@ HEURISTAIURL='https://llm-gateway.heurist.xyz' HEURISTAIMODELSURL='https://raw.githubusercontent.com/heurist-network/heurist-models/main/models.json' HEURISTAIAPIKEY='' +# If you want to use Google AI LLMs, add your key here +GEMINI_API_KEY= '' + +# If you want to use XAI LLMs, add your key here +XAI_API_KEY= '' + # Validator Configuration # JSON array of initial validators to be created on startup. # Example: VALIDATORS_CONFIG_JSON = '[{"stake": 100, "provider": "openai", "model": "gpt-4o", "amount": 2}, {"stake": 200, "provider": "anthropic", "model": "claude-3-haiku-20240307", "amount": 1}]' diff --git a/src/lib/config/simulator.ts b/src/lib/config/simulator.ts index d77bec04..38924600 100644 --- a/src/lib/config/simulator.ts +++ b/src/lib/config/simulator.ts @@ -23,8 +23,8 @@ export type RunningPlatform = (typeof AVAILABLE_PLATFORMS)[number]; export const STARTING_TIMEOUT_WAIT_CYLCE = 2000; export const STARTING_TIMEOUT_ATTEMPTS = 120; -export type AiProviders = "ollama" | "openai" | "heuristai"; -export type AiProvidersEnvVars = "ollama" | "OPENAIKEY" | "HEURISTAIAPIKEY"; +export type AiProviders = "ollama" | "openai" | "heuristai" | "geminiai" | "xai"; +export type AiProvidersEnvVars = "ollama" | "OPENAIKEY" | "HEURISTAIAPIKEY" | "GEMINI_API_KEY" | "XAI_API_KEY"; export type AiProvidersConfigType = { [key in AiProviders]: {name: string; hint: string; envVar?: AiProvidersEnvVars; cliOptionValue: string}; }; @@ -47,4 +47,16 @@ export const AI_PROVIDERS_CONFIG: AiProvidersConfigType = { envVar: "HEURISTAIAPIKEY", cliOptionValue: "heuristai", }, + geminiai: { + name: "Gemini", + hint: '(You will need to provide an API key.)', + envVar: "GEMINI_API_KEY", + cliOptionValue: "geminiai", + }, + xai: { + name: "XAI", + hint: '(You will need to provide an API key)', + envVar: "XAI_API_KEY", + cliOptionValue: "xai", + }, };