Explorar el Código

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 hace 1 mes
padre
commit
05cd7e2d8a
No account linked to committer's email address
Se han modificado 1 ficheros con 25 adiciones y 14 borrados
  1. 25
    14
      sdks/python-client/dify_client/client.py

+ 25
- 14
sdks/python-client/dify_client/client.py Ver fichero

class ChatClient(DifyClient): class ChatClient(DifyClient):
def create_chat_message( def create_chat_message(
self, 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 = { data = {
"inputs": inputs, "inputs": inputs,
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):
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) 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}


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)

Cargando…
Cancelar
Guardar