add workflowClient ,fix rename bug (#7352)

This commit is contained in:
Krasus.Chen
2024-08-16 20:21:08 +08:00
committed by GitHub
parent 5a729a69cd
commit 7d4a0a417a
4 changed files with 239 additions and 67 deletions

View File

@@ -16,6 +16,7 @@ class DifyClient:
response = requests.request(method, url, json=json, params=params, headers=headers, stream=stream)
return response
def _send_request_with_files(self, method, endpoint, data, files):
headers = {
@@ -26,24 +27,24 @@ class DifyClient:
response = requests.request(method, url, data=data, headers=headers, files=files)
return response
def message_feedback(self, message_id, rating, user):
data = {
"rating": rating,
"user": user
}
return self._send_request("POST", f"/messages/{message_id}/feedbacks", data)
def get_application_parameters(self, user):
params = {"user": user}
return self._send_request("GET", "/parameters", params=params)
def file_upload(self, user, files):
data = {
"user": user
}
return self._send_request_with_files("POST", "/files/upload", data=data, files=files)
def text_to_audio(self, text:str, user:str, streaming:bool=False):
data = {
"text": text,
@@ -83,13 +84,19 @@ class ChatClient(DifyClient):
return self._send_request("POST", "/chat-messages", data,
stream=True if response_mode == "streaming" else False)
def get_suggested(self, message_id, user:str):
params = {"user": user}
return self._send_request("GET", f"/messages/{message_id}/suggested", params=params)
def stop_message(self, task_id, user):
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}
return self._send_request("GET", "/conversations", params=params)
def get_conversation_messages(self, user, conversation_id=None, first_id=None, limit=None):
params = {"user": user}
@@ -101,14 +108,14 @@ class ChatClient(DifyClient):
params["limit"] = limit
return self._send_request("GET", "/messages", params=params)
def get_conversations(self, user, last_id=None, limit=None, pinned=None):
params = {"user": user, "last_id": last_id, "limit": limit, "pinned": pinned}
return self._send_request("GET", "/conversations", params=params)
def rename_conversation(self, conversation_id, name, user):
data = {"name": name, "user": user}
def rename_conversation(self, conversation_id, name,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):
data = {"user": user}
return self._send_request("DELETE", f"/conversations/{conversation_id}", data)
def audio_to_text(self, audio_file, user):
data = {"user": user}
@@ -116,7 +123,11 @@ class ChatClient(DifyClient):
return self._send_request_with_files("POST", "/audio-to-text", data, files)
def get_suggested(self, message_id, user:str):
params = {"user": user}
return self._send_request("GET", f"/messages/{message_id}/suggested", params=params)
class WorkflowClient(DifyClient):
def run(self, inputs:dict, response_mode:str="streaming", user:str="abc-123"):
data = {"inputs": inputs, "response_mode": response_mode, "user": user}
return self._send_request("POST", "/workflows/run", data)
def stop(self, task_id, user):
data = {"user": user}
return self._send_request("POST", f"/workflows/tasks/{task_id}/stop", data)