External/securelens-backend
1
0
Fork 0
mirror of https://github.com/Rarebuffalo/securelens-backend.git synced 2026-06-19 07:00:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: integrate LiteLLM for provider-agnostic AI (supports Gemini, OpenAI, Claude, Ollama)

Browse Source
This commit is contained in:
rarebuffalo
2026-05-03 16:09:53 +05:30
parent 8dd9f12be3
commit 03fe240642
5 changed files with 295 additions and 152 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
app/config.py
View File

@@ -1,9 +1,32 @@
"""
Application Configuration
==========================
All configuration is loaded from environment variables (or a .env file).
AI Provider Configuration
--------------------------
SecureLens supports multiple AI providers via LiteLLM.
Set AI_MODEL and AI_API_KEY in your .env to choose a provider:
Provider | AI_MODEL example | Where to get key
-------------|--------------------------------------|-----------------------------
Gemini | gemini/gemini-2.0-flash | aistudio.google.com
OpenAI | gpt-4o-mini | platform.openai.com
Claude | claude-3-5-haiku-20241022 | console.anthropic.com
OpenRouter | openrouter/google/gemini-2.0-flash | openrouter.ai
Ollama | ollama/llama3.1 | ollama.com (local, no key)
If you only set GEMINI_API_KEY (legacy), it will still work — the app
automatically maps it to the Gemini provider for backward compatibility.
"""
from pydantic_settings import BaseSettings, SettingsConfigDict
class Settings(BaseSettings):
app_name: str = "SecureLens AI"
app_version: str = "1.0.0"
app_version: str = "1.1.0"
debug: bool = False
host: str = "0.0.0.0"
@@ -22,13 +45,39 @@ class Settings(BaseSettings):
jwt_algorithm: str = "HS256"
jwt_expiry_minutes: int = 1440
# -------------------------------------------------------------------------
# AI Provider Settings (new, provider-agnostic)
# -------------------------------------------------------------------------
# AI_MODEL: the LiteLLM model string (see table in module docstring above)
ai_model: str = "gemini/gemini-2.0-flash"
# AI_API_KEY: the API key for the chosen provider.
# Leave blank for Ollama (local, no key needed).
ai_api_key: str | None = None
# -------------------------------------------------------------------------
# Legacy Gemini key — kept for backward compatibility.
# If AI_API_KEY is not set but GEMINI_API_KEY is, we use that automatically.
# -------------------------------------------------------------------------
gemini_api_key: str | None = None
# Threat Intelligence API keys (Step 3)
virustotal_api_key: str | None = None
abuseipdb_api_key: str | None = None
model_config = SettingsConfigDict(env_file=".env", env_file_encoding="utf-8")
@property
def cors_origin_list(self) -> list[str]:
return [origin.strip() for origin in self.cors_origins.split(",") if origin.strip()]
@property
def effective_ai_key(self) -> str | None:
"""
Returns the resolved AI API key.
Prefers AI_API_KEY; falls back to GEMINI_API_KEY for backward compatibility.
"""
return self.ai_api_key or self.gemini_api_key
settings = Settings()