@superatomai/sdk-web 0.0.29 → 0.0.30

package/dist/index.cjs

@@ -1000,6 +1000,69 @@ var MenusResponseMessageSchema = zod.z.object({
   to: MessageParticipantSchema.optional(),
   payload: MenusResponsePayloadSchema
 });
+var SchemaColumnStatisticsSchema = zod.z.object({
+  distinct: zod.z.number().optional(),
+  min: zod.z.number().optional(),
+  max: zod.z.number().optional()
+});
+var SchemaColumnSchema = zod.z.object({
+  name: zod.z.string(),
+  type: zod.z.string(),
+  nativeType: zod.z.string(),
+  nullable: zod.z.boolean(),
+  description: zod.z.string(),
+  statistics: SchemaColumnStatisticsSchema.optional(),
+  cardinality: zod.z.enum(["unique", "high", "medium", "low"]).optional(),
+  sampleValues: zod.z.array(zod.z.string()).optional()
+});
+var SchemaTableSchema = zod.z.object({
+  name: zod.z.string(),
+  fullName: zod.z.string(),
+  rowCount: zod.z.number(),
+  description: zod.z.string(),
+  columns: zod.z.array(SchemaColumnSchema)
+});
+var SchemaRelationshipSchema = zod.z.object({
+  from: zod.z.string(),
+  to: zod.z.string(),
+  type: zod.z.string(),
+  keys: zod.z.array(zod.z.string())
+});
+var DatabaseSchemaSchema = zod.z.object({
+  database: zod.z.string(),
+  databaseType: zod.z.string().optional(),
+  schema: zod.z.string(),
+  description: zod.z.string(),
+  tables: zod.z.array(SchemaTableSchema),
+  relationships: zod.z.array(SchemaRelationshipSchema).optional()
+});
+var SchemaRequestPayloadSchema = zod.z.object({
+  /** If true, returns the formatted documentation string in addition to raw JSON */
+  formatted: zod.z.boolean().optional()
+});
+var SchemaRequestMessageSchema = zod.z.object({
+  id: zod.z.string(),
+  type: zod.z.literal("SCHEMA_REQ"),
+  from: MessageParticipantSchema,
+  to: MessageParticipantSchema.optional(),
+  payload: SchemaRequestPayloadSchema
+});
+var SchemaResponsePayloadSchema = zod.z.object({
+  success: zod.z.boolean(),
+  error: zod.z.string().optional(),
+  data: zod.z.object({
+    schema: DatabaseSchemaSchema,
+    /** Formatted schema documentation (only if formatted: true was requested) */
+    formatted: zod.z.string().optional()
+  }).optional()
+});
+zod.z.object({
+  id: zod.z.string(),
+  type: zod.z.literal("SCHEMA_RES"),
+  from: MessageParticipantSchema,
+  to: MessageParticipantSchema.optional(),
+  payload: SchemaResponsePayloadSchema
+});
 var ClientConfigSchema = zod.z.object({
   userId: zod.z.string(),
   projectId: zod.z.string(),
@@ -1062,6 +1125,7 @@ __export(services_exports, {
   getMenusHierarchy: () => getMenusHierarchy,
   getReport: () => getReport,
   getRootMenus: () => getRootMenus,
+  getSchema: () => getSchema,
   getUI: () => getUI,
   getUser: () => getUser,
   queryArtifacts: () => queryArtifacts,
@@ -2874,6 +2938,27 @@ async function reorderMenus(client, items, timeout) {
   };
 }
 
+// src/services/schema.ts
+async function getSchema(client, options = {}) {
+  const messageId = `msg_${Date.now()}_${Math.random().toString(36).substring(7)}`;
+  const message = SchemaRequestMessageSchema.parse({
+    id: messageId,
+    type: "SCHEMA_REQ",
+    from: { type: client.type },
+    to: { type: "data-agent" },
+    payload: {
+      formatted: options.formatted ?? false
+    }
+  });
+  const response = await client.sendWithResponse(message, options.timeout);
+  const payload = response.payload;
+  return {
+    success: payload.success,
+    error: payload.error,
+    data: payload.data
+  };
+}
+
 // src/client.ts
 var SuperatomClient = class {
   constructor(config) {
@@ -3702,6 +3787,15 @@ var SuperatomClient = class {
     this.log("info", "Reordering", items.length, "menus");
     return reorderMenus(this, items, timeout);
   }
+  // ==================== Schema Methods ====================
+  /**
+   * Get the database schema
+   * Retrieves the schema.json containing table definitions, columns, and relationships
+   */
+  async getSchema(options) {
+    this.log("info", "Fetching database schema");
+    return getSchema(this, options);
+  }
   /**
    * Internal logging
    */