@meetkai/mka1 0.48.46 → 0.49.0

package/src/models/components/createvectorstorerequest.ts

@@ -5,6 +5,7 @@
 import * as z from "zod/v3";
 import { remap as remap$ } from "../../lib/primitives.js";
 import { safeParse } from "../../lib/schemas.js";
+import { ClosedEnum } from "../../types/enums.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
 import {
@@ -41,6 +42,18 @@ export type CreateVectorStoreRequestChunkingStrategy =
   | AutoChunkingStrategy
   | StaticChunkingStrategy;
 
+/**
+ * Retrieval mode, frozen at creation (cannot be changed later). 'vector' (default): standard vector similarity search. 'graph': GraphRAG — entities and relations are extracted from every chunk at ingest (metered against your usage) and search traverses the knowledge graph.
+ */
+export const RetrievalMode = {
+  Vector: "vector",
+  Graph: "graph",
+} as const;
+/**
+ * Retrieval mode, frozen at creation (cannot be changed later). 'vector' (default): standard vector similarity search. 'graph': GraphRAG — entities and relations are extracted from every chunk at ingest (metered against your usage) and search traverses the knowledge graph.
+ */
+export type RetrievalMode = ClosedEnum<typeof RetrievalMode>;
+
 /**
  * Request body for creating a vector store.
  */
@@ -54,7 +67,7 @@ export type CreateVectorStoreRequest = {
    */
   description?: string | undefined;
   /**
-   * A list of File IDs that the vector store should use. Useful for tools like file_search that can access files.
+   * A list of File IDs that the vector store should use. Useful for tools like file_search that can access files. At most 500 per request.
    */
   fileIds?: Array<string> | undefined;
   /**
@@ -77,6 +90,18 @@ export type CreateVectorStoreRequest = {
    * The number of dimensions for the embedding vectors. Only supported for models with flexible dimensions. If not specified, uses the model's default dimensions.
    */
   embeddingDimensions?: number | undefined;
+  /**
+   * Retrieval mode, frozen at creation (cannot be changed later). 'vector' (default): standard vector similarity search. 'graph': GraphRAG — entities and relations are extracted from every chunk at ingest (metered against your usage) and search traverses the knowledge graph.
+   */
+  retrievalMode?: RetrievalMode | undefined;
+  /**
+   * Model used for entity/relation extraction on graph stores (ingest-time triplet extraction and query-entity extraction). Defaults to the auto-configured model, resolved at creation time — same contract as embedding_model. Only valid when retrieval_mode is 'graph'.
+   */
+  extractionModel?: string | undefined;
+  /**
+   * Graph expansion depth for graph-mode queries (1-4, engine default 2). Only valid for graph stores.
+   */
+  maxHops?: number | undefined;
 };
 
 /** @internal */
@@ -167,6 +192,15 @@ export function createVectorStoreRequestChunkingStrategyFromJSON(
   );
 }
 
+/** @internal */
+export const RetrievalMode$inboundSchema: z.ZodNativeEnum<
+  typeof RetrievalMode
+> = z.nativeEnum(RetrievalMode);
+/** @internal */
+export const RetrievalMode$outboundSchema: z.ZodNativeEnum<
+  typeof RetrievalMode
+> = RetrievalMode$inboundSchema;
+
 /** @internal */
 export const CreateVectorStoreRequest$inboundSchema: z.ZodType<
   CreateVectorStoreRequest,
@@ -184,6 +218,9 @@ export const CreateVectorStoreRequest$inboundSchema: z.ZodType<
   metadata: z.record(z.string()).optional(),
   embedding_model: z.string().optional(),
   embedding_dimensions: z.number().int().optional(),
+  retrieval_mode: RetrievalMode$inboundSchema.optional(),
+  extraction_model: z.string().optional(),
+  max_hops: z.number().int().optional(),
 }).transform((v) => {
   return remap$(v, {
     "file_ids": "fileIds",
@@ -191,6 +228,9 @@ export const CreateVectorStoreRequest$inboundSchema: z.ZodType<
     "chunking_strategy": "chunkingStrategy",
     "embedding_model": "embeddingModel",
     "embedding_dimensions": "embeddingDimensions",
+    "retrieval_mode": "retrievalMode",
+    "extraction_model": "extractionModel",
+    "max_hops": "maxHops",
   });
 });
 /** @internal */
@@ -206,6 +246,9 @@ export type CreateVectorStoreRequest$Outbound = {
   metadata?: { [k: string]: string } | undefined;
   embedding_model?: string | undefined;
   embedding_dimensions?: number | undefined;
+  retrieval_mode?: string | undefined;
+  extraction_model?: string | undefined;
+  max_hops?: number | undefined;
 };
 
 /** @internal */
@@ -225,6 +268,9 @@ export const CreateVectorStoreRequest$outboundSchema: z.ZodType<
   metadata: z.record(z.string()).optional(),
   embeddingModel: z.string().optional(),
   embeddingDimensions: z.number().int().optional(),
+  retrievalMode: RetrievalMode$outboundSchema.optional(),
+  extractionModel: z.string().optional(),
+  maxHops: z.number().int().optional(),
 }).transform((v) => {
   return remap$(v, {
     fileIds: "file_ids",
@@ -232,6 +278,9 @@ export const CreateVectorStoreRequest$outboundSchema: z.ZodType<
     chunkingStrategy: "chunking_strategy",
     embeddingModel: "embedding_model",
     embeddingDimensions: "embedding_dimensions",
+    retrievalMode: "retrieval_mode",
+    extractionModel: "extraction_model",
+    maxHops: "max_hops",
   });
 });
 

