wehub-resource-sync
109 lines
4.0 KiB
TypeScript
109 lines
4.0 KiB
TypeScript
import { describe, it } from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
const mod = await import("../../open-sse/services/intentClassifier.ts");
|
|
|
|
describe("intentClassifier", () => {
|
|
describe("classifyPromptIntent", () => {
|
|
it("classifies code intent", () => {
|
|
assert.equal(mod.classifyPromptIntent("Write a Python function to sort"), "code");
|
|
assert.equal(mod.classifyPromptIntent("Debug this JavaScript code"), "code");
|
|
assert.equal(mod.classifyPromptIntent("How to use async/await"), "code");
|
|
});
|
|
|
|
it("classifies math intent", () => {
|
|
assert.equal(mod.classifyPromptIntent("Solve this equation: x^2 + 3x = 0"), "math");
|
|
assert.equal(mod.classifyPromptIntent("Calculate the derivative"), "math");
|
|
});
|
|
|
|
it("classifies reasoning intent", () => {
|
|
assert.equal(mod.classifyPromptIntent("Explain the reasoning behind quantum mechanics"), "reasoning");
|
|
});
|
|
|
|
it("classifies creative intent", () => {
|
|
assert.equal(mod.classifyPromptIntent("Compose a poem about the ocean"), "creative");
|
|
assert.equal(mod.classifyPromptIntent("Craft a short story about dragons"), "creative");
|
|
});
|
|
|
|
it("classifies simple intent for short prompts", () => {
|
|
assert.equal(mod.classifyPromptIntent("What is 2+2?"), "simple");
|
|
});
|
|
|
|
it("returns medium for unclassified prompts", () => {
|
|
assert.equal(mod.classifyPromptIntent("Describe the fall of the Roman Empire"), "medium");
|
|
});
|
|
|
|
it("considers system prompt in classification", () => {
|
|
assert.equal(mod.classifyPromptIntent("Hello", "You are a Python coding assistant"), "code");
|
|
});
|
|
|
|
it("code keywords take priority over math", () => {
|
|
assert.equal(mod.classifyPromptIntent("Write code to calculate derivatives"), "code");
|
|
});
|
|
|
|
it("handles empty prompt", () => {
|
|
const result = mod.classifyPromptIntent("");
|
|
assert.ok(["simple", "medium"].includes(result));
|
|
});
|
|
});
|
|
|
|
describe("classifyWithConfig", () => {
|
|
it("returns medium when disabled", () => {
|
|
const result = mod.classifyWithConfig("Write code", { enabled: false });
|
|
assert.equal(result, "medium");
|
|
});
|
|
|
|
it("uses extra keywords", () => {
|
|
const result = mod.classifyPromptIntent("deploy the application");
|
|
// With extra keywords
|
|
const withExtra = mod.classifyWithConfig("deploy the application", {
|
|
enabled: true,
|
|
extraCodeKeywords: ["deploy"],
|
|
});
|
|
assert.equal(withExtra, "code");
|
|
});
|
|
|
|
it("respects custom simpleMaxWords", () => {
|
|
const shortPrompt = "word ".repeat(10).trim();
|
|
// With default 60 words, this should be simple if it matches
|
|
const result = mod.classifyWithConfig(shortPrompt, { enabled: true, simpleMaxWords: 5 });
|
|
// Won't be simple since it doesn't match simple keywords, but the word limit is respected
|
|
assert.ok(["simple", "medium"].includes(result));
|
|
});
|
|
});
|
|
|
|
describe("keyword arrays", () => {
|
|
it("CODE_KEYWORDS is a non-empty readonly array", () => {
|
|
assert.ok(Array.isArray(mod.CODE_KEYWORDS));
|
|
assert.ok(mod.CODE_KEYWORDS.length > 0);
|
|
});
|
|
|
|
it("REASONING_KEYWORDS is a non-empty readonly array", () => {
|
|
assert.ok(Array.isArray(mod.REASONING_KEYWORDS));
|
|
assert.ok(mod.REASONING_KEYWORDS.length > 0);
|
|
});
|
|
|
|
it("MATH_KEYWORDS is a non-empty readonly array", () => {
|
|
assert.ok(Array.isArray(mod.MATH_KEYWORDS));
|
|
assert.ok(mod.MATH_KEYWORDS.length > 0);
|
|
});
|
|
|
|
it("CREATIVE_KEYWORDS is a non-empty readonly array", () => {
|
|
assert.ok(Array.isArray(mod.CREATIVE_KEYWORDS));
|
|
assert.ok(mod.CREATIVE_KEYWORDS.length > 0);
|
|
});
|
|
|
|
it("SIMPLE_KEYWORDS is a non-empty readonly array", () => {
|
|
assert.ok(Array.isArray(mod.SIMPLE_KEYWORDS));
|
|
assert.ok(mod.SIMPLE_KEYWORDS.length > 0);
|
|
});
|
|
});
|
|
|
|
describe("DEFAULT_INTENT_CONFIG", () => {
|
|
it("has expected shape", () => {
|
|
assert.equal(mod.DEFAULT_INTENT_CONFIG.enabled, true);
|
|
assert.equal(mod.DEFAULT_INTENT_CONFIG.simpleMaxWords, 60);
|
|
});
|
|
});
|
|
});
|