完善整体

Author: jayscoder

examples/utils.py

@@ -344,7 +344,7 @@ def run_example(example_fn: Callable, *args, **kwargs):
             print(f"{example_fn.__doc__.strip()}\n")
 
         return example_fn(*args, **kwargs)
-
+    
     except Exception as e:
         print(f"\n示例运行过程中发生错误: {e}")
         import traceback

pydify/agent.py

@@ -72,120 +72,30 @@ def send_message(
         endpoint = "chat-messages"
 
         return self.post_stream(endpoint, json_data=payload, **kwargs)  # 传递额外参数
-
+    
     def stop_response(self, task_id: str, user: str) -> Dict[str, Any]:
         """
-        停止正在进行的响应，仅支持流式模式。
+        停止正在进行的响应流。此方法仅在流式模式下有效。
 
         Args:
-            task_id (str): 任务ID，可在流式返回Chunk中获取
-            user (str): 用户标识，必须和发送消息接口传入user保持一致
+            task_id (str): 任务唯一标识,可从流式响应的数据块中获取
+            user (str): 用户唯一标识,需要与发送消息时的user参数保持一致
 
         Returns:
-            Dict[str, Any]: 停止响应的结果
+            Dict[str, Any]: 停止响应的结果,格式如下:
+            {
+                "result": "success"  # 表示成功停止响应
+            }
 
         Raises:
-            requests.HTTPError: 当API请求失败时
+            requests.HTTPError: API请求失败时抛出此异常,包含具体的错误信息
+            DifyAPIError: Dify服务端返回错误时抛出此异常
         """
         endpoint = f"chat-messages/{task_id}/stop"
         payload = {"user": user}
         return self.post(endpoint, json_data=payload)
 
-    def message_feedback(
-        self,
-        message_id: str,
-        user: str,
-        rating: str = None,
-        content: str = None,
-    ) -> Dict[str, Any]:
-        """
-        对消息进行反馈（点赞/点踩）。
-
-        Args:
-            message_id (str): 消息ID
-            user (str): 用户标识
-            rating (str, optional): 评价，可选值：'like'(点赞), 'dislike'(点踩), None(撤销)。默认为None
-            content (str, optional): 反馈的具体信息。默认为None
-
-        Returns:
-            Dict[str, Any]: 反馈结果
-
-        Raises:
-            ValueError: 当提供了无效的参数时
-            requests.HTTPError: 当API请求失败时
-        """
-        if rating and rating not in ["like", "dislike", None]:
-            raise ValueError("rating must be 'like', 'dislike' or None")
-
-        endpoint = f"messages/{message_id}/feedbacks"
-
-        payload = {"user": user}
-
-        if rating is not None:
-            payload["rating"] = rating
-
-        if content:
-            payload["content"] = content
-
-        return self.post(endpoint, json_data=payload)
-
-    def get_suggested_questions(
-        self, message_id: str, user: str, **kwargs
-    ) -> Dict[str, Any]:
-        """
-        获取下一轮建议问题列表。
-
-        Args:
-            message_id (str): 消息ID
-            user (str): 用户标识
-            **kwargs: 额外的请求参数，如timeout、max_retries等
-
-        Returns:
-            Dict[str, Any]: 建议问题列表
-
-        Raises:
-            DifyAPIError: 当API请求失败时
-        """
-        # 尝试多种可能的端点路径格式
-        possible_endpoints = [
-            f"messages/{message_id}/suggested",  # 原始格式
-            f"messages/{message_id}/suggested-questions",  # 新格式1
-            f"chat-messages/{message_id}/suggested-questions",  # 新格式2
-            "suggested-questions",  # 当前格式
-        ]
-
-        params = {
-            "user": user,
-        }
-
-        # 添加详细日志
-        # print(f"请求推荐问题: 消息ID={message_id}, 用户={user}")
-
-        # 尝试所有可能的端点，直到找到一个有效的
-        last_error = None
-        for endpoint in possible_endpoints:
-            try:
-                params_to_use = params.copy()
-                # 如果端点是standalone的suggested-questions，需要添加message_id参数
-                if endpoint == "suggested-questions":
-                    params_to_use["message_id"] = message_id
-                else:
-                    # 否则可能不需要在参数中包含message_id
-                    params_to_use.pop("message_id", None)
-
-                print(f"尝试端点: {endpoint}, 参数: {params_to_use}")
-                result = self.get(endpoint, params=params_to_use, **kwargs)
-                print(f"端点 {endpoint} 请求成功!")
-                return result
-            except Exception as e:
-                last_error = e
-                print(f"端点 {endpoint} 请求失败: {str(e)}")
-                continue
-
-        # 如果所有端点都失败，记录最后一个错误并返回空结果
-        print(f"所有推荐问题端点请求都失败。最后错误: {str(last_error)}")
-        return {"data": []}
-
+    
 
     def get_meta(self) -> Dict[str, Any]:
         """

pydify/chatbot.py

@@ -9,7 +9,7 @@
 import os
 from typing import Any, Dict, Generator, List, Optional, Union
 
-from .common import DifyBaseClient
+from .common import DifyBaseClient, DifyType
 
 
 class ChatbotClient(DifyBaseClient):
@@ -18,7 +18,7 @@ class ChatbotClient(DifyBaseClient):
     提供与Dify Chatbot应用API交互的方法，包括发送消息、获取历史消息、管理会话、
     上传文件、语音转文字、文字转语音等功能。
     """
-    
+    type = DifyType.Chatbot
     def send_message(
         self,
         query: str,
@@ -81,7 +81,7 @@ def send_message(
             return self.post_stream(endpoint, json_data=payload, **kwargs)
         else:
             return self.post(endpoint, json_data=payload, **kwargs)
-
+    
     def stop_response(self, task_id: str, user: str) -> Dict[str, Any]:
         """
         停止正在进行的响应，仅支持流式模式。

pydify/chatflow.py

@@ -97,61 +97,27 @@ def stop_response(self, task_id: str, user: str) -> Dict[str, Any]:
         payload = {"user": user}
         return self.post(endpoint, json_data=payload)
 
-    def message_feedback(
-        self,
-        message_id: str,
-        user: str,
-        rating: str = None,
-        content: str = None,
-    ) -> Dict[str, Any]:
-        """
-        对消息进行反馈（点赞/点踩）。
-
-        Args:
-            message_id (str): 消息ID
-            user (str): 用户标识
-            rating (str, optional): 评价，可选值：'like'(点赞), 'dislike'(点踩), None(撤销)。默认为None
-            content (str, optional): 反馈的具体信息。默认为None
-
-        Returns:
-            Dict[str, Any]: 反馈结果
-
-        Raises:
-            ValueError: 当提供了无效的参数时
-            requests.HTTPError: 当API请求失败时
-        """
-        if rating and rating not in ["like", "dislike", None]:
-            raise ValueError("rating must be 'like', 'dislike' or None")
-
-        endpoint = f"messages/{message_id}/feedbacks"
-
-        payload = {"user": user}
-
-        if rating is not None:
-            payload["rating"] = rating
-
-        if content:
-            payload["content"] = content
-
-        return self.post(endpoint, json_data=payload)
-
-    def get_suggested_questions(self, message_id: str, user: str) -> Dict[str, Any]:
+    def get_suggested_questions(self, message_id: str, user: str, **kwargs) -> Dict[str, Any]:
         """
         获取下一轮建议问题列表。
 
         Args:
             message_id (str): 消息ID
             user (str): 用户标识
+            **kwargs: 额外的请求参数，如timeout、max_retries等
 
         Returns:
             Dict[str, Any]: 建议问题列表
 
         Raises:
-            requests.HTTPError: 当API请求失败时
+            DifyAPIError: 当API请求失败时
         """
-        endpoint = f"messages/{message_id}/suggested"
-        params = {"user": user}
-        return self.get(endpoint, params=params)
+        params = {
+            "user": user,
+            "message_id": message_id,
+        }
+
+        return self.get("suggested-questions", params=params, **kwargs)
 
     def get_messages(
         self,
@@ -402,95 +368,6 @@ def text_to_audio(
 
         return self.post(endpoint, json_data=payload)
 
-    def upload_file(self, file_path: str, user: str) -> Dict[str, Any]:
-        """
-        上传文件到Dify API，可用于图文多模态理解。支持多种格式的文件。
-
-        Args:
-            file_path (str): 要上传的文件路径
-            user (str): 用户标识
-
-        Returns:
-            Dict[str, Any]: 上传文件的响应数据
-
-        Raises:
-            FileNotFoundError: 当文件不存在时
-            requests.HTTPError: 当API请求失败时
-        """
-        if not os.path.exists(file_path):
-            raise FileNotFoundError(f"File not found: {file_path}")
-
-        # 获取MIME类型
-        mime_type, _ = mimetypes.guess_type(file_path)
-        if not mime_type:
-            mime_type = "application/octet-stream"
-
-        with open(file_path, "rb") as file:
-            files = {"file": file}
-            data = {"user": user}
-
-            headers = self._get_headers()
-            # 移除Content-Type，让requests自动设置multipart/form-data
-            headers.pop("Content-Type", None)
-
-            response = self._request(
-                "POST", "files/upload", headers=headers, files=files, data=data
-            )
-            return response.json()
-
-    def upload_file_obj(
-        self, file_obj: BinaryIO, filename: str, user: str
-    ) -> Dict[str, Any]:
-        """
-        使用文件对象上传文件到Dify API。
-
-        Args:
-            file_obj (BinaryIO): 文件对象
-            filename (str): 文件名
-            user (str): 用户标识
-
-        Returns:
-            Dict[str, Any]: 上传文件的响应数据
-
-        Raises:
-            requests.HTTPError: 当API请求失败时
-        """
-        files = {"file": (filename, file_obj)}
-        data = {"user": user}
-
-        headers = self._get_headers()
-        # 移除Content-Type，让requests自动设置multipart/form-data
-        headers.pop("Content-Type", None)
-
-        response = self._request(
-            "POST", "files/upload", headers=headers, files=files, data=data
-        )
-        return response.json()
-
-    def get_app_info(self) -> Dict[str, Any]:
-        """
-        获取应用基本信息。
-
-        Returns:
-            Dict[str, Any]: 应用信息，包含名称、描述和标签
-
-        Raises:
-            requests.HTTPError: 当API请求失败时
-        """
-        return self.get("info")
-
-    def get_parameters(self) -> Dict[str, Any]:
-        """
-        获取应用参数，包括功能开关、输入参数名称、类型及默认值等。
-
-        Returns:
-            Dict[str, Any]: 应用参数配置
-
-        Raises:
-            requests.HTTPError: 当API请求失败时
-        """
-        return self.get("parameters")
-
     def get_meta(self) -> Dict[str, Any]:
         """
         获取应用Meta信息，用于获取工具icon等。