package/src/models/components/vectorstore.ts

@@ -15,6 +15,21 @@ import {
   FileCounts$outboundSchema,
 } from "./filecounts.js";
 
+/**
+ * Retrieval mode, frozen at creation. 'vector': standard vector similarity search. 'graph': GraphRAG — entities and relations are extracted at ingest and queries traverse the knowledge graph. 'hybrid' is reserved.
+ */
+export const VectorStoreRetrievalMode = {
+  Vector: "vector",
+  Hybrid: "hybrid",
+  Graph: "graph",
+} as const;
+/**
+ * Retrieval mode, frozen at creation. 'vector': standard vector similarity search. 'graph': GraphRAG — entities and relations are extracted at ingest and queries traverse the knowledge graph. 'hybrid' is reserved.
+ */
+export type VectorStoreRetrievalMode = ClosedEnum<
+  typeof VectorStoreRetrievalMode
+>;
+
 /**
  * The status of the vector store. 'expired' means the store has expired, 'in_progress' means files are still being processed, 'completed' indicates that the vector store is ready for use.
  */
@@ -76,6 +91,18 @@ export type VectorStore = {
    * The number of dimensions for the embedding vectors in this vector store. Null for legacy vector stores.
    */
   embeddingDimensions: number | null;
+  /**
+   * Retrieval mode, frozen at creation. 'vector': standard vector similarity search. 'graph': GraphRAG — entities and relations are extracted at ingest and queries traverse the knowledge graph. 'hybrid' is reserved.
+   */
+  retrievalMode: VectorStoreRetrievalMode;
+  /**
+   * The model used for entity/relation extraction on graph stores. Null for non-graph stores.
+   */
+  extractionModel?: string | null | undefined;
+  /**
+   * Graph expansion depth for graph-mode queries. Null for non-graph stores.
+   */
+  maxHops?: number | null | undefined;
   /**
    * File processing status counts.
    */
@@ -110,6 +137,15 @@ export type VectorStore = {
   lastUsedAt: number | null;
 };
 
+/** @internal */
+export const VectorStoreRetrievalMode$inboundSchema: z.ZodNativeEnum<
+  typeof VectorStoreRetrievalMode
+> = z.nativeEnum(VectorStoreRetrievalMode);
+/** @internal */
+export const VectorStoreRetrievalMode$outboundSchema: z.ZodNativeEnum<
+  typeof VectorStoreRetrievalMode
+> = VectorStoreRetrievalMode$inboundSchema;
+
 /** @internal */
 export const VectorStoreStatus$inboundSchema: z.ZodNativeEnum<
   typeof VectorStoreStatus
@@ -207,6 +243,9 @@ export const VectorStore$inboundSchema: z.ZodType<
   usage_bytes: z.number().int(),
   embedding_model: z.nullable(z.string()),
   embedding_dimensions: z.nullable(z.number().int()),
+  retrieval_mode: VectorStoreRetrievalMode$inboundSchema,
+  extraction_model: z.nullable(z.string()).optional(),
+  max_hops: z.nullable(z.number().int()).optional(),
   file_counts: FileCounts$inboundSchema,
   status: VectorStoreStatus$inboundSchema,
   expires_after: z.lazy(() => VectorStoreExpiresAfter$inboundSchema).optional(),
@@ -221,6 +260,9 @@ export const VectorStore$inboundSchema: z.ZodType<
     "usage_bytes": "usageBytes",
     "embedding_model": "embeddingModel",
     "embedding_dimensions": "embeddingDimensions",
+    "retrieval_mode": "retrievalMode",
+    "extraction_model": "extractionModel",
+    "max_hops": "maxHops",
     "file_counts": "fileCounts",
     "expires_after": "expiresAfter",
     "expires_at": "expiresAt",
@@ -237,6 +279,9 @@ export type VectorStore$Outbound = {
   usage_bytes: number;
   embedding_model: string | null;
   embedding_dimensions: number | null;
+  retrieval_mode: string;
+  extraction_model?: string | null | undefined;
+  max_hops?: number | null | undefined;
   file_counts: FileCounts$Outbound;
   status: string;
   expires_after?: VectorStoreExpiresAfter$Outbound | undefined;
@@ -260,6 +305,9 @@ export const VectorStore$outboundSchema: z.ZodType<
   usageBytes: z.number().int(),
   embeddingModel: z.nullable(z.string()),
   embeddingDimensions: z.nullable(z.number().int()),
+  retrievalMode: VectorStoreRetrievalMode$outboundSchema,
+  extractionModel: z.nullable(z.string()).optional(),
+  maxHops: z.nullable(z.number().int()).optional(),
   fileCounts: FileCounts$outboundSchema,
   status: VectorStoreStatus$outboundSchema,
   expiresAfter: z.lazy(() => VectorStoreExpiresAfter$outboundSchema).optional(),
@@ -274,6 +322,9 @@ export const VectorStore$outboundSchema: z.ZodType<
     usageBytes: "usage_bytes",
     embeddingModel: "embedding_model",
     embeddingDimensions: "embedding_dimensions",
+    retrievalMode: "retrieval_mode",
+    extractionModel: "extraction_model",
+    maxHops: "max_hops",
     fileCounts: "file_counts",
     expiresAfter: "expires_after",
     expiresAt: "expires_at",