From ef35e9febb5b245a3dbc7f3df02f34e74b9bd0d7 Mon Sep 17 00:00:00 2001 From: Novice Date: Thu, 26 Jun 2025 10:32:05 +0800 Subject: [PATCH] feat: add server identifer field --- api/core/tools/entities/api_entities.py | 2 ++ api/services/tools/tools_transform_service.py | 1 + 2 files changed, 3 insertions(+) diff --git a/api/core/tools/entities/api_entities.py b/api/core/tools/entities/api_entities.py index 025a91fae5..b94d6bba21 100644 --- a/api/core/tools/entities/api_entities.py +++ b/api/core/tools/entities/api_entities.py @@ -41,6 +41,7 @@ class ToolProviderApiEntity(BaseModel): # MCP server_url: Optional[str] = Field(default="", description="The server url of the tool") updated_at: int = Field(default_factory=lambda: int(datetime.now().timestamp())) + server_identifier: Optional[str] = Field(default="", description="The server identifier of the MCP tool") @field_validator("tools", mode="before") @classmethod @@ -62,6 +63,7 @@ class ToolProviderApiEntity(BaseModel): optional_fields = self.optional_field("server_url", self.server_url) if self.type == ToolProviderType.MCP.value: optional_fields.update(self.optional_field("updated_at", self.updated_at)) + optional_fields.update(self.optional_field("server_identifier", self.server_identifier)) return { "id": self.id, "author": self.author, diff --git a/api/services/tools/tools_transform_service.py b/api/services/tools/tools_transform_service.py index f3ad123e1c..b78d4581e5 100644 --- a/api/services/tools/tools_transform_service.py +++ b/api/services/tools/tools_transform_service.py @@ -205,6 +205,7 @@ class ToolTransformService: updated_at=int(db_provider.updated_at.timestamp()), label=I18nObject(en_US=db_provider.name, zh_Hans=db_provider.name), description=I18nObject(en_US="", zh_Hans=""), + server_identifier=db_provider.server_identifier, ) @staticmethod