如何在Dify中通过自定义Python代码处理高维矩阵数据-人工智能-PHP中文网

dify中处理高维矩阵需自建安全svd函数：仅限自托管部署，通过修改后端代码注册safe_matrix_svd工具，并在工作流中调用验证结果。

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜

如何在dify中通过自定义python代码处理高维矩阵数据

在Dify中直接运行自定义Python代码处理高维矩阵（如1000×1000以上NumPy数组）时，需绕过默认沙箱限制并确保内存与计算安全——Dify原生不支持直接import numpy或执行长时计算，必须通过服务端函数注册+API代理方式实现。

确认Dify部署模式与后端扩展能力

进入Dify管理后台 → 设置 → 系统设置 → 查看「部署类型」：仅当显示为「Self-hosted（Docker Compose或K8s）」时才可进行后续操作；SaaS版（dify.ai）完全禁止自定义Python执行环境。

打开服务器终端，执行 docker ps | grep web，确认容器名含 【web】 且状态为 Up；若无输出，说明未完成本地部署或服务未启动。

在后端服务中注册安全矩阵处理函数

进入Dify项目根目录（如 /opt/dify），定位到 api/core/llm/provider/custom 目录。

新建文件 matrix_handler.py，写入以下内容：

import numpy as np
from typing import List, Dict, Any
def safe_matrix_svd(data: List[List[float]], max_dim: int = 500) -> Dict[str, Any]:
  if len(data) > max_dim or len(data[0]) > max_dim:
    raise ValueError(f"Matrix exceeds max allowed dimension {max_dim}")
  arr = np.array(data, dtype=np.float64)
  U, s, Vt = np.linalg.svd(arr, full_matrices=False)
  return {"U": U.tolist(), "s": s.tolist(), "Vt": Vt.tolist()}

这一步必须做：修改 api/core/llm/provider/custom/__init__.py，加入 from .matrix_handler import safe_matrix_svd 并在 __all__ 列表中添加 【'safe_matrix_svd'】，否则函数无法被LLM调用识别。

python-script-generator

快速生成专业的 Python 脚本和应用代码。一键创建完整项目结构，支持CLI、API、爬虫、Bot、Django等多种项目类型，包含完整的项目结构、配置文件、依赖管理、测试、README和文档。

下载

配置自定义工具（Tool）供工作流调用

登录Dify Web UI → 左侧导航栏点击「开发」→「工具」→「创建工具」。

填写基础信息：
名称：SVD矩阵分解
描述：对二维浮点数矩阵执行截断SVD，返回U、s、Vt三部分（自动限制最大维度500）

在「API Schema」区域粘贴以下OpenAPI 3.0格式定义：

{
"type": "object",
"properties": {
  "data": {
   "type": "array",
   "items": { "type": "array", "items": { "type": "number" } }
  },
  "max_dim": { "type": "integer", "default": 500 }
},
"required": ["data"]
}

「API URL」填 http://web:5001/api/v1/tool/custom/safe_matrix_svd（注意：不是localhost，必须用Docker内部服务名 【web】）。

在应用工作流中调用矩阵处理工具

进入「应用」→「编辑工作流」→ 点击画布空白处 → 「添加节点」→ 选择「工具调用」→ 搜索并选中「SVD矩阵分解」。

配置输入参数：
① 点击「data」字段右侧「{…}」图标 → 选择「从变量中提取」→ 输入变量名 input_matrix；
② 确保上游节点已将合法二维列表（如 [[1.0,2.0],[3.0,4.0]]）赋值给该变量；
③ 若传入非数字、空数组或单行/单列超限矩阵，工具将立即返回HTTP 400错误，不会触发计算。

这一步不能跳过：在工具节点下方接一个「代码解释器」节点，用Python检查返回结果结构，例如 len($tool_result.U) == len($tool_result.s)，避免下游误用截断后的U/Vt维度。