mirror of
https://github.com/eigent-ai/eigent.git
synced 2026-05-05 23:41:06 +00:00
chore: fix pre commit format and pipeline issue (#1144)
This commit is contained in:
Wendong-Fan
GitHub
parent
2256497dff
commit
893f51fc82
970 changed files with 11235 additions and 9533 deletions
|
|
@ -12,18 +12,16 @@
|
|||
# limitations under the License.
|
||||
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
|
||||
|
||||
from typing import Annotated
|
||||
from fastapi import APIRouter, Depends, Query
|
||||
from sqlmodel import Session, col, select
|
||||
|
||||
from app.component.database import session
|
||||
from app.model.mcp.category import Category, CategoryOut
|
||||
|
||||
|
||||
router = APIRouter(prefix="/mcp", tags=["Mcp Category"])
|
||||
|
||||
|
||||
@router.get("/categories", name="category list", response_model=list[CategoryOut])
|
||||
def gets(session: Session = Depends(session)):
|
||||
stmt = select(Category).where(Category.no_delete()).order_by(col(Category.priority).asc())
|
||||
return session.exec(stmt)
|
||||
from fastapi import APIRouter, Depends
|
||||
from sqlmodel import Session, col, select
|
||||
|
||||
from app.component.database import session
|
||||
from app.model.mcp.category import Category, CategoryOut
|
||||
|
||||
router = APIRouter(prefix="/mcp", tags=["Mcp Category"])
|
||||
|
||||
|
||||
@router.get("/categories", name="category list", response_model=list[CategoryOut])
|
||||
def gets(session: Session = Depends(session)):
|
||||
stmt = select(Category).where(Category.no_delete()).order_by(col(Category.priority).asc())
|
||||
return session.exec(stmt)
|
||||
|
|
|
|||
|
|
@ -12,22 +12,23 @@
|
|||
# limitations under the License.
|
||||
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
|
||||
|
||||
import logging
|
||||
import os
|
||||
from typing import Dict
|
||||
from fastapi import Depends, HTTPException, APIRouter
|
||||
|
||||
from camel.toolkits.mcp_toolkit import MCPToolkit
|
||||
from fastapi import APIRouter, Depends, HTTPException
|
||||
from fastapi_babel import _
|
||||
from fastapi_pagination import Page
|
||||
from fastapi_pagination.ext.sqlmodel import paginate
|
||||
from sqlmodel import Session, col, select
|
||||
from sqlalchemy.orm import selectinload, with_loader_criteria
|
||||
from sqlmodel import Session, col, select
|
||||
|
||||
from app.component.auth import Auth, auth_must
|
||||
from app.component.database import session
|
||||
from app.component.environment import env
|
||||
from app.model.mcp.mcp import Mcp, McpOut, McpType
|
||||
from app.model.mcp.mcp_env import McpEnv, Status as McpEnvStatus
|
||||
from app.model.mcp.mcp_user import McpImportType, McpUser, Status
|
||||
from camel.toolkits.mcp_toolkit import MCPToolkit
|
||||
from app.component.environment import env
|
||||
import logging
|
||||
|
||||
logger = logging.getLogger("server_mcp_controller")
|
||||
|
||||
|
|
@ -59,8 +60,7 @@ async def pre_instantiate_mcp_toolkit(config_dict: dict) -> bool:
|
|||
# Set global auth directory to persist authentication across tasks
|
||||
if "MCP_REMOTE_CONFIG_DIR" not in server_config["env"]:
|
||||
server_config["env"]["MCP_REMOTE_CONFIG_DIR"] = env(
|
||||
"MCP_REMOTE_CONFIG_DIR",
|
||||
os.path.expanduser("~/.mcp-auth")
|
||||
"MCP_REMOTE_CONFIG_DIR", os.path.expanduser("~/.mcp-auth")
|
||||
)
|
||||
|
||||
# Create MCP toolkit and attempt to connect
|
||||
|
|
@ -112,10 +112,13 @@ async def gets(
|
|||
with_loader_criteria(McpUser, col(McpUser.user_id) == user_id),
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
result = paginate(session, stmt)
|
||||
total = result.total if hasattr(result, 'total') else 0
|
||||
logger.debug("MCP list retrieved", extra={"user_id": user_id, "keyword": keyword, "category_id": category_id, "mine": mine, "total": total})
|
||||
total = result.total if hasattr(result, "total") else 0
|
||||
logger.debug(
|
||||
"MCP list retrieved",
|
||||
extra={"user_id": user_id, "keyword": keyword, "category_id": category_id, "mine": mine, "total": total},
|
||||
)
|
||||
return result
|
||||
|
||||
|
||||
|
|
@ -123,11 +126,13 @@ async def gets(
|
|||
async def get(id: int, session: Session = Depends(session)):
|
||||
"""Get MCP server details."""
|
||||
try:
|
||||
stmt = select(Mcp).where(Mcp.no_delete(), Mcp.id == id).options(selectinload(Mcp.category), selectinload(Mcp.envs))
|
||||
stmt = (
|
||||
select(Mcp).where(Mcp.no_delete(), Mcp.id == id).options(selectinload(Mcp.category), selectinload(Mcp.envs))
|
||||
)
|
||||
model = session.exec(stmt).one()
|
||||
logger.debug("MCP detail retrieved", extra={"mcp_id": id, "mcp_key": model.key})
|
||||
return model
|
||||
except Exception as e:
|
||||
except Exception:
|
||||
logger.warning("MCP not found", extra={"mcp_id": id})
|
||||
raise HTTPException(status_code=404, detail=_("Mcp not found"))
|
||||
|
||||
|
|
@ -136,34 +141,39 @@ async def get(id: int, session: Session = Depends(session)):
|
|||
async def install(mcp_id: int, session: Session = Depends(session), auth: Auth = Depends(auth_must)):
|
||||
"""Install MCP server for user."""
|
||||
user_id = auth.user.id
|
||||
|
||||
|
||||
mcp = session.get_one(Mcp, mcp_id)
|
||||
if not mcp:
|
||||
logger.warning("MCP install failed: MCP not found", extra={"user_id": user_id, "mcp_id": mcp_id})
|
||||
raise HTTPException(status_code=404, detail=_("Mcp not found"))
|
||||
|
||||
|
||||
exists = session.exec(select(McpUser).where(McpUser.mcp_id == mcp.id, McpUser.user_id == user_id)).first()
|
||||
if exists:
|
||||
logger.warning("MCP install failed: already installed", extra={"user_id": user_id, "mcp_id": mcp_id, "mcp_key": mcp.key})
|
||||
logger.warning(
|
||||
"MCP install failed: already installed", extra={"user_id": user_id, "mcp_id": mcp_id, "mcp_key": mcp.key}
|
||||
)
|
||||
raise HTTPException(status_code=400, detail=_("mcp is installed"))
|
||||
|
||||
install_command: dict = mcp.install_command
|
||||
|
||||
# Pre-instantiate MCP toolkit for authentication
|
||||
config_dict = {
|
||||
"mcpServers": {
|
||||
mcp.key: install_command
|
||||
}
|
||||
}
|
||||
config_dict = {"mcpServers": {mcp.key: install_command}}
|
||||
|
||||
try:
|
||||
success = await pre_instantiate_mcp_toolkit(config_dict)
|
||||
if not success:
|
||||
logger.warning("MCP pre-instantiation failed, continuing with installation", extra={"user_id": user_id, "mcp_id": mcp_id, "mcp_key": mcp.key})
|
||||
logger.warning(
|
||||
"MCP pre-instantiation failed, continuing with installation",
|
||||
extra={"user_id": user_id, "mcp_id": mcp_id, "mcp_key": mcp.key},
|
||||
)
|
||||
else:
|
||||
logger.debug("MCP toolkit pre-instantiated", extra={"mcp_key": mcp.key})
|
||||
except Exception as e:
|
||||
logger.warning("MCP pre-instantiation exception", extra={"user_id": user_id, "mcp_key": mcp.key, "error": str(e)}, exc_info=True)
|
||||
logger.warning(
|
||||
"MCP pre-instantiation exception",
|
||||
extra={"user_id": user_id, "mcp_key": mcp.key, "error": str(e)},
|
||||
exc_info=True,
|
||||
)
|
||||
|
||||
try:
|
||||
mcp_user = McpUser(
|
||||
|
|
@ -183,7 +193,11 @@ async def install(mcp_id: int, session: Session = Depends(session), auth: Auth =
|
|||
logger.info("MCP installed", extra={"user_id": user_id, "mcp_id": mcp_id, "mcp_key": mcp.key})
|
||||
return mcp_user
|
||||
except Exception as e:
|
||||
logger.error("MCP installation failed", extra={"user_id": user_id, "mcp_id": mcp_id, "mcp_key": mcp.key, "error": str(e)}, exc_info=True)
|
||||
logger.error(
|
||||
"MCP installation failed",
|
||||
extra={"user_id": user_id, "mcp_id": mcp_id, "mcp_key": mcp.key, "error": str(e)},
|
||||
exc_info=True,
|
||||
)
|
||||
raise HTTPException(status_code=500, detail="Internal server error")
|
||||
|
||||
|
||||
|
|
@ -200,27 +214,24 @@ async def import_mcp(
|
|||
if not is_valid:
|
||||
logger.warning("Local MCP import validation failed", extra={"user_id": user_id, "error": res})
|
||||
raise HTTPException(status_code=400, detail=res)
|
||||
|
||||
mcp_data: Dict[str, McpServerItem] = res.mcpServers
|
||||
|
||||
mcp_data: dict[str, McpServerItem] = res.mcpServers
|
||||
imported_count = 0
|
||||
|
||||
for name, data in mcp_data.items():
|
||||
config_dict = {
|
||||
"mcpServers": {
|
||||
name: {
|
||||
"command": data.command,
|
||||
"args": data.args,
|
||||
"env": data.env or {}
|
||||
}
|
||||
}
|
||||
}
|
||||
config_dict = {"mcpServers": {name: {"command": data.command, "args": data.args, "env": data.env or {}}}}
|
||||
|
||||
try:
|
||||
success = await pre_instantiate_mcp_toolkit(config_dict)
|
||||
if not success:
|
||||
logger.warning("Local MCP pre-instantiation failed, continuing", extra={"user_id": user_id, "mcp_name": name})
|
||||
logger.warning(
|
||||
"Local MCP pre-instantiation failed, continuing", extra={"user_id": user_id, "mcp_name": name}
|
||||
)
|
||||
except Exception as e:
|
||||
logger.warning("Local MCP pre-instantiation exception", extra={"user_id": user_id, "mcp_name": name, "error": str(e)})
|
||||
logger.warning(
|
||||
"Local MCP pre-instantiation exception",
|
||||
extra={"user_id": user_id, "mcp_name": name, "error": str(e)},
|
||||
)
|
||||
|
||||
try:
|
||||
mcp_user = McpUser(
|
||||
|
|
@ -239,23 +250,27 @@ async def import_mcp(
|
|||
mcp_user.save()
|
||||
imported_count += 1
|
||||
except Exception as e:
|
||||
logger.error("Failed to import local MCP", extra={"user_id": user_id, "mcp_name": name, "error": str(e)}, exc_info=True)
|
||||
|
||||
logger.error(
|
||||
"Failed to import local MCP",
|
||||
extra={"user_id": user_id, "mcp_name": name, "error": str(e)},
|
||||
exc_info=True,
|
||||
)
|
||||
|
||||
logger.info("Local MCPs imported", extra={"user_id": user_id, "count": imported_count})
|
||||
return {"message": "Local MCP servers imported successfully", "count": imported_count}
|
||||
|
||||
|
||||
elif mcp_type == McpImportType.Remote:
|
||||
logger.info("Importing remote MCP server", extra={"user_id": user_id})
|
||||
is_valid, res = validate_mcp_remote_servers(mcp_data)
|
||||
if not is_valid:
|
||||
logger.warning("Remote MCP import validation failed", extra={"user_id": user_id, "error": res})
|
||||
raise HTTPException(status_code=400, detail=res)
|
||||
|
||||
|
||||
data: McpRemoteServer = res
|
||||
|
||||
try:
|
||||
# For remote servers, we don't need to pre-instantiate as they typically don't require authentication
|
||||
# but we can still try to validate the connection if needed
|
||||
# For remote servers, we don't need to pre-instantiate as they typically don't require authentication
|
||||
# but we can still try to validate the connection if needed
|
||||
mcp_user = McpUser(
|
||||
mcp_id=0,
|
||||
user_id=user_id,
|
||||
|
|
@ -265,8 +280,15 @@ async def import_mcp(
|
|||
server_url=data.server_url,
|
||||
)
|
||||
mcp_user.save()
|
||||
logger.info("Remote MCP imported", extra={"user_id": user_id, "server_name": data.server_name, "server_url": data.server_url})
|
||||
logger.info(
|
||||
"Remote MCP imported",
|
||||
extra={"user_id": user_id, "server_name": data.server_name, "server_url": data.server_url},
|
||||
)
|
||||
return mcp_user
|
||||
except Exception as e:
|
||||
logger.error("Remote MCP import failed", extra={"user_id": user_id, "server_name": data.server_name, "error": str(e)}, exc_info=True)
|
||||
raise HTTPException(status_code=500, detail="Internal server error")
|
||||
logger.error(
|
||||
"Remote MCP import failed",
|
||||
extra={"user_id": user_id, "server_name": data.server_name, "error": str(e)},
|
||||
exc_info=True,
|
||||
)
|
||||
raise HTTPException(status_code=500, detail="Internal server error")
|
||||
|
|
|
|||
|
|
@ -12,20 +12,21 @@
|
|||
# limitations under the License.
|
||||
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
|
||||
|
||||
from fastapi import APIRouter, Depends, HTTPException
|
||||
import logging
|
||||
from typing import Any, cast
|
||||
|
||||
import requests
|
||||
from exa_py import Exa
|
||||
from fastapi import APIRouter, Depends, HTTPException
|
||||
|
||||
from app.component.auth import key_must
|
||||
from app.component.environment import env_not_empty
|
||||
from app.model.mcp.proxy import ExaSearch
|
||||
from typing import Any, cast
|
||||
import requests
|
||||
import logging
|
||||
|
||||
logger = logging.getLogger("server_proxy_controller")
|
||||
|
||||
from app.model.user.key import Key
|
||||
|
||||
|
||||
router = APIRouter(prefix="/proxy", tags=["Mcp Servers"])
|
||||
|
||||
|
||||
|
|
@ -41,18 +42,26 @@ def exa_search(search: ExaSearch, key: Key = Depends(key_must)):
|
|||
|
||||
if search.include_text is not None and len(search.include_text) > 0:
|
||||
if len(search.include_text) > 1:
|
||||
logger.warning("Invalid exa search parameter", extra={"param": "include_text", "reason": "more than 1 string"})
|
||||
logger.warning(
|
||||
"Invalid exa search parameter", extra={"param": "include_text", "reason": "more than 1 string"}
|
||||
)
|
||||
raise ValueError("include_text can only contain 1 string")
|
||||
if len(search.include_text[0].split()) > 5:
|
||||
logger.warning("Invalid exa search parameter", extra={"param": "include_text", "reason": "exceeds 5 words"})
|
||||
logger.warning(
|
||||
"Invalid exa search parameter", extra={"param": "include_text", "reason": "exceeds 5 words"}
|
||||
)
|
||||
raise ValueError("include_text string cannot be longer than 5 words")
|
||||
|
||||
if search.exclude_text is not None and len(search.exclude_text) > 0:
|
||||
if len(search.exclude_text) > 1:
|
||||
logger.warning("Invalid exa search parameter", extra={"param": "exclude_text", "reason": "more than 1 string"})
|
||||
logger.warning(
|
||||
"Invalid exa search parameter", extra={"param": "exclude_text", "reason": "more than 1 string"}
|
||||
)
|
||||
raise ValueError("exclude_text can only contain 1 string")
|
||||
if len(search.exclude_text[0].split()) > 5:
|
||||
logger.warning("Invalid exa search parameter", extra={"param": "exclude_text", "reason": "exceeds 5 words"})
|
||||
logger.warning(
|
||||
"Invalid exa search parameter", extra={"param": "exclude_text", "reason": "exceeds 5 words"}
|
||||
)
|
||||
raise ValueError("exclude_text string cannot be longer than 5 words")
|
||||
|
||||
exa = Exa(EXA_API_KEY)
|
||||
|
|
@ -87,7 +96,10 @@ def exa_search(search: ExaSearch, key: Key = Depends(key_must)):
|
|||
)
|
||||
|
||||
result_count = len(results.get("results", [])) if "results" in results else 0
|
||||
logger.info("Exa search completed", extra={"query": search.query, "search_type": search.search_type, "result_count": result_count})
|
||||
logger.info(
|
||||
"Exa search completed",
|
||||
extra={"query": search.query, "search_type": search.search_type, "result_count": result_count},
|
||||
)
|
||||
return results
|
||||
|
||||
except ValueError as e:
|
||||
|
|
@ -112,7 +124,7 @@ def google_search(query: str, search_type: str = "web", key: Key = Depends(key_m
|
|||
search_language = "en"
|
||||
# How many pages to return
|
||||
num_result_pages = 10
|
||||
|
||||
|
||||
# Constructing the URL
|
||||
# Doc: https://developers.google.com/custom-search/v1/using_rest
|
||||
base_url = (
|
||||
|
|
@ -127,7 +139,7 @@ def google_search(query: str, search_type: str = "web", key: Key = Depends(key_m
|
|||
url = base_url
|
||||
|
||||
responses = []
|
||||
|
||||
|
||||
try:
|
||||
# Make the GET request
|
||||
result = requests.get(url)
|
||||
|
|
@ -194,15 +206,20 @@ def google_search(query: str, search_type: str = "web", key: Key = Depends(key_m
|
|||
"url": link,
|
||||
}
|
||||
responses.append(response)
|
||||
|
||||
logger.info("Google search completed", extra={"query": query, "search_type": search_type, "result_count": len(responses)})
|
||||
|
||||
logger.info(
|
||||
"Google search completed",
|
||||
extra={"query": query, "search_type": search_type, "result_count": len(responses)},
|
||||
)
|
||||
else:
|
||||
error_info = data.get("error", {})
|
||||
logger.error("Google search API error", extra={"query": query, "api_error": error_info})
|
||||
raise HTTPException(status_code=500, detail="Internal server error")
|
||||
|
||||
except Exception as e:
|
||||
logger.error("Google search failed", extra={"query": query, "search_type": search_type, "error": str(e)}, exc_info=True)
|
||||
logger.error(
|
||||
"Google search failed", extra={"query": query, "search_type": search_type, "error": str(e)}, exc_info=True
|
||||
)
|
||||
raise HTTPException(status_code=500, detail="Internal server error")
|
||||
|
||||
return responses
|
||||
|
||||
return responses
|
||||
|
|
|
|||
|
|
@ -12,18 +12,24 @@
|
|||
# limitations under the License.
|
||||
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
|
||||
|
||||
import os
|
||||
from typing import List, Optional
|
||||
from fastapi import Depends, HTTPException, Query, Response, APIRouter
|
||||
from sqlmodel import Session, select
|
||||
from app.component.database import session
|
||||
from app.component.auth import Auth, auth_must
|
||||
from fastapi_babel import _
|
||||
from app.model.mcp.mcp_user import McpUser, McpUserIn, McpUserOut, McpUserUpdate, Status
|
||||
from app.model.mcp.mcp import Mcp
|
||||
from camel.toolkits.mcp_toolkit import MCPToolkit
|
||||
from app.component.environment import env
|
||||
import logging
|
||||
import os
|
||||
|
||||
from camel.toolkits.mcp_toolkit import MCPToolkit
|
||||
from fastapi import APIRouter, Depends, HTTPException, Response
|
||||
from fastapi_babel import _
|
||||
from sqlmodel import Session, select
|
||||
|
||||
from app.component.auth import Auth, auth_must
|
||||
from app.component.database import session
|
||||
from app.component.environment import env
|
||||
from app.model.mcp.mcp import Mcp
|
||||
from app.model.mcp.mcp_user import (
|
||||
McpUser,
|
||||
McpUserIn,
|
||||
McpUserOut,
|
||||
McpUserUpdate,
|
||||
)
|
||||
|
||||
logger = logging.getLogger("server_mcp_user_controller")
|
||||
|
||||
|
|
@ -48,8 +54,7 @@ async def pre_instantiate_mcp_toolkit(config_dict: dict) -> bool:
|
|||
# Set global auth directory to persist authentication across tasks
|
||||
if "MCP_REMOTE_CONFIG_DIR" not in server_config["env"]:
|
||||
server_config["env"]["MCP_REMOTE_CONFIG_DIR"] = env(
|
||||
"MCP_REMOTE_CONFIG_DIR",
|
||||
os.path.expanduser("~/.mcp-auth")
|
||||
"MCP_REMOTE_CONFIG_DIR", os.path.expanduser("~/.mcp-auth")
|
||||
)
|
||||
|
||||
# Create MCP toolkit and attempt to connect
|
||||
|
|
@ -69,9 +74,9 @@ async def pre_instantiate_mcp_toolkit(config_dict: dict) -> bool:
|
|||
return False
|
||||
|
||||
|
||||
@router.get("/mcp/users", name="list mcp users", response_model=List[McpUserOut])
|
||||
@router.get("/mcp/users", name="list mcp users", response_model=list[McpUserOut])
|
||||
async def list_mcp_users(
|
||||
mcp_id: Optional[int] = None,
|
||||
mcp_id: int | None = None,
|
||||
session: Session = Depends(session),
|
||||
auth: Auth = Depends(auth_must),
|
||||
):
|
||||
|
|
@ -95,7 +100,9 @@ async def get_mcp_user(mcp_user_id: int, session: Session = Depends(session), au
|
|||
if not mcp_user:
|
||||
logger.warning("MCP user not found", extra={"user_id": auth.user.id, "mcp_user_id": mcp_user_id})
|
||||
raise HTTPException(status_code=404, detail=_("McpUser not found"))
|
||||
logger.debug("MCP user retrieved", extra={"user_id": auth.user.id, "mcp_user_id": mcp_user_id, "mcp_id": mcp_user.mcp_id})
|
||||
logger.debug(
|
||||
"MCP user retrieved", extra={"user_id": auth.user.id, "mcp_user_id": mcp_user_id, "mcp_id": mcp_user.mcp_id}
|
||||
)
|
||||
return mcp_user
|
||||
|
||||
|
||||
|
|
@ -104,10 +111,8 @@ async def create_mcp_user(mcp_user: McpUserIn, session: Session = Depends(sessio
|
|||
"""Create MCP user installation."""
|
||||
user_id = auth.user.id
|
||||
mcp_id = mcp_user.mcp_id
|
||||
|
||||
exists = session.exec(
|
||||
select(McpUser).where(McpUser.mcp_id == mcp_id, McpUser.user_id == user_id)
|
||||
).first()
|
||||
|
||||
exists = session.exec(select(McpUser).where(McpUser.mcp_id == mcp_id, McpUser.user_id == user_id)).first()
|
||||
if exists:
|
||||
logger.warning("MCP already installed", extra={"user_id": user_id, "mcp_id": mcp_id})
|
||||
raise HTTPException(status_code=400, detail=_("mcp is installed"))
|
||||
|
|
@ -115,18 +120,21 @@ async def create_mcp_user(mcp_user: McpUserIn, session: Session = Depends(sessio
|
|||
# Get MCP configuration from the main Mcp table
|
||||
mcp = session.get(Mcp, mcp_id)
|
||||
if mcp and mcp.install_command:
|
||||
config_dict = {
|
||||
"mcpServers": {
|
||||
mcp.key: mcp.install_command
|
||||
}
|
||||
}
|
||||
config_dict = {"mcpServers": {mcp.key: mcp.install_command}}
|
||||
|
||||
try:
|
||||
success = await pre_instantiate_mcp_toolkit(config_dict)
|
||||
if not success:
|
||||
logger.warning("MCP pre-instantiation failed, continuing", extra={"user_id": user_id, "mcp_id": mcp_id, "mcp_key": mcp.key})
|
||||
logger.warning(
|
||||
"MCP pre-instantiation failed, continuing",
|
||||
extra={"user_id": user_id, "mcp_id": mcp_id, "mcp_key": mcp.key},
|
||||
)
|
||||
except Exception as e:
|
||||
logger.warning("MCP pre-instantiation exception", extra={"user_id": user_id, "mcp_id": mcp_id, "error": str(e)}, exc_info=True)
|
||||
logger.warning(
|
||||
"MCP pre-instantiation exception",
|
||||
extra={"user_id": user_id, "mcp_id": mcp_id, "error": str(e)},
|
||||
exc_info=True,
|
||||
)
|
||||
|
||||
try:
|
||||
db_mcp_user = McpUser(mcp_id=mcp_id, user_id=user_id, env=mcp_user.env)
|
||||
|
|
@ -137,7 +145,9 @@ async def create_mcp_user(mcp_user: McpUserIn, session: Session = Depends(sessio
|
|||
return db_mcp_user
|
||||
except Exception as e:
|
||||
session.rollback()
|
||||
logger.error("MCP user creation failed", extra={"user_id": user_id, "mcp_id": mcp_id, "error": str(e)}, exc_info=True)
|
||||
logger.error(
|
||||
"MCP user creation failed", extra={"user_id": user_id, "mcp_id": mcp_id, "error": str(e)}, exc_info=True
|
||||
)
|
||||
raise HTTPException(status_code=500, detail="Internal server error")
|
||||
|
||||
|
||||
|
|
@ -155,9 +165,11 @@ async def update_mcp_user(
|
|||
logger.warning("MCP user not found for update", extra={"user_id": user_id, "mcp_user_id": id})
|
||||
raise HTTPException(status_code=404, detail=_("Mcp Info not found"))
|
||||
if model.user_id != user_id:
|
||||
logger.warning("Unauthorized MCP user update", extra={"user_id": user_id, "mcp_user_id": id, "owner_id": model.user_id})
|
||||
logger.warning(
|
||||
"Unauthorized MCP user update", extra={"user_id": user_id, "mcp_user_id": id, "owner_id": model.user_id}
|
||||
)
|
||||
raise HTTPException(status_code=400, detail=_("current user have no permission to modify"))
|
||||
|
||||
|
||||
try:
|
||||
update_data = update_item.model_dump(exclude_unset=True)
|
||||
model.update_fields(update_data)
|
||||
|
|
@ -166,7 +178,9 @@ async def update_mcp_user(
|
|||
logger.info("MCP user updated", extra={"user_id": user_id, "mcp_user_id": id})
|
||||
return model
|
||||
except Exception as e:
|
||||
logger.error("MCP user update failed", extra={"user_id": user_id, "mcp_user_id": id, "error": str(e)}, exc_info=True)
|
||||
logger.error(
|
||||
"MCP user update failed", extra={"user_id": user_id, "mcp_user_id": id, "error": str(e)}, exc_info=True
|
||||
)
|
||||
raise HTTPException(status_code=500, detail="Internal server error")
|
||||
|
||||
|
||||
|
|
@ -178,13 +192,19 @@ async def delete_mcp_user(mcp_user_id: int, session: Session = Depends(session),
|
|||
if not db_mcp_user:
|
||||
logger.warning("MCP user not found for deletion", extra={"user_id": user_id, "mcp_user_id": mcp_user_id})
|
||||
raise HTTPException(status_code=404, detail=_("Mcp Info not found"))
|
||||
|
||||
|
||||
try:
|
||||
session.delete(db_mcp_user)
|
||||
session.commit()
|
||||
logger.info("MCP user deleted", extra={"user_id": user_id, "mcp_user_id": mcp_user_id, "mcp_id": db_mcp_user.mcp_id})
|
||||
logger.info(
|
||||
"MCP user deleted", extra={"user_id": user_id, "mcp_user_id": mcp_user_id, "mcp_id": db_mcp_user.mcp_id}
|
||||
)
|
||||
return Response(status_code=204)
|
||||
except Exception as e:
|
||||
session.rollback()
|
||||
logger.error("MCP user deletion failed", extra={"user_id": user_id, "mcp_user_id": mcp_user_id, "error": str(e)}, exc_info=True)
|
||||
raise HTTPException(status_code=500, detail="Internal server error")
|
||||
logger.error(
|
||||
"MCP user deletion failed",
|
||||
extra={"user_id": user_id, "mcp_user_id": mcp_user_id, "error": str(e)},
|
||||
exc_info=True,
|
||||
)
|
||||
raise HTTPException(status_code=500, detail="Internal server error")
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue