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

feat: mypy for all type check (#10921)

Browse Source
This commit is contained in:
yihong
2024-12-24 18:38:51 +08:00
committed by GitHub
parent c91e8b1737
commit 56e15d09a9
584 changed files with 3975 additions and 2826 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
sdks/python-client/dify_client/client.py
View File

@@ -160,7 +160,10 @@ class WorkflowClient(DifyClient):
class KnowledgeBaseClient(DifyClient):
def __init__(
self, api_key, base_url: str = "https://api.dify.ai/v1", dataset_id: str = None
self,
api_key,
base_url: str = "https://api.dify.ai/v1",
dataset_id: str | None = None,
):
"""
Construct a KnowledgeBaseClient object.
@@ -187,7 +190,9 @@ class KnowledgeBaseClient(DifyClient):
"GET", f"/datasets?page={page}&limit={page_size}", **kwargs
)
def create_document_by_text(self, name, text, extra_params: dict = None, **kwargs):
def create_document_by_text(
self, name, text, extra_params: dict | None = None, **kwargs
):
"""
Create a document by text.
@@ -225,7 +230,7 @@ class KnowledgeBaseClient(DifyClient):
return self._send_request("POST", url, json=data, **kwargs)
def update_document_by_text(
self, document_id, name, text, extra_params: dict = None, **kwargs
self, document_id, name, text, extra_params: dict | None = None, **kwargs
):
"""
Update a document by text.
@@ -262,7 +267,7 @@ class KnowledgeBaseClient(DifyClient):
return self._send_request("POST", url, json=data, **kwargs)
def create_document_by_file(
self, file_path, original_document_id=None, extra_params: dict = None
self, file_path, original_document_id=None, extra_params: dict | None = None
):
"""
Create a document by file.
@@ -304,7 +309,7 @@ class KnowledgeBaseClient(DifyClient):
)
def update_document_by_file(
self, document_id, file_path, extra_params: dict = None
self, document_id, file_path, extra_params: dict | None = None
):
"""
Update a document by file.
@@ -372,7 +377,11 @@ class KnowledgeBaseClient(DifyClient):
return self._send_request("DELETE", url)
def list_documents(
self, page: int = None, page_size: int = None, keyword: str = None, **kwargs
self,
page: int | None = None,
page_size: int | None = None,
keyword: str | None = None,
**kwargs,
):
"""
Get a list of documents in this dataset.
@@ -402,7 +411,11 @@ class KnowledgeBaseClient(DifyClient):
return self._send_request("POST", url, json=data, **kwargs)
def query_segments(
self, document_id, keyword: str = None, status: str = None, **kwargs
self,
document_id,
keyword: str | None = None,
status: str | None = None,
**kwargs,
):
"""
Query segments in this document.