diff --git a/.env.example b/.env.example index 525d271f45..bec48db5b6 100644 --- a/.env.example +++ b/.env.example @@ -74,6 +74,15 @@ SMALL_OLLAMA_MODEL= # Default: llama3.2 MEDIUM_OLLAMA_MODEL= # Default: hermes3 LARGE_OLLAMA_MODEL= # Default: hermes3:70b +# Google Configuration +GOOGLE_MODEL= +SMALL_GOOGLE_MODEL= # Default: gemini-1.5-flash-latest +MEDIUM_GOOGLE_MODEL= # Default: gemini-1.5-flash-latest +LARGE_GOOGLE_MODEL= # Default: gemini-1.5-pro-latest +EMBEDDING_GOOGLE_MODEL= # Default: text-embedding-004 + + + #LlamaLocal Configuration LLAMALOCAL_PATH= # Default: "" which is the current directory in plugin-node/dist/ which gets destroyed and recreated on every build diff --git a/packages/core/src/models.ts b/packages/core/src/models.ts index 8965f97fb1..aeda487987 100644 --- a/packages/core/src/models.ts +++ b/packages/core/src/models.ts @@ -179,10 +179,18 @@ export const models: Models = { temperature: 0.7, }, model: { - [ModelClass.SMALL]: "gemini-1.5-flash-latest", - [ModelClass.MEDIUM]: "gemini-1.5-flash-latest", - [ModelClass.LARGE]: "gemini-1.5-pro-latest", - [ModelClass.EMBEDDING]: "text-embedding-004", + [ModelClass.SMALL]: settings.SMALL_GOOGLE_MODEL || + settings.GOOGLE_MODEL || + "gemini-1.5-flash-latest", + [ModelClass.MEDIUM]: settings.MEDIUM_GOOGLE_MODEL || + settings.GOOGLE_MODEL || + "gemini-1.5-flash-latest", + [ModelClass.LARGE]: settings.LARGE_GOOGLE_MODEL || + settings.GOOGLE_MODEL || + "gemini-1.5-pro-latest", + [ModelClass.EMBEDDING]: settings.EMBEDDING_GOOGLE_MODEL || + settings.GOOGLE_MODEL || + "text-embedding-004", }, }, [ModelProviderName.REDPILL]: {