estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

chore(api/core): apply ruff reformatting (#7624)

Browse Source
This commit is contained in:
Bowen Liang
2024-09-10 17:00:20 +08:00
committed by GitHub
parent 178730266d
commit 2cf1187b32
724 changed files with 21180 additions and 21123 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/core/tools/provider/builtin/stackexchange/stackexchange.py
View File

@@ -11,16 +11,15 @@ class StackExchangeProvider(BuiltinToolProviderController):
"credentials": credentials,
}
).invoke(
user_id='',
user_id="",
tool_parameters={
"intitle": "Test",
"sort": "relevance",
"sort": "relevance",
"order": "desc",
"site": "stackoverflow",
"accepted": True,
"pagesize": 1
"pagesize": 1,
},
)
except Exception as e:
raise ToolProviderCredentialValidationError(str(e))

8
api/core/tools/provider/builtin/stackexchange/tools/fetchAnsByStackExQuesID.py
View File

@@ -17,7 +17,9 @@ class FetchAnsByStackExQuesIDInput(BaseModel):
class FetchAnsByStackExQuesIDTool(BuiltinTool):
def _invoke(self, user_id: str, tool_parameters: dict[str, Any]) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
def _invoke(
self, user_id: str, tool_parameters: dict[str, Any]
) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
input = FetchAnsByStackExQuesIDInput(**tool_parameters)
params = {
@@ -26,7 +28,7 @@ class FetchAnsByStackExQuesIDTool(BuiltinTool):
"order": input.order,
"sort": input.sort,
"pagesize": input.pagesize,
"page": input.page
"page": input.page,
}
response = requests.get(f"https://api.stackexchange.com/2.3/questions/{input.id}/answers", params=params)
@@ -34,4 +36,4 @@ class FetchAnsByStackExQuesIDTool(BuiltinTool):
if response.status_code == 200:
return self.create_text_message(self.summary(user_id=user_id, content=response.text))
else:
return self.create_text_message(f"API request failed with status code {response.status_code}")
return self.create_text_message(f"API request failed with status code {response.status_code}")

14
api/core/tools/provider/builtin/stackexchange/tools/searchStackExQuestions.py
View File

@@ -9,26 +9,28 @@ from core.tools.tool.builtin_tool import BuiltinTool
class SearchStackExQuestionsInput(BaseModel):
intitle: str = Field(..., description="The search query.")
sort: str = Field(..., description="The sort order - relevance, activity, votes, creation.")
sort: str = Field(..., description="The sort order - relevance, activity, votes, creation.")
order: str = Field(..., description="asc or desc")
site: str = Field(..., description="The Stack Exchange site.")
tagged: str = Field(None, description="Semicolon-separated tags to include.")
nottagged: str = Field(None, description="Semicolon-separated tags to exclude.")
accepted: bool = Field(..., description="true for only accepted answers, false otherwise")
accepted: bool = Field(..., description="true for only accepted answers, false otherwise")
pagesize: int = Field(..., description="Number of results per page")
class SearchStackExQuestionsTool(BuiltinTool):
def _invoke(self, user_id: str, tool_parameters: dict[str, Any]) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
def _invoke(
self, user_id: str, tool_parameters: dict[str, Any]
) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
input = SearchStackExQuestionsInput(**tool_parameters)
params = {
"intitle": input.intitle,
"sort": input.sort,
"order": input.order,
"order": input.order,
"site": input.site,
"accepted": input.accepted,
"pagesize": input.pagesize
"pagesize": input.pagesize,
}
if input.tagged:
params["tagged"] = input.tagged
@@ -40,4 +42,4 @@ class SearchStackExQuestionsTool(BuiltinTool):
if response.status_code == 200:
return self.create_text_message(self.summary(user_id=user_id, content=response.text))
else:
return self.create_text_message(f"API request failed with status code {response.status_code}")
return self.create_text_message(f"API request failed with status code {response.status_code}")