build and test

Author: jayscoder

build_and_test.sh

@@ -24,6 +24,10 @@ echo "=== 检查包 ==="
 pip install twine
 twine check dist/*
 
+git add -A
+git commit -m "build and test"
+git push
+
 echo "=== 完成! ==="
 echo "如果要发布到PyPI，请运行:"
 echo "twine upload dist/*"

examples/test_examples.py

@@ -28,7 +28,7 @@ def create_client(type: str, base_url: str, api_key: str) -> DifyBaseClient:
         raise ValueError(f"Invalid client type: {type}")
 
 
-__version__ = "2.0.0"
+__version__ = "2.1.0"
 __all__ = [
     "WorkflowClient",
     "ChatbotClient",

pydify/README.md

@@ -10,8 +10,10 @@
 import os
 from typing import Any, BinaryIO, Dict, Generator, List, Optional, Tuple, Union
 
-from .common import DifyBaseClient, DifyType
 from .chatbot import ChatbotClient
+from .common import DifyBaseClient, DifyType
+
+
 class AgentEvent:
     """事件类型枚举
 
@@ -28,7 +30,7 @@ class AgentEvent:
     MESSAGE_REPLACE = "message_replace"  # 消息内容替换事件，用于内容审查后的替换
     ERROR = "error"  # 流式输出过程中出现的异常事件
     PING = "ping"  # 保持连接存活的ping事件，每10秒一次
-    
+
 
 class AgentClient(ChatbotClient):
     """Dify Agent应用客户端类。
@@ -38,11 +40,11 @@ class AgentClient(ChatbotClient):
     """
 
     type = DifyType.Agent
-    
+
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.task_id = None
-        
+
     def send_message(
         self,
         query: str,
@@ -53,7 +55,7 @@ def send_message(
         files: List[Dict[str, Any]] = None,
         auto_generate_name: bool = True,
         **kwargs,  # 添加kwargs参数，用于接收额外的请求参数
-    ) -> Generator[Dict[str, Any], None, None] | Dict[str, Any]:
+    ) -> Union[Generator[Dict[str, Any], None, None], Dict[str, Any]]:
         """
         发送对话消息，创建会话消息。在Agent模式下，只支持streaming流式模式。
 
@@ -76,7 +78,7 @@ def send_message(
         """
         if response_mode != "streaming":
             raise ValueError("Agent mode only supports streaming response mode")
-        
+
         payload = {
             "query": query,
             "user": user,
@@ -116,7 +118,7 @@ def stop_task(self, task_id: str, user: str) -> Dict[str, Any]:
         endpoint = f"chat-messages/{task_id}/stop"
         payload = {"user": user}
         return self.post(endpoint, json_data=payload)
-    
+
     def get_meta(self) -> Dict[str, Any]:
         """
         获取应用Meta信息，用于获取工具icon等。
@@ -128,4 +130,3 @@ def get_meta(self) -> Dict[str, Any]:
             requests.HTTPError: 当API请求失败时
         """
         return self.get("meta")
-    

pydify/__init__.py

@@ -11,6 +11,7 @@
 
 from .common import DifyBaseClient, DifyType
 
+
 class ChatbotEvent:
     """事件类型枚举
 
@@ -98,7 +99,7 @@ class ChatbotEvent:
             "event": "ping",
         }
     """
-    
+
     MESSAGE = "message"  # LLM返回文本块事件
     AGENT_MESSAGE = "agent_message"  # Agent模式下返回文本块事件
     AGENT_THOUGHT = "agent_thought"  # Agent思考步骤相关内容
@@ -109,7 +110,6 @@ class ChatbotEvent:
     MESSAGE_REPLACE = "message_replace"  # 消息内容替换事件
     ERROR = "error"  # 异常事件
     PING = "ping"  # 保持连接存活的ping事件
-    
 
 
 class ChatbotClient(DifyBaseClient):
@@ -172,18 +172,18 @@ def send_message(
 
         if files:
             payload["files"] = files
-        
+
         endpoint = "chat-messages"
-        
+
         # 打印请求信息，便于调试
         print(f"请求URL: {self.base_url}{endpoint}")
         print(f"请求参数: {json.dumps(payload)}")
-        
+
         if response_mode == "streaming":
             return self.post_stream(endpoint, json_data=payload, **kwargs)
         else:
             return self.post(endpoint, json_data=payload, **kwargs)
-    
+
     def stop_task(self, task_id: str, user: str) -> Dict[str, Any]:
         """
         停止正在进行的响应，仅支持流式模式。
@@ -282,11 +282,13 @@ def audio_to_text_obj(self, file_obj, filename: str, user: str) -> Dict[str, Any
             "POST", endpoint, headers=headers, files=files, data=data
         )
         return response.json()
-    
-    def get_messages(self, conversation_id: str, user: str, first_id: str = None, limit: int = 20) -> Dict[str, Any]:
+
+    def get_messages(
+        self, conversation_id: str, user: str, first_id: str = None, limit: int = 20
+    ) -> Dict[str, Any]:
         """
         获取会话消息列表，支持分页加载历史聊天记录。
-        
+
         Args:
             conversation_id (str): 会话ID，用于标识特定的对话会话
             user (str): 用户标识，由开发者定义规则，需保证用户标识在应用内唯一
@@ -335,7 +337,7 @@ def get_messages(self, conversation_id: str, user: str, first_id: str = None, li
             DifyAPIError: 当API请求失败时
         """
         endpoint = "messages"
-        
+
         params = {
             "conversation_id": conversation_id,
             "user": user,
@@ -345,4 +347,4 @@ def get_messages(self, conversation_id: str, user: str, first_id: str = None, li
         if first_id:
             params["first_id"] = first_id
 
-        return self.get(endpoint, params=params)
+        return self.get(endpoint, params=params)

pydify/agent.py

@@ -9,8 +9,10 @@
 import mimetypes
 import os
 from typing import Any, BinaryIO, Dict, Generator, List, Optional, Tuple, Union
-from .common import DifyBaseClient, DifyType
+
 from .chatbot import ChatbotClient
+from .common import DifyBaseClient, DifyType
+
 
 class ChatflowEvent:
     """事件类型枚举
@@ -30,7 +32,6 @@ class ChatflowEvent:
     WORKFLOW_FINISHED = "workflow_finished"  # workflow执行结束事件（包含成功/失败状态）
     ERROR = "error"  # 流式输出过程中出现的异常事件
     PING = "ping"  # 保持连接存活的ping事件，每10秒一次
-    
 
 
 class ChatflowClient(ChatbotClient):
@@ -118,7 +119,7 @@ def stop_task(self, task_id: str, user: str) -> Dict[str, Any]:
         endpoint = f"chat-messages/{task_id}/stop"
         payload = {"user": user}
         return self.post(endpoint, json_data=payload)
-    
+
     def get_suggested_questions(
         self, message_id: str, user: str, **kwargs
     ) -> Dict[str, Any]:
@@ -132,7 +133,7 @@ def get_suggested_questions(
 
         Returns:
             Dict[str, Any]: 建议问题列表
-        
+
         Raises:
             DifyAPIError: 当API请求失败时
         """
@@ -419,4 +420,3 @@ def get_meta(self) -> Dict[str, Any]:
             requests.HTTPError: 当API请求失败时
         """
         return self.get("meta")
-