@superatomai/sdk-node 0.0.27 → 0.0.28

package/dist/index.d.mts

@@ -587,6 +587,7 @@ type CollectionOperation = 'getMany' | 'getOne' | 'query' | 'mutation' | 'update
 type CollectionHandler<TParams = any, TResult = any> = (params: TParams) => Promise<TResult> | TResult;
 type LLMProvider = 'anthropic' | 'groq' | 'gemini' | 'openai';
 
+type DatabaseType = 'postgresql' | 'mssql';
 interface SuperatomSDKConfig {
     url?: string;
     apiKey?: string;
@@ -595,6 +596,7 @@ interface SuperatomSDKConfig {
     type?: string;
     bundleDir?: string;
     promptsDir?: string;
+    databaseType?: DatabaseType;
     ANTHROPIC_API_KEY?: string;
     GROQ_API_KEY?: string;
     GEMINI_API_KEY?: string;
@@ -1644,6 +1646,7 @@ declare class SuperatomSDK {
     private geminiApiKey;
     private openaiApiKey;
     private llmProviders;
+    private databaseType;
     private userManager;
     private dashboardManager;
     private reportManager;
@@ -1725,4 +1728,4 @@ declare class SuperatomSDK {
     getTools(): Tool$1[];
 }
 
-export { type Action, BM25L, type BM25LOptions, CONTEXT_CONFIG, type CapturedLog, CleanupService, type CollectionHandler, type CollectionOperation, type DBUIBlock, type HybridSearchOptions, type IncomingMessage, type KbNodesQueryFilters, type KbNodesRequestPayload, LLM, type LogLevel, type Message, type RerankedResult, SDK_VERSION, STORAGE_CONFIG, SuperatomSDK, type SuperatomSDKConfig, Thread, ThreadManager, type Tool$1 as Tool, UIBlock, UILogCollector, type User, UserManager, type UsersData, hybridRerank, logger, rerankChromaResults, rerankConversationResults };
+export { type Action, BM25L, type BM25LOptions, CONTEXT_CONFIG, type CapturedLog, CleanupService, type CollectionHandler, type CollectionOperation, type DBUIBlock, type DatabaseType, type HybridSearchOptions, type IncomingMessage, type KbNodesQueryFilters, type KbNodesRequestPayload, LLM, type LogLevel, type Message, type RerankedResult, SDK_VERSION, STORAGE_CONFIG, SuperatomSDK, type SuperatomSDKConfig, Thread, ThreadManager, type Tool$1 as Tool, UIBlock, UILogCollector, type User, UserManager, type UsersData, hybridRerank, logger, rerankChromaResults, rerankConversationResults };

package/dist/index.d.ts

@@ -587,6 +587,7 @@ type CollectionOperation = 'getMany' | 'getOne' | 'query' | 'mutation' | 'update
 type CollectionHandler<TParams = any, TResult = any> = (params: TParams) => Promise<TResult> | TResult;
 type LLMProvider = 'anthropic' | 'groq' | 'gemini' | 'openai';
 
+type DatabaseType = 'postgresql' | 'mssql';
 interface SuperatomSDKConfig {
     url?: string;
     apiKey?: string;
@@ -595,6 +596,7 @@ interface SuperatomSDKConfig {
     type?: string;
     bundleDir?: string;
     promptsDir?: string;
+    databaseType?: DatabaseType;
     ANTHROPIC_API_KEY?: string;
     GROQ_API_KEY?: string;
     GEMINI_API_KEY?: string;
@@ -1644,6 +1646,7 @@ declare class SuperatomSDK {
     private geminiApiKey;
     private openaiApiKey;
     private llmProviders;
+    private databaseType;
     private userManager;
     private dashboardManager;
     private reportManager;
@@ -1725,4 +1728,4 @@ declare class SuperatomSDK {
     getTools(): Tool$1[];
 }
 
-export { type Action, BM25L, type BM25LOptions, CONTEXT_CONFIG, type CapturedLog, CleanupService, type CollectionHandler, type CollectionOperation, type DBUIBlock, type HybridSearchOptions, type IncomingMessage, type KbNodesQueryFilters, type KbNodesRequestPayload, LLM, type LogLevel, type Message, type RerankedResult, SDK_VERSION, STORAGE_CONFIG, SuperatomSDK, type SuperatomSDKConfig, Thread, ThreadManager, type Tool$1 as Tool, UIBlock, UILogCollector, type User, UserManager, type UsersData, hybridRerank, logger, rerankChromaResults, rerankConversationResults };
+export { type Action, BM25L, type BM25LOptions, CONTEXT_CONFIG, type CapturedLog, CleanupService, type CollectionHandler, type CollectionOperation, type DBUIBlock, type DatabaseType, type HybridSearchOptions, type IncomingMessage, type KbNodesQueryFilters, type KbNodesRequestPayload, LLM, type LogLevel, type Message, type RerankedResult, SDK_VERSION, STORAGE_CONFIG, SuperatomSDK, type SuperatomSDKConfig, Thread, ThreadManager, type Tool$1 as Tool, UIBlock, UILogCollector, type User, UserManager, type UsersData, hybridRerank, logger, rerankChromaResults, rerankConversationResults };

package/dist/index.js

@@ -2936,7 +2936,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 || import_path2.default.join(process.cwd(), ".prompts");
     logger.debug(`Prompts directory set to: ${this.promptsDir}`);
@@ -3082,6 +3084,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 = import_path2.default.join(this.promptsDir, "database-rules", `${this.databaseType}.md`);
+    try {
+      if (import_fs3.default.existsSync(rulesPath)) {
+        const rules = import_fs3.default.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 || import_path2.default.join(process.cwd(), ".prompts");
 var promptLoader = new PromptLoader({
@@ -4240,6 +4312,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);
@@ -4247,6 +4320,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
       });
@@ -4507,12 +4581,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(
         {
@@ -4631,6 +4707,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,
@@ -4641,6 +4718,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
       });
@@ -9599,7 +9677,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);
@@ -9621,8 +9700,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;