mirror of
https://github.com/eigent-ai/eigent.git
synced 2026-05-25 06:27:38 +00:00
90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
|
|
|
|
from enum import IntEnum
|
|
from typing import TYPE_CHECKING
|
|
|
|
from pydantic import BaseModel
|
|
from sqlalchemy import Column, SmallInteger, String
|
|
from sqlalchemy.orm import Mapped
|
|
from sqlalchemy_utils import ChoiceType
|
|
from sqlmodel import JSON, Field, Relationship
|
|
|
|
from app.model.abstract.model import AbstractModel, DefaultTimes
|
|
from app.model.mcp.category import Category, CategoryOut
|
|
from app.model.mcp.mcp_env import McpEnv
|
|
from app.type.pydantic import HttpUrlStr
|
|
|
|
if TYPE_CHECKING:
|
|
from app.model.mcp.mcp_user import McpUser
|
|
|
|
|
|
class Status(IntEnum):
|
|
Online = 1
|
|
Offline = -1
|
|
|
|
|
|
class McpType(IntEnum):
|
|
Local = 1
|
|
Remote = 2
|
|
|
|
|
|
class Mcp(AbstractModel, DefaultTimes, table=True):
|
|
id: int = Field(default=None, primary_key=True)
|
|
category_id: int = Field(foreign_key="category.id")
|
|
name: str
|
|
key: str = Field(sa_column=Column(String(128)))
|
|
description: str = ""
|
|
home_page: str = Field(default="", sa_column=Column(String(1024)))
|
|
type: McpType = Field(default=McpType.Local, sa_column=Column(ChoiceType(McpType, SmallInteger())))
|
|
status: Status = Field(default=Status.Online, sa_column=Column(ChoiceType(Status, SmallInteger())))
|
|
sort: int = Field(default=0, sa_column=Column(SmallInteger))
|
|
server_name: str = Field(default="", sa_column=Column(String(128)))
|
|
install_command: dict = Field(default="{}", sa_column=Column(JSON))
|
|
"""{
|
|
"command": "uvx",
|
|
"args": ["mcp-server-everything-search"],
|
|
"env": {
|
|
"EVERYTHING_SDK_PATH": "path/to/Everything-SDK/dll/Everything64.dll"
|
|
}
|
|
}"""
|
|
|
|
category: Mapped[Category] = Relationship()
|
|
envs: Mapped[list[McpEnv]] = Relationship()
|
|
# user_env: Mapped[McpUser] = Relationship()
|
|
|
|
mcp_user: list["McpUser"] = Relationship(back_populates="mcp")
|
|
|
|
|
|
class McpIn(BaseModel):
|
|
category_id: int
|
|
name: str
|
|
key: str
|
|
description: str
|
|
home_page: HttpUrlStr
|
|
type: McpType
|
|
status: Status
|
|
install_command: dict
|
|
|
|
|
|
class McpOut(McpIn):
|
|
id: int
|
|
category: CategoryOut | None = None
|
|
# envs: list[McpEnvOut] = []
|
|
|
|
|
|
class McpInfo(BaseModel):
|
|
id: int
|
|
name: str
|
|
key: str
|