@superatomai/sdk-node 0.0.27 → 0.0.28

package/dist/index.mjs

@@ -2892,7 +2892,9 @@ If adaptation is not possible or would fundamentally change the component:
 var PromptLoader = class {
   constructor(config) {
     this.promptCache = /* @__PURE__ */ new Map();
+    this.databaseRulesCache = /* @__PURE__ */ new Map();
     this.isInitialized = false;
+    this.databaseType = "postgresql";
     logger.debug("Initializing PromptLoader...");
     this.promptsDir = config?.promptsDir || path3.join(process.cwd(), ".prompts");
     logger.debug(`Prompts directory set to: ${this.promptsDir}`);
@@ -3038,6 +3040,76 @@ var PromptLoader = class {
   getCacheSize() {
     return this.promptCache.size;
   }
+  /**
+   * Set the database type for SQL rules loading
+   * @param type - Database type ('postgresql' | 'mssql')
+   */
+  setDatabaseType(type) {
+    this.databaseType = type;
+    this.databaseRulesCache.clear();
+    logger.debug(`Database type set to: ${type}`);
+  }
+  /**
+   * Get current database type
+   * @returns Database type string
+   */
+  getDatabaseType() {
+    return this.databaseType;
+  }
+  /**
+   * Load database-specific SQL rules from file system
+   * Falls back to minimal default rules if file not found
+   * @returns Database rules as a string
+   */
+  async loadDatabaseRules() {
+    if (this.databaseRulesCache.has(this.databaseType)) {
+      logger.debug(`\u2713 Database rules for '${this.databaseType}' loaded from cache`);
+      return this.databaseRulesCache.get(this.databaseType);
+    }
+    const rulesPath = path3.join(this.promptsDir, "database-rules", `${this.databaseType}.md`);
+    try {
+      if (fs4.existsSync(rulesPath)) {
+        const rules = fs4.readFileSync(rulesPath, "utf-8");
+        this.databaseRulesCache.set(this.databaseType, rules);
+        logger.info(`\u2713 Loaded database rules for '${this.databaseType}' from ${rulesPath}`);
+        return rules;
+      }
+    } catch (error) {
+      logger.warn(`Could not load database rules for '${this.databaseType}' from file system: ${error}`);
+    }
+    const defaultRules = this.getDefaultDatabaseRules();
+    this.databaseRulesCache.set(this.databaseType, defaultRules);
+    logger.warn(`Using default database rules for '${this.databaseType}' (file not found at ${rulesPath})`);
+    return defaultRules;
+  }
+  /**
+   * Get default database rules as fallback
+   * @returns Minimal database rules
+   */
+  getDefaultDatabaseRules() {
+    if (this.databaseType === "mssql") {
+      return `**Database Type: Microsoft SQL Server**
+
+**SQL Query Rules:**
+- Use \`TOP N\` for row limiting (e.g., \`SELECT TOP 32 * FROM table\`)
+- Use \`1\` for true, \`0\` for false (no native boolean)
+- Use \`+\` or \`CONCAT()\` for string concatenation
+- Use \`GETDATE()\` for current timestamp
+- Use \`CAST()\` or \`CONVERT()\` for type casting
+- Use \`OUTPUT INSERTED.*\` instead of \`RETURNING\`
+- NULL values: Use \`NULL\` keyword without quotes`;
+    }
+    return `**Database Type: PostgreSQL**
+
+**SQL Query Rules:**
+- Use \`LIMIT N\` for row limiting (e.g., \`SELECT * FROM table LIMIT 32\`)
+- Use \`true\` / \`false\` for boolean values
+- Use \`||\` for string concatenation
+- Use \`NOW()\` for current timestamp
+- Use \`::TYPE\` or \`CAST()\` for type casting
+- Use \`RETURNING\` clause for mutations
+- NULL values: Use \`NULL\` keyword without quotes`;
+  }
 };
 var defaultPromptsPath = process.env.PROMPTS_DIR || path3.join(process.cwd(), ".prompts");
 var promptLoader = new PromptLoader({
@@ -4196,6 +4268,7 @@ ${JSON.stringify(tool.requiredFields || [], null, 2)}`;
         }).join("\n\n");
       }
       const schemaDoc = schema.generateSchemaDocumentation();
+      const databaseRules = await promptLoader.loadDatabaseRules();
       logger.file("\n=============================\nText analysis response:", analysisContent);
       logger.file("\n=============================\nDeferred tools:", deferredToolsText);
       logger.file("\n=============================\nExecuted tools:", executedToolsText);
@@ -4203,6 +4276,7 @@ ${JSON.stringify(tool.requiredFields || [], null, 2)}`;
         ANALYSIS_CONTENT: analysisContent,
         AVAILABLE_COMPONENTS: availableComponentsText,
         SCHEMA_DOC: schemaDoc,
+        DATABASE_RULES: databaseRules,
         DEFERRED_TOOLS: deferredToolsText,
         EXECUTED_TOOLS: executedToolsText
       });
@@ -4463,12 +4537,14 @@ ${JSON.stringify(tool.requiredFields || [], null, 2)}`;
         };
       }
       const schemaDoc = schema.generateSchemaDocumentation();
+      const databaseRules = await promptLoader.loadDatabaseRules();
       const prompts = await promptLoader.loadPrompts("adapt-ui-block-params", {
         ORIGINAL_USER_PROMPT: originalUserPrompt,
         CURRENT_USER_PROMPT: currentUserPrompt,
         MATCHED_UI_BLOCK_COMPONENT: JSON.stringify(component, null, 2),
         COMPONENT_PROPS: JSON.stringify(component.props, null, 2),
-        SCHEMA_DOC: schemaDoc || "No schema available"
+        SCHEMA_DOC: schemaDoc || "No schema available",
+        DATABASE_RULES: databaseRules
       });
       const result = await LLM.stream(
         {
@@ -4587,6 +4663,7 @@ ${JSON.stringify(tool.requiredFields || [], null, 2)}`;
         availableToolsDoc = toolsDocParts.join("\n\n---\n\n");
       }
       const schemaDoc = schema.generateSchemaDocumentation();
+      const databaseRules = await promptLoader.loadDatabaseRules();
       const knowledgeBaseContext = await knowledge_base_default.getKnowledgeBase({
         prompt: userPrompt,
         collections,
@@ -4597,6 +4674,7 @@ ${JSON.stringify(tool.requiredFields || [], null, 2)}`;
         USER_PROMPT: userPrompt,
         CONVERSATION_HISTORY: conversationHistory || "No previous conversation",
         SCHEMA_DOC: schemaDoc,
+        DATABASE_RULES: databaseRules,
         KNOWLEDGE_BASE_CONTEXT: knowledgeBaseContext || "No additional knowledge base context available.",
         AVAILABLE_EXTERNAL_TOOLS: availableToolsDoc
       });
@@ -9555,7 +9633,8 @@ var SuperatomSDK = class {
     this.geminiApiKey = config.GEMINI_API_KEY || process.env.GEMINI_API_KEY || "";
     this.openaiApiKey = config.OPENAI_API_KEY || process.env.OPENAI_API_KEY || "";
     this.llmProviders = config.LLM_PROVIDERS || getLLMProviders();
-    logger.info(`Initializing Superatom SDK v${SDK_VERSION} for project ${this.projectId}, llm providers: ${this.llmProviders.join(", ")}, config llm providers: ${config.LLM_PROVIDERS}`);
+    this.databaseType = config.databaseType || "postgresql";
+    logger.info(`Initializing Superatom SDK v${SDK_VERSION} for project ${this.projectId}, llm providers: ${this.llmProviders.join(", ")}, database type: ${this.databaseType}`);
     this.userManager = new UserManager(this.projectId, 5e3);
     this.dashboardManager = new DashboardManager(this.projectId);
     this.reportManager = new ReportManager(this.projectId);
@@ -9577,8 +9656,9 @@ var SuperatomSDK = class {
       if (promptsDir) {
         promptLoader.setPromptsDir(promptsDir);
       }
+      promptLoader.setDatabaseType(this.databaseType);
       await promptLoader.initialize();
-      logger.info(`PromptLoader initialized with ${promptLoader.getCacheSize()} prompts from ${promptLoader.getPromptsDir()}`);
+      logger.info(`PromptLoader initialized with ${promptLoader.getCacheSize()} prompts from ${promptLoader.getPromptsDir()}, database type: ${this.databaseType}`);
     } catch (error) {
       logger.error("Failed to initialize PromptLoader:", error);
       throw error;