mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-24 21:15:37 -04:00
Maries
c6f83a63e1
* chore: fix ruff issue * feat(oauth): implement OAuth * feat(invoke-message): refactor message handling and introduce InvokeMessage class * feat(plugin-oauth): add credential_id and credential_type to tool parameters * feat(plugin-oauth): add credential_id and credential_type to tool parameters * chore: update dify_plugin version to 0.5.0b4 and clean up github.yaml * chore: update plugin version to 0.1.2 in manifest.yaml * feat(session): session context and tool backwards invocation credential support * feat(oauth): session context and tool backwards invocation credential support * feat: update README and requirements for OAuth support in version 0.4.2 * feat: add .gitignore to exclude IDE files and secret keys * chore: apply ruff * feat: bump version to 0.4.2b1 * feat: update GitHub plugin configuration for OAuth support and improve credential handling * feat: update .gitignore to exclude dify plugin files and public keys * feat: fix credential validation for GitHub API and bump version to 0.2.1 * feat: update GitHub plugin to support multiple access tokens and bump version to 0.2.5 * chore: apply ruff * feat: add ToolProviderOAuthError for improved OAuth error handling in GitHub plugin * chore: apply ruff * chore: bump version to 0.4.2 * chore: update examples sdk version to 0.4.2 * fix: thread deadlock in PluginRunner when running tests without gevent monkey patching * feat: add support for refreshing OAuth credentials in Plugin and GitHub provider * feat: refactor OAuth credential handling to return structured OAuthCredentials object * apply ruff * feat: refactor OAuth credential handling to use ToolOAuthCredentials for improved structure * feat: reorganize imports in __init__.py for improved clarity and structure * feat: add Microsoft To Do plugin for refresh token example * chore: apply ruff * fix: update author in GitHub configuration and clean up Microsoft To Do schema * chore: bump version to 0.4.2b2 in pyproject.toml * feat: update Microsoft To Do plugin to handle OAuth token encoding and version bump * feat:remove inelegant example * chore: update dify_plugin version to 0.4.2 * chore: bump version to 0.4.2 in pyproject.toml --------- Co-authored-by: Yeuoly <admin@srmxy.cn>
80 lines
2.8 KiB
Python
80 lines
2.8 KiB
Python
import inspect
|
|
import logging
|
|
from collections.abc import Callable
|
|
from typing import Any
|
|
|
|
from dify_plugin.core.runtime import Session
|
|
from dify_plugin.core.server.__base.request_reader import RequestReader
|
|
from dify_plugin.core.server.__base.response_writer import ResponseWriter
|
|
|
|
logger = logging.getLogger(__file__)
|
|
|
|
|
|
class Route:
|
|
filter: Callable[[dict], bool]
|
|
func: Callable
|
|
|
|
def __init__(self, filter: Callable[[dict], bool], func) -> None: # noqa: A002
|
|
self.filter = filter
|
|
self.func = func
|
|
|
|
|
|
class Router:
|
|
routes: list[Route]
|
|
request_reader: RequestReader
|
|
|
|
def __init__(self, request_reader: RequestReader, response_writer: ResponseWriter | None) -> None:
|
|
self.routes = []
|
|
self.request_reader = request_reader
|
|
self.response_writer = response_writer
|
|
|
|
def register_route(self, f: Callable, filter: Callable[[dict], bool], instance: Any = None): # noqa: A002
|
|
sig = inspect.signature(f)
|
|
parameters = list(sig.parameters.values())
|
|
if len(parameters) == 0:
|
|
raise ValueError("Route function must have at least one parameter")
|
|
|
|
if instance:
|
|
# get first parameter of func
|
|
parameter = parameters[2]
|
|
# get annotation of the first parameter
|
|
annotation = parameter.annotation
|
|
|
|
def wrapper(session: Session, data: dict):
|
|
try:
|
|
data = annotation(**data)
|
|
except TypeError as e:
|
|
if not self.response_writer:
|
|
logger.exception("failed to route request: %s")
|
|
else:
|
|
self.response_writer.error(
|
|
session_id=session.session_id,
|
|
data={"error": str(e), "error_type": type(e).__name__},
|
|
)
|
|
return f(instance, session, data)
|
|
else:
|
|
# get first parameter of func
|
|
parameter = parameters[1]
|
|
# get annotation of the first parameter
|
|
annotation = parameter.annotation
|
|
|
|
def wrapper(session: Session, data: dict):
|
|
try:
|
|
data = annotation(**data)
|
|
except TypeError as e:
|
|
if not self.response_writer:
|
|
logger.exception("failed to route request: %s")
|
|
else:
|
|
self.response_writer.error(
|
|
session_id=session.session_id,
|
|
data={"error": str(e), "error_type": type(e).__name__},
|
|
)
|
|
return f(session, data)
|
|
|
|
self.routes.append(Route(filter, wrapper))
|
|
|
|
def dispatch(self, session: Session, data: dict) -> Any:
|
|
for route in self.routes:
|
|
if route.filter(data):
|
|
return route.func(session, data)
|