cencori 1.3.1 → 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) {
@@ -707,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) {
@@ -1231,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();
@@ -1657,6 +1729,7 @@ export {
   Cencori,
   CencoriError,
   ComputeNamespace,
+  DocumentsNamespace,
   MemoryClient,
   RateLimitError,
   SafetyError,