107 lines
5.5 KiB
C++
107 lines
5.5 KiB
C++
#include "../src/cli/cli.h"
|
|
#include "../src/shell/sandbox.h"
|
|
#include "../src/shell/session.h"
|
|
#include "../src/shell/pty.h"
|
|
#include "../src/shell/spinner.h"
|
|
#include "../src/shell/livestream.h"
|
|
#include "../src/shell/lanes.h"
|
|
#include "../src/llm/llm.h"
|
|
#include "../src/util/ansi.h"
|
|
#include "../src/util/text.h"
|
|
#include "../src/config/config.h"
|
|
#include "../src/config/consts.h"
|
|
#include <cassert>
|
|
#include <filesystem>
|
|
#include <cstdio>
|
|
using detz::llm::ChatClient;
|
|
using detz::llm::ChatRequest;
|
|
int main() {
|
|
detz::cli::Parser p;
|
|
const char* a0 = "detz";
|
|
auto c = p.parse(1, (char**)&a0);
|
|
assert(c.name == "--help");
|
|
assert(c.run() == 0);
|
|
// shell
|
|
detz::shell::Sandbox sb("/tmp", {"echo"});
|
|
auto r = sb.run("echo hi");
|
|
assert(r.code == 0 && r.out.find("hi") != std::string::npos);
|
|
auto denied = sb.run("rm -rf /");
|
|
assert(denied.code == 126);
|
|
assert(!detz::shell::curl_hint("http://x").empty());
|
|
detz::shell::ShellSession sh;
|
|
auto e = sh.exec("echo ok");
|
|
assert(e.out.find("ok") != std::string::npos);
|
|
assert(sh.target_paths("a:b").size() == 2);
|
|
assert(sh.segment_targets({"a"}).size() == 1);
|
|
detz::shell::PtyProc pty;
|
|
assert(pty.spawn("echo pty") && pty.read_all().find("pty") != std::string::npos);
|
|
detz::shell::Spinner sp("x"); sp.tick(); sp.done(); assert(sp.ticks() == 2);
|
|
detz::shell::LiveStream ls; ls.push("a"); ls.push("b");
|
|
assert(ls.joined() == "ab");
|
|
int n = 0; ls.drain([&](const std::string&) { ++n; return true; }); assert(n == 2);
|
|
detz::shell::LaneBoard lb; lb.set("a", "1"); lb.set("a", "2");
|
|
assert(lb.size() == 1 && lb.render().find("2") != std::string::npos);
|
|
assert(!detz::shell::Glow::highlight("t").empty());
|
|
assert(detz::shell::speak_text("hi") == "hi");
|
|
// llm mocked transport
|
|
auto mock = [](const std::string&, const std::string&, const std::map<std::string,std::string>&) {
|
|
return std::string("{\"content\":\"hello\"}");
|
|
};
|
|
ChatClient cc("openai", "m", mock);
|
|
auto cr = cc.chat(ChatRequest{"m", {{"user", "hi"}}, false});
|
|
assert(cr.text == "hello");
|
|
auto smock = [](const std::string&, const std::string&, const std::map<std::string,std::string>&) {
|
|
return std::string("data: {\"content\":\"a\"}\n\ndata: [DONE]\n\n");
|
|
};
|
|
ChatClient cs("anthropic", "m", smock);
|
|
std::vector<std::string> got;
|
|
cs.chat_stream(ChatRequest{"m", {}, true}, [&](const std::string& d) { got.push_back(d); return true; });
|
|
assert(got.size() == 1);
|
|
bool threw = false;
|
|
try { ChatClient bad("openai", "m"); bad.chat(ChatRequest{}); } catch (const detz::llm::BackendError&) { threw = true; }
|
|
assert(threw);
|
|
detz::llm::ModelHealth h; h.record("a", false, 5); h.record("b", true, 9);
|
|
assert(h.size() == 2 && h.best()->model == "b");
|
|
assert(detz::llm::provider_names().find("openrouter") != std::string::npos);
|
|
assert(detz::llm::provider_info("xai").base_url.find("x.ai") != std::string::npos);
|
|
detz::llm::Opencode oc("http://localhost:9", mock);
|
|
assert(oc.complete("m", {}).text == "hello");
|
|
bool be = false;
|
|
try { detz::llm::Opencode no; no.complete("m", {}); } catch (const detz::llm::BackendError&) { be = true; }
|
|
assert(be);
|
|
// M2 util+config
|
|
assert(detz::Consts::kTotalStepCap == 500 && detz::Consts::kAuditMaxChars == 20000);
|
|
assert(std::string(detz::kDefaultModel) == "muse");
|
|
assert(!detz::ZenFreeModels().empty() && !detz::PollinationsModels().empty());
|
|
assert(detz::Paint("x", detz::Ansi::kRed).find("x") != std::string::npos);
|
|
assert(detz::Truncate("abc", 1500, 500) == "abc");
|
|
std::string big(3000, 'a');
|
|
assert(detz::Truncate(big, 100, 100).find("chars truncated") != std::string::npos);
|
|
assert(detz::TruncMarker(5).find("5") != std::string::npos);
|
|
assert(detz::CappedText("hi") == "hi");
|
|
assert(detz::ChunkLines("").empty());
|
|
assert(detz::ChunkLines("a\nb").size() == 1);
|
|
assert(detz::CollapseRepeatedLines("x\nx\nx").find("repeated 2") != std::string::npos);
|
|
assert(detz::CollapseRepeatedLines("a\nb").find("repeated") == std::string::npos);
|
|
{ auto [tags, rest] = detz::SplitLeadingTags("@a @b hello"); assert(tags.size() == 2 && rest == "hello"); }
|
|
{ auto [tags, rest] = detz::SplitLeadingTags("hi"); assert(tags.empty() && rest == "hi"); }
|
|
{ auto dir = std::filesystem::temp_directory_path().string();
|
|
std::string f = dir + "/detz_m2_test.txt"; FILE* fh = std::fopen(f.c_str(), "w"); std::fputs("x", fh); std::fclose(fh);
|
|
assert(detz::ExpandFileMentions("@detz_m2_test.txt", dir).find("detz_m2_test") != std::string::npos);
|
|
assert(detz::ExpandFileMentions("see @nope-missing-xyz", dir).find("@nope") != std::string::npos);
|
|
auto cf = detz::CompleteFiles("detz_m2_test", dir); assert(!cf.empty());
|
|
assert(detz::CompleteFiles("zzz-no-match-xyz", dir).empty());
|
|
std::filesystem::remove(f); }
|
|
{ detz::Config cfg = detz::Config::FromEnv(); assert(!cfg.model.empty());
|
|
assert(!detz::SettingsCatalog().empty());
|
|
assert(!detz::SplitModels("a, b;c").empty());
|
|
auto ids = detz::KnownModelIds(cfg); assert(!ids.empty());
|
|
auto rt = detz::ResolveTag("auto", {}, ids, "/tmp"); assert(rt.first == "auto");
|
|
auto ru = detz::ResolveTag("zzz-unknown-xyz", {}, ids, "/tmp"); assert(ru.first == "unknown");
|
|
std::map<std::string,std::string> bots = {{"bob", "BotBob"}};
|
|
auto rb = detz::ResolveTag("bob", bots, ids, "/tmp"); assert(rb.first == "bot"); }
|
|
std::puts("M2 OK");
|
|
std::puts("M4+M5 OK");
|
|
return 0;
|
|
}
|