@babav/knowledge-core-client 0.7.0 → 0.9.0

package/dist/index.d.ts

@@ -169,6 +169,8 @@ export interface Feedback {
 export interface Agent {
     id: UUID;
     name: string;
+    identity_prompt: string | null;
+    response_prompt: string | null;
     generation_model: string | null;
     top_k: number | null;
     rerank_instruction: string | null;
@@ -294,9 +296,11 @@ export declare class KnowledgeCoreClient extends HttpBase {
             limit?: number;
             cursor?: string;
         }) => Promise<Page<Document>>;
-        /** Upload a file and ingest it ASYNCHRONOUSLY — resolves with the created
-         *  document at status `pending` (HTTP 202); track via documents.get() polling
-         *  the status to `indexed`/`failed`, or the tenant result webhook. */
+        /** Ingest a document ASYNCHRONOUSLY — resolves with the created document at
+         *  status `pending` (HTTP 202); track via documents.get() polling to
+         *  `indexed`/`failed`, or the tenant result webhook. Auto-routes by size: files
+         *  over ~30 MB go via the signed-URL upload path (Cloud Run caps requests at
+         *  ~32 MB), smaller files via one multipart request. One method, any size. */
         ingestDocument: (id: UUID, a: {
             file: FileData;
             filename: string;
@@ -459,6 +463,8 @@ export declare class AdminClient extends HttpBase {
     agents: {
         create: (b: {
             name: string;
+            identity_prompt?: string;
+            response_prompt?: string;
             generation_model?: string;
             top_k?: number;
             rerank_instruction?: string;

package/dist/index.js

@@ -148,6 +148,16 @@ function toBlob(data, contentType) {
         return data;
     return new Blob([data], { type: contentType ?? "application/octet-stream" });
 }
+function fileSize(data) {
+    if (data instanceof Blob)
+        return data.size;
+    if (data instanceof ArrayBuffer)
+        return data.byteLength;
+    return data.byteLength;
+}
+// Cloud Run caps request bodies at ~32 MB; over this, ingestDocument auto-routes to
+// the signed-URL upload path instead of a single multipart request.
+const MULTIPART_MAX_BYTES = 30 * 1024 * 1024;
 // ---------------------------------------------------------------------------
 // Tenant client (data ops) — use a TENANT key
 // ---------------------------------------------------------------------------
@@ -203,10 +213,14 @@ export class KnowledgeCoreClient extends HttpBase {
         countDocuments: (id) => this.request("GET", `/v1/corpora/${id}/documents/count`),
         /** Documents in the corpus whose filename contains `q` (case-insensitive), paginated. */
         searchDocuments: (id, q, opts) => this.request("GET", `/v1/corpora/${id}/documents/search`, { query: { q, ...opts } }),
-        /** Upload a file and ingest it ASYNCHRONOUSLY — resolves with the created
-         *  document at status `pending` (HTTP 202); track via documents.get() polling
-         *  the status to `indexed`/`failed`, or the tenant result webhook. */
+        /** Ingest a document ASYNCHRONOUSLY — resolves with the created document at
+         *  status `pending` (HTTP 202); track via documents.get() polling to
+         *  `indexed`/`failed`, or the tenant result webhook. Auto-routes by size: files
+         *  over ~30 MB go via the signed-URL upload path (Cloud Run caps requests at
+         *  ~32 MB), smaller files via one multipart request. One method, any size. */
         ingestDocument: (id, a) => {
+            if (fileSize(a.file) > MULTIPART_MAX_BYTES)
+                return this.corpora.uploadDocument(id, a);
             const fd = new FormData();
             fd.append("file", toBlob(a.file, a.content_type), a.filename);
             if (a.custom_metadata)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@babav/knowledge-core-client",
-  "version": "0.7.0",
+  "version": "0.9.0",
   "description": "TypeScript client for the Babav Knowledge Core API (Deno + Node 18+, zero deps).",
   "type": "module",
   "main": "./dist/index.js",

package/src/index.ts

@@ -183,6 +183,8 @@ export interface Feedback {
 export interface Agent {
   id: UUID;
   name: string;
+  identity_prompt: string | null;   // answer-system: who/purpose (null => server default)
+  response_prompt: string | null;   // answer-system: output guidelines (null => server default)
   generation_model: string | null;
   top_k: number | null;
   rerank_instruction: string | null;
@@ -363,6 +365,16 @@ function toBlob(data: FileData, contentType?: string): Blob {
   return new Blob([data as BlobPart], { type: contentType ?? "application/octet-stream" });
 }
 
+function fileSize(data: FileData): number {
+  if (data instanceof Blob) return data.size;
+  if (data instanceof ArrayBuffer) return data.byteLength;
+  return (data as Uint8Array).byteLength;
+}
+
+// Cloud Run caps request bodies at ~32 MB; over this, ingestDocument auto-routes to
+// the signed-URL upload path instead of a single multipart request.
+const MULTIPART_MAX_BYTES = 30 * 1024 * 1024;
+
 // ---------------------------------------------------------------------------
 // SSE handlers
 // ---------------------------------------------------------------------------
@@ -438,10 +450,13 @@ export class KnowledgeCoreClient extends HttpBase {
     /** Documents in the corpus whose filename contains `q` (case-insensitive), paginated. */
     searchDocuments: (id: UUID, q: string, opts?: { limit?: number; cursor?: string }) =>
       this.request<Page<Document>>("GET", `/v1/corpora/${id}/documents/search`, { query: { q, ...opts } }),
-    /** Upload a file and ingest it ASYNCHRONOUSLY — resolves with the created
-     *  document at status `pending` (HTTP 202); track via documents.get() polling
-     *  the status to `indexed`/`failed`, or the tenant result webhook. */
+    /** Ingest a document ASYNCHRONOUSLY — resolves with the created document at
+     *  status `pending` (HTTP 202); track via documents.get() polling to
+     *  `indexed`/`failed`, or the tenant result webhook. Auto-routes by size: files
+     *  over ~30 MB go via the signed-URL upload path (Cloud Run caps requests at
+     *  ~32 MB), smaller files via one multipart request. One method, any size. */
     ingestDocument: (id: UUID, a: { file: FileData; filename: string; content_type?: string; visibility?: Visibility; folder_id?: UUID; custom_metadata?: Record<string, unknown> }) => {
+      if (fileSize(a.file) > MULTIPART_MAX_BYTES) return this.corpora.uploadDocument(id, a);
       const fd = new FormData();
       fd.append("file", toBlob(a.file, a.content_type), a.filename);
       if (a.custom_metadata) fd.append("custom_metadata", JSON.stringify(a.custom_metadata));
@@ -588,7 +603,7 @@ export class AdminClient extends HttpBase {
 
   // Agents are query agents, global for now.
   agents = {
-    create: (b: { name: string; generation_model?: string; top_k?: number; rerank_instruction?: string; max_hops?: number; groundedness_threshold?: number; grounding_enabled?: boolean }) =>
+    create: (b: { name: string; identity_prompt?: string; response_prompt?: string; generation_model?: string; top_k?: number; rerank_instruction?: string; max_hops?: number; groundedness_threshold?: number; grounding_enabled?: boolean }) =>
       this.request<Agent>("POST", "/v1/agents", { json: b }),
     update: (id: UUID, b: Partial<Omit<Agent, "id">>) =>
       this.request<Agent>("PATCH", `/v1/agents/${id}`, { json: b }),