害怕Dify工作流被白嫖?教你一招加上登录验证功能

相信很多开发和使用Dify工作流的小伙伴都遇到过一个问题，就是害怕自己的工作流被滥用，尤其是部署在公网上的工作流，只要别人知道了地址，就都能使用。

今天教大家怎么给工作流加一个登录验证功能，这样就不怕自己的工作流再被滥用了。

这是一个可以在Dify工作流中加入账号密码验证的功能，主要是提供思路。工作流已经上传到网盘老地方，老粉应该知道如何获取，新粉可以私信我。

使用的时候要先输入账号信息，不正确就返回登录框，正确了就流转到我们正常的业务流程上，这样就可以给工作流加上权限验证功能了。甚至还可以根据账号查找权限，对接不同的工作流。

主要的验证逻辑在登录代码执行节点里。我这里为了说明思路，只是在代码里写死了账号，实际使用的时候，大家可以从数据库里查询出正确的账号密码。

import json

def main(input_string):
    try:
        data = json.loads(input_string)
        username = data['username']
        password = data['password']
        # 这里的实际逻辑可以改成从数据库或者验证接口来完成验证
        if username == "north1":
            return {"is_login":1,"user_token":"user_token_test"}
        else:
            return {"is_login":0,"user_token":""}
    except:
        return {"is_login":0,"user_token":""}

如果有需要的小伙伴可以私信我获取工作流，觉得有用就点赞关注一下吧。