农饮工程实施职责单位的法律依据主要来自于《农村饮水安全工程建设管理办法》及地方性相关法规，如《新疆维吾尔自治区农村饮水工程管理办法》。

一. 《农村饮水安全工程建设管理办法》中关于职责单位的规定

1. 行政首长负责制：根据该《办法》，农村饮水安全保障实行行政首长负责制，地方政府对农村饮水安全负总责，中央给予指导和资金支持。这意味着地方政府是农饮工程实施的主要责任单位。
2. 部门分工：各有关部门要在政府的统一领导下，各负其责，密切配合，共同做好农村饮水安全工作。这明确了政府各部门在农饮工程实施中的具体职责和协作要求。
3. 项目法人责任制：农村饮水安全工程建设实行项目法人责任制，对“千吨万人”以上的集中供水工程，要按有关规定组建项目建设管理单位，负责工程建设和建后运行管理。这规定了项目法人在农饮工程建设和管理中的核心地位。

二. 《新疆维吾尔自治区农村饮水工程管理办法》中关于职责单位的规定

1. 管理主体：该《办法》明确了农村饮水工程的管理主体，包括县级人民政府水行政主管部门农村饮水工程管理机构、受益范围内的村民委员会或者用水户协会，以及投资主体自主管理等。这根据农饮工程的投资主体和规模，详细划分了管理职责。
2. 部门职责：县级以上人民政府水行政主管部门负责本行政区域内农村饮水工程管理的指导和监督工作，其他有关部门如发展改革、财政、卫生、国土资源、环境保护等也按照各自的职责范围，负责农村饮水工程管理的相关工作。这进一步细化了政府各部门在农饮工程管理中的具体职责。

综上所述，农饮工程实施职责单位的法律依据主要来自于上述两个法规，它们共同构成了农饮工程实施和管理中职责单位划分的法律依据。如有进一步需要，您可以继续咨询我。