@superatomai/sdk-node 0.0.16-mds → 0.0.17-mds

package/dist/index.js

@@ -6902,7 +6902,7 @@ function extractEntityDetails(description) {
     });
     seenNames.add(name.toLowerCase());
   }
-  const catalogMatch = description.match(/(?:All \d+ tables|Other tables)[^:]*:\n?([\s\S]*?)(?:\n\n|$)/i);
+  const catalogMatch = description.match(/(?:All \d+ tables|Other tables|Remaining \d+ tables)[^:]*:\n?([\s\S]*?)(?:\n\n|$)/i);
   if (catalogMatch) {
     const catalogText = catalogMatch[1];
     const entryPattern = /([\w.]+)\(([\d.]+[KMB]?)\s*rows?\)/gi;
@@ -7225,13 +7225,21 @@ Analyze the error and try again with a corrected query.`;
     const sourceType = this.extractSourceType();
     const fullSchema = this.tool.fullSchema || this.tool.description || "No schema available";
     const hasSchemaSearch = !!this.tool.schemaSearchFn;
-    const schemaSearchInstructions = hasSchemaSearch ? `## Schema Search
-This source has a large schema. The schema above shows a catalog of table names only.
+    const schemaTier = this.tool.schemaTier || "";
+    let schemaSearchInstructions = "";
+    if (hasSchemaSearch && schemaTier === "very_large") {
+      schemaSearchInstructions = `## Schema Search
+This source has a very large schema. The schema above shows only table names \u2014 no column details.
 Before writing any query, you MUST use the search_schema tool to find exact table and column names.
 1. Search with keywords related to the requested data
-2. Review the returned column details carefully
+2. Review the returned column details and sample values carefully
 3. Then write your SQL query using the exact names from the search results
-You may search multiple times with different keywords if the first search doesn't find what you need.` : "";
+You may search multiple times with different keywords if the first search doesn't find what you need.`;
+    } else if (hasSchemaSearch) {
+      schemaSearchInstructions = `## Schema Search
+This source has a large schema. The top tables are shown in detail above, but not all tables are listed with columns.
+If you need data from a table that is only listed in the catalog (no columns shown), use the search_schema tool to look up its column details before writing a query.`;
+    }
     const prompts = await promptLoader.loadPrompts("agent-source-query", {
       SOURCE_NAME: sourceName,
       SOURCE_TYPE: sourceType,