From 499d3b3e14491d245778d770ea9cf1a0bc073f8a Mon Sep 17 00:00:00 2001 From: Siddhesh Agarwal Date: Thu, 31 Oct 2024 12:18:51 +0530 Subject: [PATCH] Added Gemini API Key to settings + imported Gemini Provider --- simplemind/providers/__init__.py | 3 ++- simplemind/settings.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/simplemind/providers/__init__.py b/simplemind/providers/__init__.py index f9b1983..d142a4d 100644 --- a/simplemind/providers/__init__.py +++ b/simplemind/providers/__init__.py @@ -2,9 +2,10 @@ from typing import List, Type from ._base import BaseProvider from .anthropic import Anthropic +from .gemini import Gemini from .groq import Groq from .openai import OpenAI from .ollama import Ollama from .xai import XAI -providers: List[Type[BaseProvider]] = [Anthropic, Groq, OpenAI, Ollama, XAI] +providers: List[Type[BaseProvider]] = [Anthropic, Gemini, Groq, OpenAI, Ollama, XAI] diff --git a/simplemind/settings.py b/simplemind/settings.py index 4068ebd..b0acf98 100644 --- a/simplemind/settings.py +++ b/simplemind/settings.py @@ -11,6 +11,7 @@ class Settings(BaseSettings): None, description="API key for Anthropic" ) GROQ_API_KEY: Optional[SecretStr] = Field(None, description="API key for Groq") + GEMINI_API_KEY: Optional[SecretStr] = Field(None, description="API key for Gemini") OPENAI_API_KEY: Optional[SecretStr] = Field(None, description="API key for OpenAI") OLLAMA_HOST_URL: Optional[str] = Field( "http://127.0.0.1:11434", description="Fully qualified host URL for Ollama"