write-language 0.1.94 → 0.1.96

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "write-language",
-  "version": "0.1.94",
+  "version": "0.1.96",
   "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

@@ -0,0 +1,69 @@
+/**
+ * @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;
+  prompt: string;
+  color?: string;
+}
+
+export const DEFAULT_REWRITE_MODES: RewriteMode[] = [
+  {
+    id: 'clarity',
+    name: 'Clarity',
+    prompt: 'Rewrite this paragraph for maximum clarity and straightforwardness, keeping all original meaning but removing ambiguity and simplifying complex sentences:',
+    color: 'blue',
+  },
+  {
+    id: 'concise',
+    name: 'Concise',
+    prompt: 'Rewrite this text to be more concise, removing redundancy and filler while preserving all key points and tone. Aim for about 50% of the original length:',
+    color: 'purple',
+  },
+  {
+    id: 'summarize',
+    name: 'Summarize',
+    prompt: 'Summarize the following text into a shorter paragraph, keeping the main ideas and overall tone but removing details and repetition:',
+    color: 'green',
+  },
+  {
+    id: 'rephrase',
+    name: 'Rephrase',
+    prompt: 'Rephrase this paragraph with fresh wording and more engaging style, varying sentence structure and word choice while preserving the core message:',
+    color: 'orange',
+  },
+  {
+    id: 'expand',
+    name: 'Expand',
+    prompt: 'Keep the original paragraph as-is, then expand it by adding one additional paragraph that elaborates on the main idea, gives an example, or adds helpful context for the reader:',
+    color: 'pink',
+  },
+];
+
+const STORAGE_KEY = 'REASON-rewrite-modes';
+
+export const getRewriteModes = (): RewriteMode[] => {
+  try {
+    const stored = localStorage.getItem(STORAGE_KEY);
+    if (stored) return JSON.parse(stored);
+  } catch (e) {
+    console.error('Failed to load rewrite modes:', e);
+  }
+  return DEFAULT_REWRITE_MODES;
+};
+
+export const saveRewriteModes = (modes: RewriteMode[]): void => {
+  try {
+    localStorage.setItem(STORAGE_KEY, JSON.stringify(modes));
+  } catch (e) {
+    console.error('Failed to save rewrite modes:', e);
+  }
+};
+
+export const resetRewriteModes = (): void => {
+  saveRewriteModes(DEFAULT_REWRITE_MODES);
+};

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/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" });
-  });
-});