diff options
Diffstat (limited to 'backend/app/auth/routes.py')
| -rw-r--r-- | backend/app/auth/routes.py | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/backend/app/auth/routes.py b/backend/app/auth/routes.py index 3c906b5..eaf897e 100644 --- a/backend/app/auth/routes.py +++ b/backend/app/auth/routes.py @@ -228,14 +228,20 @@ async def get_api_keys(current_user: User = Depends(get_current_user)): return { "openai_api_key": mask_key(current_user.openai_api_key), "gemini_api_key": mask_key(current_user.gemini_api_key), + "claude_api_key": mask_key(current_user.claude_api_key), + "openrouter_api_key": mask_key(current_user.openrouter_api_key), "has_openai_key": bool(current_user.openai_api_key), "has_gemini_key": bool(current_user.gemini_api_key), + "has_claude_key": bool(current_user.claude_api_key), + "has_openrouter_key": bool(current_user.openrouter_api_key), } class ApiKeysUpdate(BaseModel): openai_api_key: Optional[str] = None gemini_api_key: Optional[str] = None + claude_api_key: Optional[str] = None + openrouter_api_key: Optional[str] = None @router.post("/api-keys") @@ -253,7 +259,13 @@ async def update_api_keys( if keys.gemini_api_key is not None: current_user.gemini_api_key = keys.gemini_api_key if keys.gemini_api_key else None - + + if keys.claude_api_key is not None: + current_user.claude_api_key = keys.claude_api_key if keys.claude_api_key else None + + if keys.openrouter_api_key is not None: + current_user.openrouter_api_key = keys.openrouter_api_key if keys.openrouter_api_key else None + db.commit() return {"message": "API keys updated successfully"} |