feat: add oauth schema to datasource

This commit is contained in:
Yeuoly 2025-05-21 20:29:59 +08:00
parent 9bafd3a226
commit 5fa2aca2c8
1 changed files with 2 additions and 0 deletions

View File

@ -7,6 +7,7 @@ from typing import Any, Optional, Union
from pydantic import BaseModel, ConfigDict, Field, ValidationInfo, field_serializer, field_validator, model_validator
from core.entities.provider_entities import ProviderConfig
from core.plugin.entities.oauth import OAuthSchema
from core.plugin.entities.parameters import (
PluginParameter,
PluginParameterOption,
@ -349,6 +350,7 @@ class ToolProviderEntity(BaseModel):
identity: ToolProviderIdentity
plugin_id: Optional[str] = None
credentials_schema: list[ProviderConfig] = Field(default_factory=list)
oauth_schema: Optional[OAuthSchema] = Field(default=None, description="The oauth schema of the tool provider")
class ToolProviderEntityWithPlugin(ToolProviderEntity):