feat: support tool plugin id

This commit is contained in:
Yeuoly 2024-10-29 12:32:11 +08:00
parent 5bf86ff66d
commit 739e15f88b
No known key found for this signature in database
GPG Key ID: A66E7E320FB19F61
2 changed files with 7 additions and 0 deletions

View File

@ -110,6 +110,12 @@ class PluginEntity(BasePluginEntity):
source: PluginInstallationSource
meta: Mapping[str, Any]
@model_validator(mode="after")
def set_plugin_id(self):
if self.declaration.tool:
self.declaration.tool.plugin_id = self.plugin_id
return self
class GithubPackage(BaseModel):
repo: str

View File

@ -377,6 +377,7 @@ class ToolEntity(BaseModel):
class ToolProviderEntity(BaseModel):
identity: ToolProviderIdentity
plugin_id: Optional[str] = Field(None, description="The id of the plugin")
credentials_schema: list[ProviderConfig] = Field(default_factory=list)