Fabio
c36510c21e
Some checks failed
Linux Build / Linux cpu (push) Has been cancelled
Linux Build / Linux cpu-clang (push) Has been cancelled
Linux Build / Linux vulkan (push) Has been cancelled
mac-build / macOS CPU (push) Has been cancelled
Nix Build / Nix (cpu) (push) Has been cancelled
Nix Build / Nix (vulkan) (push) Has been cancelled
windows-build / Windows CPU (push) Has been cancelled
37 lines
1.4 KiB
C++
37 lines
1.4 KiB
C++
#pragma once
|
|
|
|
#include "engine/framework/io/json.h"
|
|
#include "engine/framework/runtime/errors.h"
|
|
|
|
#include <string>
|
|
|
|
namespace minitts::server {
|
|
|
|
// /v1/tasks/stream answers with one JSON body once the run is over, unless the
|
|
// request's top-level "stream_format" is "sse": then each event goes out as a
|
|
// server-sent event as the model produces it. Absent or null keeps the JSON
|
|
// body. Any other value is turned away, so that a misspelt one is not answered
|
|
// with a body the client did not ask for.
|
|
inline bool task_stream_sse_requested(const engine::io::json::Value & body) {
|
|
const auto * value = body.find("stream_format");
|
|
if (value == nullptr || value->is_null()) {
|
|
return false;
|
|
}
|
|
if (value->is_string() && value->as_string() == "sse") {
|
|
return true;
|
|
}
|
|
throw engine::runtime::InvalidRequestError("task stream stream_format must be sse");
|
|
}
|
|
|
|
// The data of the SSE lines, the same in both runtimes: each event as the
|
|
// object the JSON response puts in "events", then the object it puts under
|
|
// "result".
|
|
inline std::string task_stream_event_line(const std::string & event_json) {
|
|
return "{\"type\":\"task.stream.event\",\"event\":" + event_json + "}";
|
|
}
|
|
|
|
inline std::string task_stream_done_line(const std::string & result_json) {
|
|
return "{\"type\":\"task.stream.done\",\"result\":" + result_json + "}";
|
|
}
|
|
|
|
} // namespace minitts::server
|