gemini 3 pro dify workflow-engine test

dify-workflow-engine/models/__init__.py

@@ -0,0 +1,4 @@
+from .workflow import Workflow
+from .enums import UserFrom
+from .dataset import Dataset, Document, DocumentSegment, DatasetMetadata, RateLimitLog
+from .tools import ToolFile

dify-workflow-engine/models/dataset.py

@@ -0,0 +1,14 @@
+class Document:
+    doc_metadata = {}
+
+class Dataset:
+    pass
+
+class DocumentSegment:
+    pass
+
+class DatasetMetadata:
+    pass
+
+class RateLimitLog:
+    pass

dify-workflow-engine/models/enums.py

@@ -0,0 +1,5 @@
+from enum import Enum
+
+class UserFrom(Enum):
+    ACCOUNT = "account"
+    END_USER = "end_user"

dify-workflow-engine/models/model.py

@@ -0,0 +1,14 @@
+class App:
+    pass
+
+class EndUser:
+    pass
+
+class Account:
+    pass
+
+class UploadFile:
+    pass
+
+class Conversation:
+    pass

dify-workflow-engine/models/tools.py

@@ -0,0 +1,4 @@
+from pydantic import BaseModel
+
+class ToolFile(BaseModel):
+    pass

dify-workflow-engine/models/workflow.py

@@ -0,0 +1,12 @@
+class Workflow:
+    def __init__(self, tenant_id, app_id, id, graph_dict):
+        self.tenant_id = tenant_id
+        self.app_id = app_id
+        self.id = id
+        self.graph_dict = graph_dict
+
+    def get_node_config_by_id(self, node_id):
+        for node in self.graph_dict.get("nodes", []):
+            if node["id"] == node_id:
+                return node
+        return {}