codemie-sdk 0.1.229 → 0.1.230

package/README.md

@@ -212,7 +212,7 @@ await client.assistants.delete("assistant-id");
 
 #### Advanced Features
 
-6. **Chat with Assistant**
+6. **Chat with Assistant (with MCP header propagation)**
 ```typescript
 const params: AssistantChatParams = {
   text: "Your message here",
@@ -224,11 +224,38 @@ const params: AssistantChatParams = {
   system_prompt: "Override default system prompt",
   background_task: false,
   content_raw: "", // Raw content if needed
-  file_name: "", // For file processing
   top_k: 1,
-  metadata: {} // Additional metadata
+  metadata: {}, // Additional metadata
+  propagate_headers: true, // Enable propagation of X-* headers
+};
+
+// Pass X-* headers to be forwarded to MCP servers
+const response = await client.assistants.chat(
+  "assistant-id",
+  params,
+  {
+    "X-Tenant-ID": "tenant-abc-123",
+    "X-User-ID": "user-456",
+    "X-Request-ID": "req-123",
+  }
+);
+```
+
+6.a. **Chat with Assistant by slug (with MCP header propagation)**
+```typescript
+const params: AssistantChatParams = {
+  text: "Your message here",
+  propagate_headers: true,
 };
-const response = await client.assistants.chat("assistant-id", params);
+
+const response = await client.assistants.chatBySlug(
+  "assistant-slug",
+  params,
+  {
+    "X-Environment": "production",
+    "X-Feature-Flag-Beta": "true",
+  }
+);
 ```
 7. **Utilize structured outputs with Assistant
 Uzing zod objects
@@ -679,10 +706,19 @@ const result = await client.workflows.delete("workflow-id");
 
 The SDK provides comprehensive workflow execution management through the WorkflowExecutionService:
 
