87 lines
2.3 KiB
TypeScript
87 lines
2.3 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import { formatAlphaSearchContext, sanitizeAlphaSearchPayload } from "../extensions/research-tools/alpha.js";
|
|
import { formatToolText } from "../extensions/research-tools/shared.js";
|
|
|
|
test("sanitizeAlphaSearchPayload drops raw alpha search text while keeping parsed hits", () => {
|
|
const payload = sanitizeAlphaSearchPayload({
|
|
query: "scaling laws",
|
|
mode: "all",
|
|
semantic: {
|
|
raw: "\n\n\n1. **Paper A**\n- Abstract: noisy raw block",
|
|
results: [
|
|
{
|
|
rank: 1,
|
|
title: "Paper A",
|
|
publishedAt: "2025-09-28",
|
|
organizations: "Stanford University, EPFL",
|
|
authors: "A. Author, B. Author",
|
|
abstract: "Line one.\n\n\nLine two.",
|
|
arxivId: "2509.24012",
|
|
arxivUrl: "https://arxiv.org/abs/2509.24012",
|
|
alphaXivUrl: "https://www.alphaxiv.org/overview/2509.24012",
|
|
raw: "internal raw block that should be dropped",
|
|
},
|
|
],
|
|
},
|
|
keyword: {
|
|
raw: "\n\n\nNoisy keyword fallback",
|
|
results: [],
|
|
},
|
|
});
|
|
|
|
assert.deepEqual(payload, {
|
|
query: "scaling laws",
|
|
mode: "all",
|
|
semantic: {
|
|
count: 1,
|
|
results: [
|
|
{
|
|
rank: 1,
|
|
title: "Paper A",
|
|
publishedAt: "2025-09-28",
|
|
organizations: "Stanford University, EPFL",
|
|
authors: "A. Author, B. Author",
|
|
abstract: "Line one. Line two.",
|
|
arxivId: "2509.24012",
|
|
arxivUrl: "https://arxiv.org/abs/2509.24012",
|
|
alphaXivUrl: "https://www.alphaxiv.org/overview/2509.24012",
|
|
},
|
|
],
|
|
},
|
|
keyword: {
|
|
count: 0,
|
|
results: [],
|
|
note: "Noisy keyword fallback",
|
|
},
|
|
});
|
|
});
|
|
|
|
test("formatAlphaSearchContext emits compact model-facing text without raw JSON escapes", () => {
|
|
const text = formatAlphaSearchContext({
|
|
query: "scaling laws",
|
|
mode: "semantic",
|
|
results: [
|
|
{
|
|
rank: 1,
|
|
title: "Paper A",
|
|
abstract: "First line.\n\n\nSecond line.",
|
|
arxivId: "2509.24012",
|
|
raw: "should not appear",
|
|
},
|
|
],
|
|
raw: "\n\n\nvery noisy raw payload",
|
|
});
|
|
|
|
assert.match(text, /query: scaling laws/);
|
|
assert.match(text, /1\. Paper A/);
|
|
assert.match(text, /abstract: First line\. Second line\./);
|
|
assert.ok(!text.includes("\\n"));
|
|
assert.ok(!text.includes("raw"));
|
|
});
|
|
|
|
test("formatToolText collapses excess blank lines in plain strings", () => {
|
|
assert.equal(formatToolText("alpha\n\n\n\nbeta"), "alpha\n\nbeta");
|
|
});
|