mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-22 10:25:23 -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>
70 lines
2.8 KiB
Python
70 lines
2.8 KiB
Python
import base64
|
|
from collections.abc import Generator
|
|
from typing import Any
|
|
|
|
import requests
|
|
|
|
from dify_plugin import Tool
|
|
from dify_plugin.entities.provider_config import CredentialType
|
|
from dify_plugin.entities.tool import ToolInvokeMessage
|
|
from dify_plugin.errors.model import InvokeError
|
|
|
|
|
|
class GithubRepositoryReadmeTool(Tool):
|
|
def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessage, None, None]:
|
|
"""
|
|
invoke tools
|
|
"""
|
|
owner = tool_parameters.get("owner", "")
|
|
repo = tool_parameters.get("repo", "")
|
|
ref = tool_parameters.get("ref", "")
|
|
dir_path = tool_parameters.get("dir", "")
|
|
credential_type = self.runtime.credential_type
|
|
if not owner:
|
|
yield self.create_text_message("Please input owner")
|
|
if not repo:
|
|
yield self.create_text_message("Please input repo")
|
|
if credential_type == CredentialType.API_KEY and "access_tokens" not in self.runtime.credentials:
|
|
yield self.create_text_message("GitHub API Access Tokens is required.")
|
|
|
|
if credential_type == CredentialType.OAUTH and "access_tokens" not in self.runtime.credentials:
|
|
yield self.create_text_message("GitHub OAuth Access Tokens is required.")
|
|
|
|
access_token = self.runtime.credentials.get("access_tokens")
|
|
try:
|
|
headers = {
|
|
"Content-Type": "application/vnd.github+json",
|
|
"Authorization": f"Bearer {access_token}",
|
|
"X-GitHub-Api-Version": "2022-11-28",
|
|
}
|
|
s = requests.session()
|
|
api_domain = "https://api.github.com"
|
|
url = f"{api_domain}/repos/{owner}/{repo}/readme"
|
|
if dir_path:
|
|
url = f"{url}/{dir_path}"
|
|
if ref:
|
|
url = f"{url}?ref={ref}"
|
|
response = s.request(
|
|
method="GET",
|
|
headers=headers,
|
|
url=url,
|
|
)
|
|
response_data = response.json()
|
|
if response.status_code == 200:
|
|
if response_data.get("encoding") != "base64":
|
|
raise InvokeError(
|
|
f"Can not get base64 encoded readme, response encoding is {response_data.get('encoding')}"
|
|
)
|
|
content = response_data.get("content")
|
|
if not content:
|
|
raise InvokeError("README content is empty")
|
|
decoded_bytes = base64.b64decode(content)
|
|
decoded_str = decoded_bytes.decode("utf-8")
|
|
yield self.create_text_message(decoded_str)
|
|
else:
|
|
raise InvokeError(f"Request failed: {response.status_code} {response_data.get('message')}")
|
|
except InvokeError as e:
|
|
raise e
|
|
except Exception as e:
|
|
raise InvokeError(f"Request failed: {e}") from e
|