-1. **Run Workflow**
+1. **Run Workflow (with MCP header propagation)**
 ```typescript
-// Simple workflow execution
-const execution = await client.workflows.run("workflow-id", {userInput: "optional input"});
+// Enable propagation in payload and pass X-* headers to forward to MCP servers
+const execution = await client.workflows.run(
+  "workflow-id",
+  "optional input",
+  undefined, // fileName
+  true,      // propagateHeaders
+  {
+    "X-Request-ID": "req-abc-123",
+    "X-Source-App": "analytics-ui",
+  },
+);
 
 // Get execution service for advanced operations
 const executionService = client.workflows.executions("workflow-id");
@@ -702,8 +738,14 @@ const execution = await executionService.get("execution-id");
 // Abort running execution
 const result = await executionService.abort("execution-id");
 
-// Resume interrupted execution
-const result = await executionService.resume("execution-id");
+// Resume interrupted execution with header propagation (query param + headers)
+const result = await executionService.resume(
+  "execution-id",
+  true, // propagateHeaders
+  {
+    "X-Correlation-ID": "corr-456",
+  }
+);
 
 // Delete all executions
 const result = await executionService.deleteAll();

package/dist/index.cjs

@@ -226,6 +226,8 @@ var AssistantChatParamsSchema = import_zod.z.object({
   ]),
   history_index: import_zod.z.number().optional(),
   stream: import_zod.z.boolean().optional(),
+  propagate_headers: import_zod.z.boolean().optional(),
+  custom_metadata: import_zod.z.record(import_zod.z.unknown()).optional(),
   top_k: import_zod.z.number().optional(),
   system_prompt: import_zod.z.string().optional(),
   background_task: import_zod.z.boolean().optional(),
@@ -296,11 +298,22 @@ var ApiRequestHandler = class {
     });
     return response.data;
   }
-  async get(path, params, config = {}) {
-    return this.request("get", path, { ...config, params });
+  /**
+   * Merge extra headers into the request (used for X-* propagation)
+   */
+  withHeaders(config = {}, extraHeaders) {
+    if (!extraHeaders || Object.keys(extraHeaders).length === 0) return config;
+    const headers = config.headers ?? {};
+    return { ...config, headers: { ...headers, ...extraHeaders } };
+  }
+  async get(path, params, config = {}, extraHeaders) {
+    return this.request("get", path, this.withHeaders({ ...config, params }, extraHeaders));
+  }
+  async post(path, data, config = {}, extraHeaders) {
+    return this.request("post", path, this.withHeaders({ ...config, data }, extraHeaders));
   }
-  async post(path, data, config = {}) {
-    return this.request("post", path, { ...config, data });
+  async put(path, data, config = {}, extraHeaders) {
+    return this.request("put", path, this.withHeaders({ ...config, data }, extraHeaders));
   }
   async postMultipart(url, formData, params) {
     const response = await this.client.post(url, formData, {
@@ -311,9 +324,6 @@ var ApiRequestHandler = class {
     });
     return response.data;
   }
-  async put(path, data, config = {}) {
-    return this.request("put", path, { ...config, data });
-  }
   async delete(path, config = {}) {
     return this.request("delete", path, config);
   }
@@ -446,7 +456,7 @@ var AssistantService = class {
   /**
    * Send a chat request to an assistant.
    */
-  async chat(assistantId, _params) {
+  async chat(assistantId, _params, headers) {
     let zodSchema = void 0;
     let params = { ..._params };
     if (params.output_schema && params.output_schema instanceof import_zod2.z.ZodType) {
@@ -461,7 +471,33 @@ var AssistantService = class {
       params,
       {
         responseType: params.stream ? "stream" : void 0
-      }
+      },
+      headers
+    );
+    const mapped = AssistantMapper.mapBaseModelApiResponse(response);
+    if (!params.stream && zodSchema && mapped.generated) {
+      mapped.generated = zodSchema.parse(mapped.generated);
+    }
+    return mapped;
+  }
+  /**
+   * Send a chat request to an assistant by slug.
+   */
+  async chatBySlug(assistantSlug, _params, headers) {
+    let zodSchema = void 0;
+    let params = { ..._params };
+    if (params.output_schema && params.output_schema instanceof import_zod2.z.ZodType) {
+      zodSchema = params.output_schema;
+      params.output_schema = (0, import_zod_to_json_schema.zodToJsonSchema)(zodSchema, { definitions: {} });
+    }
+    params = AssistantChatParamsSchema.parse(params);
+    const response = await this.api.post(
+      `/v1/assistants/slug/${assistantSlug}/model`,
+      params,
+      {
+        responseType: params.stream ? "stream" : void 0
+      },
+      headers
     );
     const mapped = AssistantMapper.mapBaseModelApiResponse(response);
     if (!params.stream && zodSchema && mapped.generated) {
@@ -1158,7 +1194,9 @@ var WorkflowExecutionListParamsSchema = import_zod5.z.object({
   per_page: import_zod5.z.number().default(10)
 }).readonly();
 var WorkflowExecutionCreateParamsSchema = import_zod5.z.object({
-  user_input: import_zod5.z.string().default("")
+  user_input: import_zod5.z.union([import_zod5.z.string(), import_zod5.z.record(import_zod5.z.unknown()), import_zod5.z.array(import_zod5.z.unknown()), import_zod5.z.number(), import_zod5.z.boolean()]).optional(),
+  file_name: import_zod5.z.string().optional(),
+  propagate_headers: import_zod5.z.boolean().optional()
 }).readonly();
 var WorkflowExecutionStateListParamsSchema = import_zod5.z.object({
   page: import_zod5.z.number().default(0),
@@ -1221,9 +1259,13 @@ var WorkflowExecutionService = class {
   /**
    * Create a new workflow execution.
    */
-  async create(userInput = "") {
-    const params = WorkflowExecutionCreateParamsSchema.parse({ user_input: userInput });
-    return this.api.post(`/v1/workflows/${this.workflowId}/executions`, params);
+  async create(userInput, fileName, propagateHeaders, headers) {
+    const params = WorkflowExecutionCreateParamsSchema.parse({
+      user_input: userInput,
+      file_name: fileName,
+      propagate_headers: propagateHeaders
+    });
+    return this.api.post(`/v1/workflows/${this.workflowId}/executions`, params, {}, headers);
   }
   /**
    * Get workflow execution by ID.
@@ -1255,8 +1297,9 @@ var WorkflowExecutionService = class {
   /**
    * Resume an interrupted workflow execution.
    */
-  async resume(executionId) {
-    return this.api.put(`/v1/workflows/${this.workflowId}/executions/${executionId}/resume`);
+  async resume(executionId, propagateHeaders, headers) {
+    const config = { params: { propagate_headers: propagateHeaders } };
+    return this.api.put(`/v1/workflows/${this.workflowId}/executions/${executionId}/resume`, {}, config, headers);
   }
 };
 
@@ -1319,8 +1362,8 @@ var WorkflowService = class {
   /**
    * Run a workflow by ID.
    */
-  async run(workflowId, userInput = "") {
-    return this.executions(workflowId).create(userInput);
+  async run(workflowId, userInput, fileName, propagateHeaders, headers) {
+    return this.executions(workflowId).create(userInput, fileName, propagateHeaders, headers);
   }
   /**
    * Get workflow execution service for the specified workflow.