write-language 0.1.95 → 0.1.97

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "write-language",
-  "version": "0.1.95",
+  "version": "0.1.97",
   "description": "Multi-provider language generation toolkit using Vercel AI SDK - generate responses with 10+ LLM providers including OpenAI, Anthropic, Google, and more.",
   "author": "vtempest <grokthiscontact@gmail.com>",
   "license": "rights.institute/PROSPER",
@@ -43,6 +43,7 @@
   "scripts": {
     "build": "vite build",
     "test": "vitest",
+    "test:coverage": "vitest run --coverage",
     "test-ui": "vitest --ui --watch",
     "make": "rm -rf dist/*; NODE_OPTIONS=--max-old-space-size=15192 vite build",
     "ship": "npx standard-version --release-as patch && npm publish --access public"
@@ -60,6 +61,7 @@
     }
   },
   "devDependencies": {
+    "@vitest/coverage-v8": "^4.0.18",
     "@vitest/ui": "^4.0.18",
     "terser": "^5.46.0",
     "typescript": "^5.9.3",

package/src/language-model-families.ts

@@ -1,3 +1,8 @@
+/**
+ * @fileoverview Static metadata for well-known LLM model families (Claude, ChatGPT,
+ * Gemini, Llama, etc.) used for display purposes such as provider/maker attribution,
+ * flagship model naming, and open vs. closed licensing status.
+ */
 
 export interface LanguageModelFamily {
   model_family: string;

package/src/provider-factory.ts

@@ -71,6 +71,12 @@ export function createLLMProvider(
     case "openrouter":
       return createOpenRouter({ apiKey }).chat(model);
 
+    case "anyapi":
+      return createOpenAI({
+        apiKey,
+        baseURL: "https://api.anyapi.ai/v1",
+      })(model);
+
     case "perplexity":
       return createOpenAI({
         apiKey,

package/src/rewrite-modes.ts

@@ -1,3 +1,9 @@
+/**
+ * @fileoverview Defines the default text-rewrite mode presets (clarity, concise,
+ * summarize, rephrase, expand) and helpers to read/write/reset the user's custom
+ * modes from `localStorage`.
+ */
+
 export interface RewriteMode {
   id: string;
   name: string;

package/src/tools/qwksearch-api-tools.ts

@@ -1 +1,9 @@
+/**
+ * @fileoverview Registry of agent tools (function-calling definitions backed by the
+ * qwksearch API) that can be attached to agent prompts in prompt-templates.ts.
+ *
+ * Currently empty; populate with `AgentTool` entries (name, schema, func) as
+ * qwksearch-backed tools are implemented.
+ */
+
 export const AGENT_TOOLS: any[] = [];

package/src/utils/prism.ts

@@ -1,3 +1,8 @@
+/**
+ * @fileoverview Configures Prism.js with a fixed set of language grammars and exposes
+ * a small `highlightCode` helper used by markdown-to-html.ts for code-block syntax
+ * highlighting.
+ */
 import Prism from "prismjs";
 import "prismjs/components/prism-markup";
 import "prismjs/components/prism-css";

package/dist/generate-response.attachments.test.d.ts

@@ -1 +0,0 @@
-export {};

package/dist/rewrite-modes.test.d.ts

@@ -1 +0,0 @@
-export {};

package/src/generate-response.attachments.test.ts

@@ -1,110 +0,0 @@
-/**
- * @fileoverview Verifies that file/image attachments are forwarded to the
- * Vercel AI SDK as multimodal `messages` content parts, and that plain
- * (attachment-free) calls keep using the simpler `prompt` form.
- */
-import { afterEach, describe, expect, it, vi } from "vitest";
-
-// Capture the exact arguments passed to generateText so we can assert on the
-// shape of the request the SDK receives.
-const generateTextMock = vi.fn((_args: unknown) => Promise.resolve({ text: "ok" }));
-
-vi.mock("ai", () => ({
-  generateText: (args: unknown) => generateTextMock(args),
-  stepCountIs: (n: number) => n,
-  tool: (t: unknown) => t,
-}));
-
-// Return a truthy fake model so writeLanguageResponse proceeds to generateText.
-vi.mock("./provider-factory", () => ({
-  createLLMProvider: () => ({ id: "fake-model" }),
-}));
-
-// Avoid pulling the markdown/prism stack into the test.
-vi.mock("./utils/markdown-to-html", () => ({
-  convertMarkdownToHTMLEscaped: async (s: string) => s,
-}));
-
-import { writeLanguageResponse } from "./generate-response";
-
-afterEach(() => {
-  generateTextMock.mockClear();
-});
-
-describe("writeLanguageResponse attachments", () => {
-  const base = {
-    provider: "groq",
-    apiKey: "test-key",
-    agent: "question",
-    query: "Describe the attached file",
-    html: false,
-    applyContextLimit: false,
-  } as const;
-
-  it("sends a plain text prompt when there are no attachments", async () => {
-    await writeLanguageResponse({ ...base });
-
-    expect(generateTextMock).toHaveBeenCalledTimes(1);
-    const arg = generateTextMock.mock.calls[0][0] as Record<string, unknown>;
-    expect(typeof arg.prompt).toBe("string");
-    expect(arg.messages).toBeUndefined();
-  });
-
-  it("sends an image attachment as an image content part", async () => {
-    await writeLanguageResponse({
-      ...base,
-      attachments: [
-        { mediaType: "image/png", data: "data:image/png;base64,AAAA" },
-      ],
-    });
-
-    const arg = generateTextMock.mock.calls[0][0] as any;
-    expect(arg.prompt).toBeUndefined();
-    expect(Array.isArray(arg.messages)).toBe(true);
-    const parts = arg.messages[0].content;
-    expect(parts[0]).toMatchObject({ type: "text" });
-    expect(parts[1]).toMatchObject({
-      type: "image",
-      image: "data:image/png;base64,AAAA",
-    });
-  });
-
-  it("sends a document attachment as a file content part with its mediaType", async () => {
-    await writeLanguageResponse({
-      ...base,
-      attachments: [
-        {
-          mediaType: "application/pdf",
-          data: "JVBERi0=",
-          filename: "report.pdf",
-        },
-      ],
-    });
-
-    const arg = generateTextMock.mock.calls[0][0] as any;
-    const parts = arg.messages[0].content;
-    expect(parts[1]).toMatchObject({
-      type: "file",
-      mediaType: "application/pdf",
-      data: "JVBERi0=",
-      filename: "report.pdf",
-    });
-  });
-
-  it("skips malformed attachments (missing data or mediaType)", async () => {
-    await writeLanguageResponse({
-      ...base,
-      attachments: [
-        { mediaType: "", data: "x" } as any,
-        { mediaType: "image/png" } as any,
-        { mediaType: "image/jpeg", data: "data:image/jpeg;base64,BBBB" },
-      ],
-    });
-
-    const arg = generateTextMock.mock.calls[0][0] as any;
-    const parts = arg.messages[0].content;
-    // 1 text part + 1 valid image part only.
-    expect(parts).toHaveLength(2);
-    expect(parts[1]).toMatchObject({ type: "image" });
-  });
-});

package/src/rewrite-modes.test.ts

@@ -1,96 +0,0 @@
-import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
-import {
-  DEFAULT_REWRITE_MODES,
-  getRewriteModes,
-  saveRewriteModes,
-  resetRewriteModes,
-  type RewriteMode,
-} from './rewrite-modes'
-
-const STORAGE_KEY = 'REASON-rewrite-modes'
-
-// Minimal in-memory localStorage stub for the node test environment.
-function createLocalStorageStub() {
-  const store = new Map<string, string>()
-  return {
-    getItem: vi.fn((k: string) => (store.has(k) ? store.get(k)! : null)),
-    setItem: vi.fn((k: string, v: string) => {
-      store.set(k, v)
-    }),
-    removeItem: vi.fn((k: string) => {
-      store.delete(k)
-    }),
-    clear: vi.fn(() => store.clear()),
-    _store: store,
-  }
-}
-
-let ls: ReturnType<typeof createLocalStorageStub>
-
-beforeEach(() => {
-  ls = createLocalStorageStub()
-  vi.stubGlobal('localStorage', ls)
-})
-
-afterEach(() => {
-  vi.unstubAllGlobals()
-})
-
-describe('DEFAULT_REWRITE_MODES', () => {
-  it('includes the five built-in modes with unique ids', () => {
-    const ids = DEFAULT_REWRITE_MODES.map((m) => m.id)
-    expect(ids).toEqual(['clarity', 'concise', 'summarize', 'rephrase', 'expand'])
-    expect(new Set(ids).size).toBe(ids.length)
-  })
-
-  it('gives every mode a name and prompt', () => {
-    for (const mode of DEFAULT_REWRITE_MODES) {
-      expect(mode.name).toBeTruthy()
-      expect(mode.prompt.length).toBeGreaterThan(0)
-    }
-  })
-})
-
-describe('getRewriteModes', () => {
-  it('returns the defaults when nothing is stored', () => {
-    expect(getRewriteModes()).toEqual(DEFAULT_REWRITE_MODES)
-  })
-
-  it('returns the stored modes when present', () => {
-    const custom: RewriteMode[] = [{ id: 'x', name: 'X', prompt: 'do x' }]
-    ls._store.set(STORAGE_KEY, JSON.stringify(custom))
-    expect(getRewriteModes()).toEqual(custom)
-  })
-
-  it('falls back to defaults on malformed JSON', () => {
-    ls._store.set(STORAGE_KEY, '{not valid json')
-    const spy = vi.spyOn(console, 'error').mockImplementation(() => {})
-    expect(getRewriteModes()).toEqual(DEFAULT_REWRITE_MODES)
-    expect(spy).toHaveBeenCalled()
-  })
-})
-
-describe('saveRewriteModes', () => {
-  it('serialises the modes to localStorage', () => {
-    const modes: RewriteMode[] = [{ id: 'y', name: 'Y', prompt: 'do y' }]
-    saveRewriteModes(modes)
-    expect(JSON.parse(ls._store.get(STORAGE_KEY)!)).toEqual(modes)
-  })
-
-  it('logs but does not throw when storage rejects', () => {
-    ls.setItem.mockImplementation(() => {
-      throw new Error('quota exceeded')
-    })
-    const spy = vi.spyOn(console, 'error').mockImplementation(() => {})
-    expect(() => saveRewriteModes([])).not.toThrow()
-    expect(spy).toHaveBeenCalled()
-  })
-})
-
-describe('resetRewriteModes', () => {
-  it('writes the defaults back to storage', () => {
-    ls._store.set(STORAGE_KEY, JSON.stringify([{ id: 'z', name: 'Z', prompt: 'z' }]))
-    resetRewriteModes()
-    expect(JSON.parse(ls._store.get(STORAGE_KEY)!)).toEqual(DEFAULT_REWRITE_MODES)
-  })
-})