12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970 |
- --[[
- Copyright (c) 2022, Vsevolod Stakhov <vsevolod@rspamd.com>
-
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
-
- http://www.apache.org/licenses/LICENSE-2.0
-
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- ]]--
-
- local neural_common = require "plugins/neural"
- local ts = require("tableshape").types
- local ucl = require "ucl"
-
- local E = {}
-
- -- Controller neural plugin
-
- local learn_request_schema = ts.shape{
- ham_vec = ts.array_of(ts.array_of(ts.number)),
- rule = ts.string:is_optional(),
- spam_vec = ts.array_of(ts.array_of(ts.number)),
- }
-
- local function handle_learn(task, conn)
- local parser = ucl.parser()
- local ok, err = parser:parse_text(task:get_rawbody())
- if not ok then
- conn:send_error(400, err)
- return
- end
- local req_params = parser:get_object()
-
- ok, err = learn_request_schema:transform(req_params)
- if not ok then
- conn:send_error(400, err)
- return
- end
-
- local rule_name = req_params.rule or 'default'
- local rule = neural_common.settings.rules[rule_name]
- local set = neural_common.get_rule_settings(task, rule)
- local version = ((set.ann or E).version or 0) + 1
-
- neural_common.spawn_train{
- ev_base = task:get_ev_base(),
- ann_key = neural_common.new_ann_key(rule, set, version),
- set = set,
- rule = rule,
- ham_vec = req_params.ham_vec,
- spam_vec = req_params.spam_vec,
- worker = task:get_worker(),
- }
-
- conn:send_string('{"success" : true}')
- end
-
- return {
- learn = {
- handler = handle_learn,
- enable = true,
- need_task = true,
- },
- }
|