From 5fa2aca2c8d7f9edd00c32ad0240f9aae8ba818b Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Wed, 21 May 2025 20:29:59 +0800 Subject: [PATCH] feat: add oauth schema to datasource --- api/core/tools/entities/tool_entities.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/api/core/tools/entities/tool_entities.py b/api/core/tools/entities/tool_entities.py index 37375f4a71..9884d93e9d 100644 --- a/api/core/tools/entities/tool_entities.py +++ b/api/core/tools/entities/tool_entities.py @@ -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):