cencori 1.3.0 → 1.4.0

package/dist/index.mjs

@@ -600,6 +600,17 @@ function toBody(request) {
     temperature: request.temperature,
     response_format: request.responseFormat
   };
+  if (request.images && request.images.length > 0) {
+    body.images = request.images.map((img) => {
+      if (img.url) return { url: img.url };
+      if (img.base64) return { base64: img.base64, mime_type: img.mimeType };
+      throw new Error("Each entry in images requires url or base64");
+    });
+    return body;
+  }
+  if (!request.image) {
+    throw new Error("vision request requires `image` or `images[]`");
+  }
   if (request.image.url) {
     body.image_url = request.image.url;
   } else if (request.image.base64) {
@@ -633,6 +644,57 @@ var VisionNamespace = class {
   async classify(request) {
     return this.post("/api/ai/vision/classify", request);
   }
+  /**
+   * Stream a vision analysis. Yields `{ delta }` chunks as text arrives, then
+   * a final `{ done: true, model, provider, usage, cost }` chunk with totals.
+   *
+   * @example
+   * for await (const chunk of cencori.vision.analyzeStream({ image: { url } })) {
+   *   if (chunk.delta) process.stdout.write(chunk.delta);
+   *   if (chunk.done) console.log(chunk.cost);
+   * }
+   */
+  async *analyzeStream(request) {
+    const body = { ...toBody(request), stream: true };
+    const response = await fetch(`${this.config.baseUrl}/api/ai/vision`, {
+      method: "POST",
+      headers: {
+        "CENCORI_API_KEY": this.config.apiKey,
+        "Content-Type": "application/json",
+        ...this.config.headers
+      },
+      body: JSON.stringify(body)
+    });
+    if (!response.ok) {
+      const data = await response.json().catch(() => ({}));
+      const message = data && typeof data === "object" && "message" in data && typeof data.message === "string" ? data.message : `Vision stream request failed with status ${response.status}`;
+      throw new Error(message);
+    }
+    if (!response.body) throw new Error("Response body is null");
+    const reader = response.body.getReader();
+    const decoder = new TextDecoder();
+    let buffer = "";
+    try {
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        buffer += decoder.decode(value, { stream: true });
+        const lines = buffer.split("\n");
+        buffer = lines.pop() ?? "";
+        for (const line of lines) {
+          if (!line.startsWith("data: ")) continue;
+          const data = line.slice(6);
+          if (data === "[DONE]") return;
+          try {
+            yield JSON.parse(data);
+          } catch {
+          }
+        }
+      }
+    } finally {
+      reader.releaseLock();
+    }
+  }
   async post(path, request) {
     const response = await fetch(`${this.config.baseUrl}${path}`, {
       method: "POST",
@@ -656,6 +718,66 @@ var VisionNamespace = class {
   }
 };
 
+// src/documents/index.ts
+function toBody2(request, extra = {}) {
+  const body = {
+    prompt: request.prompt,
+    model: request.model,
+    max_tokens: request.maxTokens,
+    temperature: request.temperature,
+    ...extra
+  };
+  if (request.document.url) {
+    body.document_url = request.document.url;
+  } else if (request.document.base64) {
+    body.document_base64 = request.document.base64;
+    body.mime_type = request.document.mimeType;
+    body.filename = request.document.filename;
+  } else {
+    throw new Error("documents request requires document.url or document.base64");
+  }
+  return body;
+}
+var DocumentsNamespace = class {
+  constructor(config) {
+    this.config = config;
+  }
+  /** Extract text from a PDF (native, no LLM) or image (via Vision OCR). */
+  async extract(request) {
+    return this.post("/api/ai/documents/extract", toBody2(request));
+  }
+  /** Extract then summarize the document with a chat model. */
+  async summarize(request) {
+    return this.post("/api/ai/documents/summarize", toBody2(request));
+  }
+  /** Extract then answer a question about the document. */
+  async query(request) {
+    if (!request.question) throw new Error("documents.query requires a `question`");
+    return this.post("/api/ai/documents/query", toBody2(request, { question: request.question }));
+  }
+  async post(path, body) {
+    const response = await fetch(`${this.config.baseUrl}${path}`, {
+      method: "POST",
+      headers: {
+        "CENCORI_API_KEY": this.config.apiKey,
+        "Content-Type": "application/json",
+        ...this.config.headers
+      },
+      body: JSON.stringify(body)
+    });
+    const data = await response.json();
+    if (!response.ok) {
+      const message = data && typeof data === "object" && "message" in data && typeof data.message === "string" ? data.message : `Documents request failed with status ${response.status}`;
+      const code = data && typeof data === "object" && "error" in data && typeof data.error === "string" ? data.error : "request_failed";
+      const err = new Error(message);
+      err.code = code;
+      err.details = data;
+      throw err;
+    }
+    return data;
+  }
+};
+
 // src/sessions/index.ts
 var SessionsNamespace = class {
   constructor(config) {
@@ -1180,6 +1302,7 @@ var Cencori = class {
     };
     this.ai = new AINamespace(this.config);
     this.vision = new VisionNamespace(this.config);
+    this.documents = new DocumentsNamespace(this.config);
     this.agents = new AgentsNamespace(this.config);
     this.compute = new ComputeNamespace();
     this.workflow = new WorkflowNamespace();
@@ -1606,6 +1729,7 @@ export {
   Cencori,
   CencoriError,
   ComputeNamespace,
+  DocumentsNamespace,
   MemoryClient,
   RateLimitError,
   SafetyError,