feat: add server identifer field

pull/22036/head
Novice 11 months ago
parent 8f61408cb6
commit ef35e9febb

@ -41,6 +41,7 @@ class ToolProviderApiEntity(BaseModel):
# MCP # MCP
server_url: Optional[str] = Field(default="", description="The server url of the tool") server_url: Optional[str] = Field(default="", description="The server url of the tool")
updated_at: int = Field(default_factory=lambda: int(datetime.now().timestamp())) 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") @field_validator("tools", mode="before")
@classmethod @classmethod
@ -62,6 +63,7 @@ class ToolProviderApiEntity(BaseModel):
optional_fields = self.optional_field("server_url", self.server_url) optional_fields = self.optional_field("server_url", self.server_url)
if self.type == ToolProviderType.MCP.value: if self.type == ToolProviderType.MCP.value:
optional_fields.update(self.optional_field("updated_at", self.updated_at)) optional_fields.update(self.optional_field("updated_at", self.updated_at))
optional_fields.update(self.optional_field("server_identifier", self.server_identifier))
return { return {
"id": self.id, "id": self.id,
"author": self.author, "author": self.author,

@ -205,6 +205,7 @@ class ToolTransformService:
updated_at=int(db_provider.updated_at.timestamp()), updated_at=int(db_provider.updated_at.timestamp()),
label=I18nObject(en_US=db_provider.name, zh_Hans=db_provider.name), label=I18nObject(en_US=db_provider.name, zh_Hans=db_provider.name),
description=I18nObject(en_US="", zh_Hans=""), description=I18nObject(en_US="", zh_Hans=""),
server_identifier=db_provider.server_identifier,
) )
@staticmethod @staticmethod

Loading…
Cancel
Save