|
@@ -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)
|