This commit is contained in:
marko-kraemer 2025-07-27 05:53:25 +02:00
parent e38d8f327e
commit aed291e7c3
8 changed files with 45 additions and 52 deletions

View File

@ -1,8 +1,10 @@
from fastapi import APIRouter, HTTPException, Depends
from typing import Optional
from typing import Optional, Dict
from utils.auth_utils import verify_admin_api_key
from utils.suna_default_agent_service import SunaDefaultAgentService
from utils.logger import logger
from utils.config import config, EnvMode
from dotenv import load_dotenv, set_key, find_dotenv, dotenv_values
router = APIRouter(prefix="/admin", tags=["admin"])
@ -39,4 +41,42 @@ async def admin_install_suna_for_user(
raise HTTPException(
status_code=500,
detail=f"Failed to install Suna agent for user {account_id}"
)
)
@router.get("/env-vars")
def get_env_vars() -> Dict[str, str]:
"""Get environment variables (local mode only)."""
if config.ENV_MODE != EnvMode.LOCAL:
raise HTTPException(status_code=403, detail="Env vars management only available in local mode")
try:
env_path = find_dotenv()
if not env_path:
logger.error("Could not find .env file")
return {}
return dotenv_values(env_path)
except Exception as e:
logger.error(f"Failed to get env vars: {e}")
raise HTTPException(status_code=500, detail=f"Failed to get env variables: {e}")
@router.post("/env-vars")
def save_env_vars(request: Dict[str, str]) -> Dict[str, str]:
"""Save environment variables (local mode only)."""
if config.ENV_MODE != EnvMode.LOCAL:
raise HTTPException(status_code=403, detail="Env vars management only available in local mode")
try:
env_path = find_dotenv()
if not env_path:
raise HTTPException(status_code=500, detail="Could not find .env file")
for key, value in request.items():
set_key(env_path, key, value)
load_dotenv(override=True)
logger.info(f"Env variables saved successfully: {request}")
return {"message": "Env variables saved successfully"}
except Exception as e:
logger.error(f"Failed to save env variables: {e}")
raise HTTPException(status_code=500, detail=f"Failed to save env variables: {e}")

View File

@ -189,8 +189,7 @@ api_router.include_router(pipedream_api.router)
# MFA functionality moved to frontend
from local_env_manager import api as local_env_manager_api
api_router.include_router(local_env_manager_api.router)
from admin import api as admin_api
api_router.include_router(admin_api.router)

View File

@ -1,44 +0,0 @@
from fastapi import APIRouter
from utils.config import config, EnvMode
from fastapi import HTTPException
from typing import Dict
from dotenv import load_dotenv, set_key, find_dotenv, dotenv_values
from utils.logger import logger
router = APIRouter(tags=["local-env-manager"])
@router.get("/env-vars")
def get_env_vars() -> Dict[str, str]:
if config.ENV_MODE != EnvMode.LOCAL:
raise HTTPException(status_code=403, detail="Env vars management only available in local mode")
try:
env_path = find_dotenv()
if not env_path:
logger.error("Could not find .env file")
return {}
return dotenv_values(env_path)
except Exception as e:
logger.error(f"Failed to get env vars: {e}")
raise HTTPException(status_code=500, detail=f"Failed to get env variables: {e}")
@router.post("/env-vars")
def save_env_vars(request: Dict[str, str]) -> Dict[str, str]:
if config.ENV_MODE != EnvMode.LOCAL:
raise HTTPException(status_code=403, detail="Env vars management only available in local mode")
try:
env_path = find_dotenv()
if not env_path:
raise HTTPException(status_code=500, detail="Could not find .env file")
for key, value in request.items():
set_key(env_path, key, value)
load_dotenv(override=True)
logger.info(f"Env variables saved successfully: {request}")
return {"message": "Env variables saved successfully"}
except Exception as e:
logger.error(f"Failed to save env variables: {e}")
raise HTTPException(status_code=500, detail=f"Failed to save env variables: {e}")

View File

@ -8,8 +8,6 @@ import { CustomMCPDialog } from './custom-mcp-dialog';
import { PipedreamRegistry } from '@/components/agents/pipedream/pipedream-registry';
import { ToolsManager } from './tools-manager';
import { toast } from 'sonner';
import { useQueryClient } from '@tanstack/react-query';
import { agentKeys } from '@/hooks/react-query/agents/keys';
export const MCPConfigurationNew: React.FC<MCPConfigurationProps> = ({
configuredMCPs,

View File

@ -24,7 +24,7 @@ export function LocalEnvManager() {
const {data: apiKeys, isLoading} = useQuery({
queryKey: ['api-keys'],
queryFn: async() => {
const response = await backendApi.get('/env-vars');
const response = await backendApi.get('/admin/env-vars');
return response.data;
},
enabled: isLocalMode()
@ -82,7 +82,7 @@ export function LocalEnvManager() {
const updateApiKeys = useMutation({
mutationFn: async (data: APIKeyForm) => {
const response = await backendApi.post('/env-vars', data);
const response = await backendApi.post('/admin/env-vars', data);
await queryClient.invalidateQueries({ queryKey: ['api-keys'] });
return response.data;
},