mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-22 10:25:23 -04:00
Harry
1fa18bfd1c
- Renamed `WebhookValidationError` to `TriggerValidationError` for consistency in error handling across trigger operations. - Updated references to the new `TriggerValidationError` in the GitHub provider and trigger interfaces. - Enhanced the `TriggerProvider` class to raise the new validation error for webhook signature issues. - Adjusted the `IssueCommentTrigger` to return the correct event response type. - Improved type hints and imports in various modules for better clarity and organization.
96 lines
4.2 KiB
Python
96 lines
4.2 KiB
Python
import json
|
|
from collections.abc import Generator
|
|
from datetime import datetime
|
|
from typing import Any
|
|
from urllib.parse import quote
|
|
|
|
import requests
|
|
|
|
from dify_plugin import Tool
|
|
from dify_plugin.entities import I18nObject, ParameterOption
|
|
from dify_plugin.entities.provider_config import CredentialType
|
|
from dify_plugin.entities.tool import ToolInvokeMessage
|
|
|
|
|
|
class GithubRepositoriesTool(Tool):
|
|
def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessage, None, None]:
|
|
"""
|
|
invoke tools
|
|
"""
|
|
top_n = tool_parameters.get("top_n", 5)
|
|
query = tool_parameters.get("query", "")
|
|
credential_type = self.runtime.credential_type
|
|
if not query:
|
|
yield self.create_text_message("Please input symbol")
|
|
|
|
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}",
|
|
# fixed api version
|
|
"X-GitHub-Api-Version": "2022-11-28",
|
|
}
|
|
s = requests.session()
|
|
api_domain = "https://api.github.com"
|
|
response = s.request(
|
|
method="GET",
|
|
headers=headers,
|
|
url=f"{api_domain}/search/repositories?q={quote(query)}&sort=stars&per_page={top_n}&order=desc",
|
|
)
|
|
response_data = response.json()
|
|
if response.status_code == 200 and isinstance(response_data.get("items"), list):
|
|
contents = []
|
|
if len(response_data.get("items")) > 0:
|
|
for item in response_data.get("items"):
|
|
content = {}
|
|
updated_at_object = datetime.strptime(item["updated_at"], "%Y-%m-%dT%H:%M:%SZ")
|
|
content["owner"] = item["owner"]["login"]
|
|
content["name"] = item["name"]
|
|
if item["description"] is not None:
|
|
content["description"] = (
|
|
item["description"][:100] + "..."
|
|
if len(item["description"]) > 100
|
|
else item["description"]
|
|
)
|
|
else:
|
|
content["description"] = ""
|
|
content["url"] = item["html_url"]
|
|
content["star"] = item["watchers"]
|
|
content["forks"] = item["forks"]
|
|
content["updated"] = updated_at_object.strftime("%Y-%m-%d")
|
|
contents.append(content)
|
|
s.close()
|
|
yield self.create_text_message(
|
|
self.session.model.summary.invoke(
|
|
text=json.dumps(contents, ensure_ascii=False),
|
|
instruction="Summarize the text",
|
|
)
|
|
)
|
|
else:
|
|
yield self.create_text_message(f"No items related to {query} were found.")
|
|
else:
|
|
yield self.create_text_message(response.json().get("message"))
|
|
except Exception as e:
|
|
yield self.create_text_message(f"GitHub API Key and Api Version is invalid. {e}")
|
|
|
|
def _fetch_parameter_options(self, parameter: str) -> list[ParameterOption]:
|
|
return [
|
|
ParameterOption(
|
|
value="iamjoel",
|
|
label=I18nObject(en_US="Joel"),
|
|
icon="https://avatars.githubusercontent.com/u/2120155?s=40&v=4",
|
|
),
|
|
ParameterOption(
|
|
value="yeuoly",
|
|
label=I18nObject(en_US="Yeuoly"),
|
|
icon="https://avatars.githubusercontent.com/u/45712896?s=60&v=4",
|
|
),
|
|
]
|