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

add type annotations for Python SDK ChatClient Class (#24018)

Browse Source 
Co-authored-by: EchterTimo <EchterTimo@users.noreply.github.com>
This commit is contained in:
Timo
2025-09-05 12:12:46 +02:00
committed by GitHub
parent a9da8edbde
commit 05cd7e2d8a
1 changed files with 25 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -73,12 +73,12 @@ class CompletionClient(DifyClient):
class ChatClient(DifyClient): class ChatClient(DifyClient):
def create_chat_message( def create_chat_message(
self, self,
inputs, inputs: dict,
query, query: str,
user, user: str,
response_mode="blocking", response_mode: str = "blocking",
conversation_id=None, conversation_id: str | None = None,
files=None, files: dict | None = None,
): ):
data = { data = {
"inputs": inputs, "inputs": inputs,
@@ -97,22 +97,33 @@ class ChatClient(DifyClient):
stream=True if response_mode == "streaming" else False, stream=True if response_mode == "streaming" else False,
) )
def get_suggested(self, message_id, user: str): def get_suggested(self, message_id: str, user: str):
params = {"user": user} params = {"user": user}
return self._send_request( return self._send_request(
"GET", f"/messages/{message_id}/suggested", params=params "GET", f"/messages/{message_id}/suggested", params=params
) )
def stop_message(self, task_id, user): def stop_message(self, task_id: str, user: str):
data = {"user": user} data = {"user": user}
return self._send_request("POST", f"/chat-messages/{task_id}/stop", data) return self._send_request("POST", f"/chat-messages/{task_id}/stop", data)
def get_conversations(self, user, last_id=None, limit=None, pinned=None): def get_conversations(
params = {"user": user, "last_id": last_id, "limit": limit, "pinned": pinned} self,
user: str,
last_id: str | None = None,
limit: int | None = None,
pinned: bool | None = None
):
params = {"user": user, "last_id": last_id,
"limit": limit, "pinned": pinned}
return self._send_request("GET", "/conversations", params=params) return self._send_request("GET", "/conversations", params=params)
def get_conversation_messages( def get_conversation_messages(
self, user, conversation_id=None, first_id=None, limit=None self,
user: str,
conversation_id: str | None = None,
first_id: str | None = None,
limit: int | None = None
): ):
params = {"user": user} params = {"user": user}
@@ -126,18 +137,18 @@ class ChatClient(DifyClient):
return self._send_request("GET", "/messages", params=params) return self._send_request("GET", "/messages", params=params)
def rename_conversation( def rename_conversation(
self, conversation_id, name, auto_generate: bool, user: str self, conversation_id: str, name: str, auto_generate: bool, user: str
): ):
data = {"name": name, "auto_generate": auto_generate, "user": user} data = {"name": name, "auto_generate": auto_generate, "user": user}
return self._send_request( return self._send_request(
"POST", f"/conversations/{conversation_id}/name", data "POST", f"/conversations/{conversation_id}/name", data
) )
def delete_conversation(self, conversation_id, user): def delete_conversation(self, conversation_id: str, user: str):
data = {"user": user} data = {"user": user}
return self._send_request("DELETE", f"/conversations/{conversation_id}", data) return self._send_request("DELETE", f"/conversations/{conversation_id}", data)
def audio_to_text(self, audio_file, user): def audio_to_text(self, audio_file: dict, user: str):
data = {"user": user} data = {"user": user}
files = {"audio_file": audio_file} files = {"audio_file": audio_file}
return self._send_request_with_files("POST", "/audio-to-text", data, files) return self._send_request_with_files("POST", "/audio-to-text", data, files)