瀏覽代碼

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

Co-authored-by: EchterTimo <EchterTimo@users.noreply.github.com>
tags/2.0.0-beta.2^2
Timo 1 月之前
父節點
當前提交
05cd7e2d8a
No account linked to committer's email address
共有 1 個文件被更改,包括 25 次插入14 次删除
  1. 25
    14
      sdks/python-client/dify_client/client.py

+ 25
- 14
sdks/python-client/dify_client/client.py 查看文件

@@ -73,12 +73,12 @@ class CompletionClient(DifyClient):
class ChatClient(DifyClient):
def create_chat_message(
self,
inputs,
query,
user,
response_mode="blocking",
conversation_id=None,
files=None,
inputs: dict,
query: str,
user: str,
response_mode: str = "blocking",
conversation_id: str | None = None,
files: dict | None = None,
):
data = {
"inputs": inputs,
@@ -97,22 +97,33 @@ class ChatClient(DifyClient):
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}
return self._send_request(
"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}
return self._send_request("POST", f"/chat-messages/{task_id}/stop", data)

def get_conversations(self, user, last_id=None, limit=None, pinned=None):
params = {"user": user, "last_id": last_id, "limit": limit, "pinned": pinned}
def get_conversations(
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)

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}

@@ -126,18 +137,18 @@ class ChatClient(DifyClient):
return self._send_request("GET", "/messages", params=params)

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}
return self._send_request(
"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}
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}
files = {"audio_file": audio_file}
return self._send_request_with_files("POST", "/audio-to-text", data, files)

Loading…
取消
儲存