@superblocksteam/sdk-api 2.0.96 → 2.0.97-next.0

package/src/integrations/mongodb/README.md

@@ -4,9 +4,9 @@ Execute MongoDB operations with full type safety and runtime validation.
 
 ## Methods
 
-| Method                                    | Description                 |
-| ----------------------------------------- | --------------------------- |
-| `run(collection, action, schema, params)` | Execute a MongoDB operation |
+| Method                                               | Description                 |
+| ---------------------------------------------------- | --------------------------- |
+| `run(collection, action, schema, params, metadata?)` | Execute a MongoDB operation |
 
 ## Actions
 
@@ -215,6 +215,10 @@ const users = await ctx.integrations.mongodb.run("users", "find", UserSchema, {
 | `skip`       | `number` | Documents to skip            |
 | `field`      | `string` | Field for distinct operation |
 
+## Trace Metadata
+
+All methods accept an optional `metadata` parameter as the last argument for diagnostics labeling. See the [root SDK README](../../../README.md#trace-metadata) for details.
+
 ## Error Handling
 
 ```typescript

package/src/integrations/mongodb/client.ts

@@ -8,7 +8,7 @@ import type { z } from "zod";
 import type { PartialMessage } from "@bufbuild/protobuf";
 import type { Plugin as MongoDBPlugin } from "@superblocksteam/types/dist/src/plugins/mongodb/v1/plugin_pb";
 import type { IntegrationConfig, IntegrationClientImpl } from "../types.js";
-import type { QueryExecutor } from "../registry.js";
+import type { QueryExecutor, TraceMetadata } from "../registry.js";
 import { RestApiValidationError } from "../../errors.js";
 import { IntegrationError } from "../../runtime/errors.js";
 import type { MongoDBClient, MongoDBAction, MongoDBParams } from "./types.js";
@@ -98,12 +98,15 @@ export class MongoDBClientImpl implements MongoDBClient, IntegrationClientImpl {
     action: MongoDBAction,
     schema: z.ZodSchema<T>,
     params?: MongoDBParams,
+    metadata?: TraceMetadata,
   ): Promise<T> {
     const request = this.buildRequest(collection, action, params);
 
     try {
       const result = await this.executeQuery(
         request as Record<string, unknown>,
+        undefined,
+        metadata,
       );
 
       // Validate result against schema

package/src/integrations/mongodb/types.ts

@@ -4,6 +4,7 @@
 
 import type { z } from "zod";
 import type { BaseIntegrationClient } from "../../types.js";
+import type { TraceMetadata } from "../registry.js";
 
 /**
  * MongoDB operation types.
@@ -99,5 +100,6 @@ export interface MongoDBClient extends BaseIntegrationClient {
     action: MongoDBAction,
     schema: z.ZodSchema<T>,
     params?: MongoDBParams,
+    metadata?: TraceMetadata,
   ): Promise<T>;
 }

package/src/integrations/mssql/README.md

@@ -4,10 +4,10 @@ Execute SQL queries and statements against Microsoft SQL Server databases with f
 
 ## Methods
 
-| Method                           | Description                                                                |
-| -------------------------------- | -------------------------------------------------------------------------- |
-| `query<T>(sql, schema, params?)` | Execute a SELECT query and return validated, typed results                 |
-| `execute(sql, params?)`          | Execute a statement (INSERT, UPDATE, DELETE) and return affected row count |
+| Method                                      | Description                                                                |
+| ------------------------------------------- | -------------------------------------------------------------------------- |
+| `query<T>(sql, schema, params?, metadata?)` | Execute a SELECT query and return validated, typed results                 |
+| `execute(sql, params?, metadata?)`          | Execute a statement (INSERT, UPDATE, DELETE) and return affected row count |
 
 ## Usage
 
@@ -94,6 +94,10 @@ const analytics = await ctx.integrations.mssql.query(
 );
 ```
 
+## Trace Metadata
+
+All methods accept an optional `metadata` parameter as the last argument for diagnostics labeling. See the [root SDK README](../../../README.md#trace-metadata) for details.
+
 ## Common Pitfalls
 
 ### BIGINT Values Returned as Strings

package/src/integrations/mssql/client.ts

@@ -9,7 +9,7 @@ import type { z } from "zod";
 import type { PartialMessage } from "@bufbuild/protobuf";
 import type { Plugin as MSSQLPlugin } from "@superblocksteam/types/dist/src/plugins/mssql/v1/plugin_pb";
 import type { IntegrationConfig, IntegrationClientImpl } from "../types.js";
-import type { QueryExecutor } from "../registry.js";
+import type { QueryExecutor, TraceMetadata } from "../registry.js";
 import { QueryValidationError } from "../../errors.js";
 import { IntegrationError } from "../../runtime/errors.js";
 import { describeType } from "../utils.js";
@@ -62,11 +62,12 @@ export class MSSQLClientImpl implements MSSQLClient, IntegrationClientImpl {
     sql: string,
     schema: z.ZodSchema<T>,
     params?: unknown[],
+    metadata?: TraceMetadata,
   ): Promise<T[]> {
     const request = this.buildRequest(sql, params);
 
     try {
-      const result = await this.executeQuery(request);
+      const result = await this.executeQuery(request, undefined, metadata);
 
       if (!Array.isArray(result)) {
         throw new IntegrationError(
@@ -114,11 +115,12 @@ export class MSSQLClientImpl implements MSSQLClient, IntegrationClientImpl {
   async execute(
     sql: string,
     params?: unknown[],
+    metadata?: TraceMetadata,
   ): Promise<{ rowCount: number }> {
     const request = this.buildRequest(sql, params);
 
     try {
-      const result = await this.executeQuery(request);
+      const result = await this.executeQuery(request, undefined, metadata);
 
       if (
         typeof result === "object" &&

package/src/integrations/mssql/types.ts

@@ -6,6 +6,7 @@
 
 import type { z } from "zod";
 import type { BaseIntegrationClient } from "../../types.js";
+import type { TraceMetadata } from "../registry.js";
 
 /**
  * Client for executing MSSQL queries with type-safe validation.
@@ -45,6 +46,7 @@ export interface MSSQLClient extends BaseIntegrationClient {
     sql: string,
     schema: z.ZodSchema<T>,
     params?: unknown[],
+    metadata?: TraceMetadata,
   ): Promise<T[]>;
 
   /**
@@ -54,5 +56,9 @@ export interface MSSQLClient extends BaseIntegrationClient {
    * @param params - Optional parameters for prepared statements
    * @returns Promise resolving to execution result with row count
    */
-  execute(sql: string, params?: unknown[]): Promise<{ rowCount: number }>;
+  execute(
+    sql: string,
+    params?: unknown[],
+    metadata?: TraceMetadata,
+  ): Promise<{ rowCount: number }>;
 }

package/src/integrations/mysql/README.md

@@ -4,10 +4,10 @@ Execute SQL queries and statements against MySQL databases with full type safety
 
 ## Methods
 
-| Method                           | Description                                                                |
-| -------------------------------- | -------------------------------------------------------------------------- |
-| `query<T>(sql, schema, params?)` | Execute a SELECT query and return validated, typed results                 |
-| `execute(sql, params?)`          | Execute a statement (INSERT, UPDATE, DELETE) and return affected row count |
+| Method                                      | Description                                                                |
+| ------------------------------------------- | -------------------------------------------------------------------------- |
+| `query<T>(sql, schema, params?, metadata?)` | Execute a SELECT query and return validated, typed results                 |
+| `execute(sql, params?, metadata?)`          | Execute a statement (INSERT, UPDATE, DELETE) and return affected row count |
 
 ## Usage
 
@@ -96,6 +96,10 @@ const orders = await ctx.integrations.mysql.query(
 );
 ```
 
+## Trace Metadata
+
+All methods accept an optional `metadata` parameter as the last argument for diagnostics labeling. See the [root SDK README](../../../README.md#trace-metadata) for details.
+
 ## Common Pitfalls
 
 ### BIGINT Values Returned as Strings

package/src/integrations/mysql/client.ts

@@ -9,7 +9,7 @@ import type { z } from "zod";
 import type { PartialMessage } from "@bufbuild/protobuf";
 import type { Plugin as MySQLPlugin } from "@superblocksteam/types/dist/src/plugins/mysql/v1/plugin_pb";
 import type { IntegrationConfig, IntegrationClientImpl } from "../types.js";
-import type { QueryExecutor } from "../registry.js";
+import type { QueryExecutor, TraceMetadata } from "../registry.js";
 import { QueryValidationError } from "../../errors.js";
 import { IntegrationError } from "../../runtime/errors.js";
 import { describeType } from "../utils.js";
@@ -62,11 +62,12 @@ export class MySQLClientImpl implements MySQLClient, IntegrationClientImpl {
     sql: string,
     schema: z.ZodSchema<T>,
     params?: unknown[],
+    metadata?: TraceMetadata,
   ): Promise<T[]> {
     const request = this.buildRequest(sql, params);
 
     try {
-      const result = await this.executeQuery(request);
+      const result = await this.executeQuery(request, undefined, metadata);
 
       if (!Array.isArray(result)) {
         throw new IntegrationError(
@@ -114,11 +115,12 @@ export class MySQLClientImpl implements MySQLClient, IntegrationClientImpl {
   async execute(
     sql: string,
     params?: unknown[],
+    metadata?: TraceMetadata,
   ): Promise<{ rowCount: number }> {
     const request = this.buildRequest(sql, params);
 
     try {
-      const result = await this.executeQuery(request);
+      const result = await this.executeQuery(request, undefined, metadata);
 
       if (
         typeof result === "object" &&

package/src/integrations/mysql/types.ts

@@ -6,6 +6,7 @@
 
 import type { z } from "zod";
 import type { BaseIntegrationClient } from "../../types.js";
+import type { TraceMetadata } from "../registry.js";
 
 /**
  * Client for executing MySQL queries with type-safe validation.
@@ -60,6 +61,7 @@ export interface MySQLClient extends BaseIntegrationClient {
     sql: string,
     schema: z.ZodSchema<T>,
     params?: unknown[],
+    metadata?: TraceMetadata,
   ): Promise<T[]>;
 
   /**
@@ -78,5 +80,9 @@ export interface MySQLClient extends BaseIntegrationClient {
    * console.log(`Updated ${result.rowCount} rows`);
    * ```
    */
-  execute(sql: string, params?: unknown[]): Promise<{ rowCount: number }>;
+  execute(
+    sql: string,
+    params?: unknown[],
+    metadata?: TraceMetadata,
+  ): Promise<{ rowCount: number }>;
 }

package/src/integrations/notion/README.md

@@ -4,9 +4,9 @@ Query databases, create pages, and manage content in Notion workspaces.
 
 ## Methods
 
-| Method                        | Description                                        |
-| ----------------------------- | -------------------------------------------------- |
-| `apiRequest(options, schema)` | Make any Notion API request with schema validation |
+| Method                                   | Description                                        |
+| ---------------------------------------- | -------------------------------------------------- |
+| `apiRequest(options, schema, metadata?)` | Make any Notion API request with schema validation |
 
 ## Usage
 
@@ -250,6 +250,10 @@ const result = await ctx.integrations.notion.apiRequest(
 console.log("Properties:", Object.keys(result.properties));
 ```
 
+## Trace Metadata
+
+All methods accept an optional `metadata` parameter as the last argument for diagnostics labeling. See the [root SDK README](../../../README.md#trace-metadata) for details.
+
 ## Common Pitfalls
 
 ### No Specialized Methods

package/src/integrations/openai_v2/README.md

@@ -4,10 +4,10 @@ Interact with OpenAI's APIs for chat completions, embeddings, image generation,
 
 ## Methods
 
-| Method                              | Description                                        |
-| ----------------------------------- | -------------------------------------------------- |
-| `apiRequest(options, schema)`       | Make any OpenAI API request with schema validation |
-| `streamApiRequest(options, schema)` | Stream responses with real-time chunk validation   |
+| Method                                         | Description                                        |
+| ---------------------------------------------- | -------------------------------------------------- |
+| `apiRequest(options, schema, metadata?)`       | Make any OpenAI API request with schema validation |
+| `streamApiRequest(options, schema, metadata?)` | Stream responses with real-time chunk validation   |
 
 ## Usage
 
@@ -311,6 +311,10 @@ OpenAI's streaming chunks have this structure:
 }
 ```
 
+## Trace Metadata
+
+All methods accept an optional `metadata` parameter as the last argument for diagnostics labeling. See the [root SDK README](../../../README.md#trace-metadata) for details.
+
 ## Common Pitfalls
 
 ### No Specialized Methods

package/src/integrations/oracledb/README.md

@@ -4,10 +4,10 @@ Execute SQL queries and statements against Oracle databases with full type safet
 
 ## Methods
 
-| Method                           | Description                                                                |
-| -------------------------------- | -------------------------------------------------------------------------- |
-| `query<T>(sql, schema, params?)` | Execute a SELECT query and return validated, typed results                 |
-| `execute(sql, params?)`          | Execute a statement (INSERT, UPDATE, DELETE) and return affected row count |
+| Method                                      | Description                                                                |
+| ------------------------------------------- | -------------------------------------------------------------------------- |
+| `query<T>(sql, schema, params?, metadata?)` | Execute a SELECT query and return validated, typed results                 |
+| `execute(sql, params?, metadata?)`          | Execute a statement (INSERT, UPDATE, DELETE) and return affected row count |
 
 ## Usage
 
@@ -94,6 +94,10 @@ const analytics = await ctx.integrations.oracledb.query(
 );
 ```
 
+## Trace Metadata
+
+All methods accept an optional `metadata` parameter as the last argument for diagnostics labeling. See the [root SDK README](../../../README.md#trace-metadata) for details.
+
 ## Common Pitfalls
 
 ### Uppercase Column Names

package/src/integrations/oracledb/client.ts

@@ -9,7 +9,7 @@ import type { z } from "zod";
 import type { PartialMessage } from "@bufbuild/protobuf";
 import type { Plugin as OracleDBPlugin } from "@superblocksteam/types/dist/src/plugins/oracledb/v1/plugin_pb";
 import type { IntegrationConfig, IntegrationClientImpl } from "../types.js";
-import type { QueryExecutor } from "../registry.js";
+import type { QueryExecutor, TraceMetadata } from "../registry.js";
 import { QueryValidationError } from "../../errors.js";
 import { IntegrationError } from "../../runtime/errors.js";
 import { describeType } from "../utils.js";
@@ -66,11 +66,12 @@ export class OracleDBClientImpl
     sql: string,
     schema: z.ZodSchema<T>,
     params?: unknown[],
+    metadata?: TraceMetadata,
   ): Promise<T[]> {
     const request = this.buildRequest(sql, params);
 
     try {
-      const result = await this.executeQuery(request);
+      const result = await this.executeQuery(request, undefined, metadata);
 
       if (!Array.isArray(result)) {
         throw new IntegrationError(
@@ -118,11 +119,12 @@ export class OracleDBClientImpl
   async execute(
     sql: string,
     params?: unknown[],
+    metadata?: TraceMetadata,
   ): Promise<{ rowCount: number }> {
     const request = this.buildRequest(sql, params);
 
     try {
-      const result = await this.executeQuery(request);
+      const result = await this.executeQuery(request, undefined, metadata);
 
       if (
         typeof result === "object" &&

package/src/integrations/oracledb/types.ts

@@ -6,6 +6,7 @@
 
 import type { z } from "zod";
 import type { BaseIntegrationClient } from "../../types.js";
+import type { TraceMetadata } from "../registry.js";
 
 /**
  * Client for executing OracleDB queries with type-safe validation.
@@ -45,6 +46,7 @@ export interface OracleDBClient extends BaseIntegrationClient {
     sql: string,
     schema: z.ZodSchema<T>,
     params?: unknown[],
+    metadata?: TraceMetadata,
   ): Promise<T[]>;
 
   /**
@@ -54,5 +56,9 @@ export interface OracleDBClient extends BaseIntegrationClient {
    * @param params - Optional parameters for prepared statements
    * @returns Promise resolving to execution result with row count
    */
-  execute(sql: string, params?: unknown[]): Promise<{ rowCount: number }>;
+  execute(
+    sql: string,
+    params?: unknown[],
+    metadata?: TraceMetadata,
+  ): Promise<{ rowCount: number }>;
 }

package/src/integrations/pagerduty/README.md

@@ -4,9 +4,9 @@ Create incidents, manage services, and interact with PagerDuty's incident manage
 
 ## Methods
 
-| Method                        | Description                                           |
-| ----------------------------- | ----------------------------------------------------- |
-| `apiRequest(options, schema)` | Make any PagerDuty API request with schema validation |
+| Method                                   | Description                                           |
+| ---------------------------------------- | ----------------------------------------------------- |
+| `apiRequest(options, schema, metadata?)` | Make any PagerDuty API request with schema validation |
 
 ## Usage
 
@@ -315,6 +315,10 @@ await ctx.integrations.pagerduty.apiRequest(
 );
 ```
 
+## Trace Metadata
+
+All methods accept an optional `metadata` parameter as the last argument for diagnostics labeling. See the [root SDK README](../../../README.md#trace-metadata) for details.
+
 ## Common Pitfalls
 
 ### No Specialized Methods

package/src/integrations/perplexity/README.md

@@ -4,9 +4,9 @@ Interact with Perplexity's search-augmented AI for real-time information retriev
 
 ## Methods
 
-| Method                        | Description                                            |
-| ----------------------------- | ------------------------------------------------------ |
-| `apiRequest(options, schema)` | Make any Perplexity API request with schema validation |
+| Method                                   | Description                                            |
+| ---------------------------------------- | ------------------------------------------------------ |
+| `apiRequest(options, schema, metadata?)` | Make any Perplexity API request with schema validation |
 
 ## Usage
 
@@ -183,6 +183,10 @@ const result = await ctx.integrations.perplexity.apiRequest(
 );
 ```
 
+## Trace Metadata
+
+All methods accept an optional `metadata` parameter as the last argument for diagnostics labeling. See the [root SDK README](../../../README.md#trace-metadata) for details.
+
 ## Common Pitfalls
 
 ### No Specialized Methods

package/src/integrations/postgres/README.md

@@ -4,10 +4,10 @@ Execute SQL queries and statements against PostgreSQL databases with full type s
 
 ## Methods
 
-| Method                           | Description                                                                |
-| -------------------------------- | -------------------------------------------------------------------------- |
-| `query<T>(sql, schema, params?)` | Execute a SELECT query and return validated, typed results                 |
-| `execute(sql, params?)`          | Execute a statement (INSERT, UPDATE, DELETE) and return affected row count |
+| Method                                      | Description                                                                |
+| ------------------------------------------- | -------------------------------------------------------------------------- |
+| `query<T>(sql, schema, params?, metadata?)` | Execute a SELECT query and return validated, typed results                 |
+| `execute(sql, params?, metadata?)`          | Execute a statement (INSERT, UPDATE, DELETE) and return affected row count |
 
 ## Usage
 
@@ -173,6 +173,30 @@ const users = await ctx.integrations.db.query(
 );
 ```
 
+## Trace Metadata
+
+All methods accept an optional `metadata` parameter as the last argument for diagnostics labeling:
+
+```typescript
+const users = await ctx.integrations.db.query(
+  "SELECT * FROM users WHERE status = $1",
+  UserSchema,
+  ["active"],
+  {
+    label: "Fetch active users",
+    description: "Paginated user list for dashboard",
+  },
+);
+
+await ctx.integrations.db.execute(
+  "DELETE FROM sessions WHERE expires_at < NOW()",
+  undefined, // no params
+  { label: "Clean expired sessions" },
+);
+```
+
+When `includeDiagnostics` is enabled, `label` and `description` appear in the trace view. See the [root SDK README](../../../README.md#trace-metadata) for details.
+
 ## Common Pitfalls
 
 ### Aggregate Functions (COUNT, SUM) Return Strings

package/src/integrations/postgres/client.ts

@@ -9,7 +9,7 @@ import type { z } from "zod";
 import type { PartialMessage } from "@bufbuild/protobuf";
 import type { Plugin as PostgresPlugin } from "@superblocksteam/types/dist/src/plugins/postgresql/v1/plugin_pb";
 import type { IntegrationConfig, IntegrationClientImpl } from "../types.js";
-import type { QueryExecutor } from "../registry.js";
+import type { QueryExecutor, TraceMetadata } from "../registry.js";
 import type { PostgresClient } from "./types.js";
 import { QueryValidationError } from "../../errors.js";
 import { IntegrationError } from "../../runtime/errors.js";
@@ -68,11 +68,12 @@ export class PostgresClientImpl
     sql: string,
     schema: z.ZodSchema<T>,
     params?: unknown[],
+    metadata?: TraceMetadata,
   ): Promise<T[]> {
     const request = this.buildRequest(sql, params);
 
     try {
-      const result = await this.executeQuery(request);
+      const result = await this.executeQuery(request, undefined, metadata);
 
       if (!Array.isArray(result)) {
         throw new IntegrationError(
@@ -117,11 +118,12 @@ export class PostgresClientImpl
   async execute(
     sql: string,
     params?: unknown[],
+    metadata?: TraceMetadata,
   ): Promise<{ rowCount: number }> {
     const request = this.buildRequest(sql, params);
 
     try {
-      const result = await this.executeQuery(request);
+      const result = await this.executeQuery(request, undefined, metadata);
 
       if (
         typeof result === "object" &&

package/src/integrations/postgres/types.ts

@@ -4,6 +4,7 @@
 
 import type { z } from "zod";
 import type { BaseIntegrationClient } from "../../types.js";
+import type { TraceMetadata } from "../registry.js";
 
 /**
  * PostgreSQL client for database operations.
@@ -76,6 +77,7 @@ export interface PostgresClient extends BaseIntegrationClient {
     sql: string,
     schema: z.ZodSchema<T>,
     params?: unknown[],
+    metadata?: TraceMetadata,
   ): Promise<T[]>;
 
   /**
@@ -96,5 +98,9 @@ export interface PostgresClient extends BaseIntegrationClient {
    * console.log(`Deleted ${result.rowCount} expired sessions`);
    * ```
    */
-  execute(sql: string, params?: unknown[]): Promise<{ rowCount: number }>;
+  execute(
+    sql: string,
+    params?: unknown[],
+    metadata?: TraceMetadata,
+  ): Promise<{ rowCount: number }>;
 }

package/src/integrations/python/README.md

@@ -63,7 +63,7 @@ export default api({
 
 ## API Reference
 
-### `ctx.integrations.<name>.run<T>(code, schema, bindings?): Promise<T>`
+### `ctx.integrations.<name>.run<T>(code, schema, bindings?, metadata?): Promise<T>`
 
 Executes Python code with validated output. Use the **key name** you declared in `integrations` (e.g. `ctx.integrations.runtime.run(...)` if you declared `runtime: python(...)`).
 
@@ -71,7 +71,8 @@ Executes Python code with validated output. Use the **key name** you declared in
 
 - `code` (string): Python script to execute. Bindings are automatically injected as Python variables.
 - `schema` (ZodSchema<T>): **Required** Zod schema for validating the return value.
-- `bindings` (Record<string, unknown>, optional): Key/value pairs that become Python variables in your code.
+- `bindings` (Record<string, unknown>, optional): Key/value pairs that become Python variables in your code. Do not pass `TraceMetadata` here; use the `metadata` parameter instead.
+- `metadata` (TraceMetadata, optional): Trace metadata for diagnostics labeling (see [Trace Metadata](#trace-metadata)).
 
 **Returns:** Promise resolving to validated result matching the schema.
 
@@ -79,6 +80,16 @@ Executes Python code with validated output. Use the **key name** you declared in
 
 **Important:** Your Python code must use a `return` statement to return a value.
 
+> **Note:** Because `bindings` and `metadata` are both plain objects, take care not
+> to pass metadata as the third argument. If you need metadata but no bindings,
+> pass `undefined` for `bindings` explicitly:
+>
+> ```typescript
+> await ctx.integrations.runtime.run(code, schema, undefined, {
+>   label: "My trace label",
+> });
+> ```
+
 ## Binding Syntax
 
 Bindings are passed as a map and automatically injected as Python variables. Reference them directly in your code by their key names:
@@ -302,6 +313,10 @@ export default api({
 });
 ```
 
+## Trace Metadata
+
+All methods accept an optional `metadata` parameter as the last argument for diagnostics labeling. See the [root SDK README](../../../README.md#trace-metadata) for details.
+
 ## Error Handling
 
 The client throws `CodeExecutionError` when execution or validation fails:
@@ -546,6 +561,6 @@ For specific package requirements, consult your Superblocks environment document
 
 ## See Also
 
-- [API Reference](../../README.md) - Main SDK documentation
-- [Error Handling](../../README.md#error-handling) - Error types and handling strategies
-- [Best Practices](../../README.md#best-practices) - General SDK best practices
+- [API Reference](../../../README.md) - Main SDK documentation
+- [Error Handling](../../../README.md#error-handling) - Error types and handling strategies
+- [Best Practices](../../../README.md#best-practices) - General SDK best practices

package/src/integrations/python/client.ts

@@ -8,7 +8,7 @@ import type { z } from "zod";
 import type { PartialMessage } from "@bufbuild/protobuf";
 import type { Plugin as PythonPlugin } from "@superblocksteam/types/dist/src/plugins/python/v1/plugin_pb";
 import type { IntegrationConfig, IntegrationClientImpl } from "../types.js";
-import type { QueryExecutor } from "../registry.js";
+import type { QueryExecutor, TraceMetadata } from "../registry.js";
 import type { PythonClient } from "./types.js";
 import { CodeExecutionError } from "../../errors.js";
 
@@ -89,6 +89,7 @@ export class PythonClientImpl implements PythonClient, IntegrationClientImpl {
     code: string,
     schema: z.ZodSchema<T>,
     bindings?: Record<string, unknown>,
+    metadata?: TraceMetadata,
   ): Promise<T> {
     // Build request with bindings injected as Python variables
     const request = this.buildRequest(code, bindings);
@@ -98,6 +99,8 @@ export class PythonClientImpl implements PythonClient, IntegrationClientImpl {
       // No need to pass bindings as input since they're already in the code
       const result = await this.executeQuery(
         request as unknown as Record<string, unknown>,
+        undefined,
+        metadata,
       );
 
       // Validate result against schema