@horribleprogram/sdk 0.1.2 → 0.1.4

package/README.md

@@ -1,97 +1,109 @@
-# @botcierge/sdk
-
-[![npm version](https://img.shields.io/npm/v/@botcierge/sdk.svg)](https://www.npmjs.com/package/@botcierge/sdk)
-[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
-
-Official JavaScript/TypeScript SDK for the **Botcierge Intent Classification API**.
-
-## Features
-
-- **TypeScript Native**: Full type definitions for all API responses.
-- **Lightweight**: Zero dependencies (uses native `fetch`).
-- **Flexible**: Easy to use for both simple scripts and complex applications.
-- **Cross-platform**: Works in Node.js, Browsers, and Edge environments.
-
-## Installation
-
-```bash
-npm install @botcierge/sdk
-```
-
-## Quick Start
-
-```typescript
-import { query_intent } from '@botcierge/sdk';
-
-try {
-  const result = await query_intent("I'd like to share some food");
-  
-  console.log(`Domain: ${result.domain}`);         // foodshare
-  console.log(`Intent: ${result.intent}`);         // share_food
-  console.log(`Confidence: ${result.confidence}`); // high
-} catch (error) {
-  console.error("Classification failed:", error.message);
-}
-```
-
-## API Reference
-
-### `query_intent(utterance: string): Promise<IntentResult>`
-
-A helper function for quick classification using the default configuration (connecting to `http://localhost:8000`).
-
-### `class Botcierge`
-
-The main class for interacting with the Botcierge API.
-
-#### `constructor(config?: BotciergeConfig)`
-
-- `config.baseUrl`: The base URL of your Botcierge API (default: `http://localhost:8000`).
-
-#### `query_intent(utterance: string): Promise<IntentResult>`
-
-Classifies a string into a specific domain and intent.
-
-### Types
-
-#### `IntentResult`
-```typescript
-interface IntentResult {
-  domain: string;                   // The high-level category (e.g., 'moneyshare')
-  intent: string;                   // The specific action (e.g., 'request_loan')
-  confidence: "high" | "medium" | "low";
-  scores: Record<string, number>;   // Raw similarity scores for all intents
-}
-```
-
-## Error Handling
-
-The SDK throws standard `Error` objects if the network request fails or if the API returns a non-2xx status code.
-
-```typescript
-import { Botcierge } from '@botcierge/sdk';
-
-const client = new Botcierge({ baseUrl: 'https://invalid-api.com' });
-
-try {
-  await client.query_intent("hello");
-} catch (e) {
-  console.log(e.message); // "Botcierge API error: ..."
-}
-```
-
-## Development
-
-### Running Tests
-```bash
-npm test
-```
-
-### Building
-```bash
-npm run build
-```
-
-## License
-
-MIT © [horribleprogram](https://npmjs.com/~horribleprogram)
+# @horribleprogram/sdk
+
+[![npm version](https://img.shields.io/npm/v/@horribleprogram/sdk.svg)](https://www.npmjs.com/package/@horribleprogram/sdk)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+Official JavaScript/TypeScript SDK for the **Botcierge Intent Classification API**.
+
+## Features
+
+- **TypeScript Native**: Full type definitions for all API responses.
+- **Lightweight**: Zero dependencies (uses native `fetch`).
+- **Flexible**: Easy to use for both simple scripts and complex applications.
+- **Cross-platform**: Works in Node.js, Browsers, and Edge environments.
+
+## Installation
+
+```bash
+npm install @horribleprogram/sdk
+```
+
+## Quick Start
+
+```typescript
+import { query_intent } from '@horribleprogram/sdk';
+
+const result = await query_intent("I'd like to share some food");
+
+console.log(result.domain);     // FOODLINK
+console.log(result.intent);     // share_food
+console.log(result.confidence); // high
+```
+
+No configuration needed — the SDK points to the hosted API at `https://horribleprogram-intentapi.hf.space` by default.
+
+## Try it in Node
+
+Create a file `demo.mjs` and run it with `node demo.mjs`:
+
+```js
+import { Botcierge, query_intent } from '@horribleprogram/sdk'
+
+// Zero-config — hits the hosted API by default
+console.log(await query_intent("I'm hungry"))
+console.log(await query_intent("I need help with a bill"))
+
+// Or with an explicit client
+const client = new Botcierge()
+console.log(await client.query_intent("I want to add milk to my list"))
+```
+
+Example output:
+```json
+{ "domain": "FOODLINK",   "intent": "request_food",      "confidence": "high" }
+{ "domain": "BILLBRIDGE", "intent": "request_bill_help",  "confidence": "high" }
+{ "domain": "SHOP_SAVVY", "intent": "add_item",           "confidence": "high" }
+```
+
+## API Reference
+
+### `query_intent(utterance: string): Promise<IntentResult>`
+
+Classifies a string using the default client (hosted API).
+
+### `class Botcierge`
+
+#### `constructor(config?: BotciergeConfig)`
+
+- `config.baseUrl`: Override the API base URL (default: `https://horribleprogram-intentapi.hf.space`).
+
+#### `query_intent(utterance: string): Promise<IntentResult>`
+
+Classifies a string into a specific domain and intent.
+
+### Types
+
+#### `IntentResult`
+```typescript
+interface IntentResult {
+  domain: string;
+  intent: string;
+  confidence: "high" | "medium" | "low";
+  scores?: Record<string, number>;
+}
+```
+
+## Error Handling
+
+```typescript
+import { Botcierge } from '@horribleprogram/sdk';
+
+const client = new Botcierge({ baseUrl: 'https://my-own-instance.com' });
+
+try {
+  await client.query_intent("hello");
+} catch (e) {
+  console.log(e.message); // "Botcierge API error: ..."
+}
+```
+
+## Development
+
+```bash
+npm test
+npm run build
+```
+
+## License
+
+MIT © [horribleprogram](https://npmjs.com/~horribleprogram)

package/dist/chunk-4DZAWXQM.js

@@ -0,0 +1,280 @@
+// src/index.ts
+import { pipeline } from "@xenova/transformers";
+
+// src/intents.ts
+var INTENTS_TAXONOMY = {
+  "GIFTMINDER": {
+    "view_events": [
+      "I want to see events coming up"
+    ],
+    "view_wishlists": [
+      "I want to see wishlists"
+    ],
+    "buy_gifts": [
+      "I want to buy gifts for my loved ones"
+    ]
+  },
+  "REMEMBOT": {
+    "remember": [
+      "I want to remember something"
+    ],
+    "recall": [
+      "I want to recall something"
+    ]
+  },
+  "RESCUEBIDS": {
+    "avoid_fee": [
+      "I want to avoid a bank fee"
+    ],
+    "request_loan": [
+      "I need a small loan"
+    ]
+  },
+  "FOODLINK": {
+    "request_food": [
+      "I'm hungry",
+      "I am hungry"
+    ],
+    "share_food": [
+      "I have food to share",
+      "I want to share some food"
+    ]
+  },
+  "BILLBRIDGE": {
+    "request_bill_help": [
+      "I need help with a bill"
+    ]
+  },
+  "BLOODSHARE": {
+    "request_blood": [
+      "I need blood"
+    ],
+    "share_blood": [
+      "I have blood to share"
+    ]
+  },
+  "MATH": {
+    "compute_expression": [
+      "compute xplusonesquared",
+      "compute xsquaredplusone",
+      "Compute x squared plus one",
+      "Compute x plus one squared"
+    ]
+  },
+  "SHOP_SAVVY": {
+    "add_item": [
+      "Add an item to my shopping list",
+      "I want to add items to my shopping list"
+    ],
+    "remove_item": [
+      "Remove an item from my shopping list"
+    ],
+    "view_list": [
+      "What's on my shopping list",
+      "I want to see my shopping list"
+    ],
+    "edit_list": [
+      "I want to change my shopping list"
+    ],
+    "mark_purchased": [
+      "I've purchased some items on my shopping list",
+      "I have purchased some items on my shopping list"
+    ]
+  },
+  "BOT_STORE": {
+    "add_package": [
+      "I want to buy a package",
+      "I want to add a package"
+    ],
+    "remove_package": [
+      "I want to stop using a package",
+      "I want to remove a package"
+    ]
+  },
+  "TASKMASTERAI": {
+    "add_task": [
+      "I want to add a to do list item to my list",
+      "Add a to do list item",
+      "I want to add a to do list item"
+    ],
+    "view_tasks": [
+      "I want to see my to do list",
+      "Show to do list",
+      "Show my to do list"
+    ],
+    "edit_tasks": [
+      "I want to change my to do list",
+      "I want to edit my to do list"
+    ],
+    "remove_task": [
+      "I want to remove items from my to do list"
+    ]
+  },
+  "FLOWPLANNER": {
+    "create_flow_plan": [
+      "Create a flow plan for tomorrow"
+    ]
+  },
+  "CGM": {
+    "find_company": [
+      "I need a company that...",
+      "I want a company that..."
+    ]
+  },
+  "TIFFANY": {
+    "process_dental_patient": [
+      "I want to process the patient visit for <patient-name>"
+    ]
+  },
+  "CAREMEDIC": {
+    "process_medical_patient": [
+      "I want to process the patient visit for <patient-name>"
+    ]
+  },
+  "DISCRETION": {
+    "retrieval_augmented_generation": [
+      "I need to retrieve information from my documents"
+    ]
+  }
+};
+
+// src/index.ts
+var extractorPromise = null;
+function getExtractor() {
+  if (!extractorPromise) {
+    extractorPromise = (async () => {
+      console.log("Loading local embedding model 'Xenova/all-MiniLM-L12-v2'...");
+      console.log("Note: On first run, this will download the model files (approx. 120MB).");
+      const extractor = await pipeline("feature-extraction", "Xenova/all-MiniLM-L12-v2");
+      console.log("Local embedding model loaded successfully.");
+      return extractor;
+    })();
+  }
+  return extractorPromise;
+}
+var cachedTaxonomyEmbeddingsPromise = null;
+async function getTaxonomyEmbeddings() {
+  if (cachedTaxonomyEmbeddingsPromise) {
+    return cachedTaxonomyEmbeddingsPromise;
+  }
+  cachedTaxonomyEmbeddingsPromise = (async () => {
+    const extractor = await getExtractor();
+    const items = [];
+    for (const [domain, intents] of Object.entries(INTENTS_TAXONOMY)) {
+      for (const [intent, utterances] of Object.entries(intents)) {
+        for (const utterance of utterances) {
+          if (utterance.trim()) {
+            items.push({ domain, intent, utterance });
+          }
+        }
+      }
+    }
+    const texts = items.map((item) => item.utterance);
+    const output = await extractor(texts, { pooling: "mean", normalize: true });
+    const dim = output.dims[1];
+    const data = output.data;
+    const embeddings = [];
+    for (let i = 0; i < items.length; i++) {
+      const startIndex = i * dim;
+      const vector = Array.from(data.subarray(startIndex, startIndex + dim));
+      embeddings.push({
+        ...items[i],
+        vector
+      });
+    }
+    return embeddings;
+  })();
+  return cachedTaxonomyEmbeddingsPromise;
+}
+function cosineSimilarity(v1, v2) {
+  let dotProduct = 0;
+  let norm1 = 0;
+  let norm2 = 0;
+  for (let i = 0; i < v1.length; i++) {
+    dotProduct += v1[i] * v2[i];
+    norm1 += v1[i] * v1[i];
+    norm2 += v2[i] * v2[i];
+  }
+  if (norm1 === 0 || norm2 === 0) return 0;
+  return dotProduct / (Math.sqrt(norm1) * Math.sqrt(norm2));
+}
+var Botcierge = class {
+  baseUrl;
+  apiKey;
+  constructor(config = {}) {
+    this.baseUrl = config.baseUrl || "https://horribleprogram-intentapi.hf.space";
+    this.apiKey = config.apiKey;
+  }
+  getApiKey() {
+    if (this.apiKey) return this.apiKey;
+    if (typeof process !== "undefined" && process.env) {
+      return process.env.BOTCIERGE_API_KEY || process.env.OPENAI_API_KEY;
+    }
+    return void 0;
+  }
+  async query_intent(utterance) {
+    const apiKey = this.getApiKey();
+    if (apiKey) {
+      const headers = {
+        "Content-Type": "application/json",
+        "Authorization": `Bearer ${apiKey}`,
+        "X-API-Key": apiKey
+      };
+      const response = await fetch(`${this.baseUrl}/classify`, {
+        method: "POST",
+        headers,
+        body: JSON.stringify({ utterance })
+      });
+      if (!response.ok) {
+        const error = await response.json().catch(() => ({ detail: "Unknown error" }));
+        throw new Error(`Botcierge API error: ${error.detail || response.statusText}`);
+      }
+      return await response.json();
+    } else {
+      const extractor = await getExtractor();
+      const output = await extractor(utterance, { pooling: "mean", normalize: true });
+      const queryVector = Array.from(output.data);
+      const taxEmbeddings = await getTaxonomyEmbeddings();
+      let bestSim = -1;
+      let bestItem = null;
+      for (const item of taxEmbeddings) {
+        const sim = cosineSimilarity(queryVector, item.vector);
+        if (sim > bestSim) {
+          bestSim = sim;
+          bestItem = item;
+        }
+      }
+      if (!bestItem || bestSim < 0.35) {
+        return {
+          domain: "unknown",
+          intent: "unknown",
+          confidence: "low",
+          scores: {}
+        };
+      }
+      let confidence = "low";
+      if (bestSim >= 0.7) {
+        confidence = "high";
+      } else if (bestSim >= 0.5) {
+        confidence = "medium";
+      }
+      return {
+        domain: bestItem.domain,
+        intent: bestItem.intent,
+        confidence,
+        scores: {
+          [bestItem.intent]: bestSim
+        }
+      };
+    }
+  }
+};
+var defaultClient = new Botcierge();
+async function query_intent(utterance) {
+  return defaultClient.query_intent(utterance);
+}
+
+export {
+  Botcierge,
+  query_intent
+};

package/dist/index.cjs

@@ -24,24 +24,274 @@ __export(index_exports, {
   query_intent: () => query_intent
 });
 module.exports = __toCommonJS(index_exports);
+var import_transformers = require("@xenova/transformers");
+
+// src/intents.ts
+var INTENTS_TAXONOMY = {
+  "GIFTMINDER": {
+    "view_events": [
+      "I want to see events coming up"
+    ],
+    "view_wishlists": [
+      "I want to see wishlists"
+    ],
+    "buy_gifts": [
+      "I want to buy gifts for my loved ones"
+    ]
+  },
+  "REMEMBOT": {
+    "remember": [
+      "I want to remember something"
+    ],
+    "recall": [
+      "I want to recall something"
+    ]
+  },
+  "RESCUEBIDS": {
+    "avoid_fee": [
+      "I want to avoid a bank fee"
+    ],
+    "request_loan": [
+      "I need a small loan"
+    ]
+  },
+  "FOODLINK": {
+    "request_food": [
+      "I'm hungry",
+      "I am hungry"
+    ],
+    "share_food": [
+      "I have food to share",
+      "I want to share some food"
+    ]
+  },
+  "BILLBRIDGE": {
+    "request_bill_help": [
+      "I need help with a bill"
+    ]
+  },
+  "BLOODSHARE": {
+    "request_blood": [
+      "I need blood"
+    ],
+    "share_blood": [
+      "I have blood to share"
+    ]
+  },
+  "MATH": {
+    "compute_expression": [
+      "compute xplusonesquared",
+      "compute xsquaredplusone",
+      "Compute x squared plus one",
+      "Compute x plus one squared"
+    ]
+  },
+  "SHOP_SAVVY": {
+    "add_item": [
+      "Add an item to my shopping list",
+      "I want to add items to my shopping list"
+    ],
+    "remove_item": [
+      "Remove an item from my shopping list"
+    ],
+    "view_list": [
+      "What's on my shopping list",
+      "I want to see my shopping list"
+    ],
+    "edit_list": [
+      "I want to change my shopping list"
+    ],
+    "mark_purchased": [
+      "I've purchased some items on my shopping list",
+      "I have purchased some items on my shopping list"
+    ]
+  },
+  "BOT_STORE": {
+    "add_package": [
+      "I want to buy a package",
+      "I want to add a package"
+    ],
+    "remove_package": [
+      "I want to stop using a package",
+      "I want to remove a package"
+    ]
+  },
+  "TASKMASTERAI": {
+    "add_task": [
+      "I want to add a to do list item to my list",
+      "Add a to do list item",
+      "I want to add a to do list item"
+    ],
+    "view_tasks": [
+      "I want to see my to do list",
+      "Show to do list",
+      "Show my to do list"
+    ],
+    "edit_tasks": [
+      "I want to change my to do list",
+      "I want to edit my to do list"
+    ],
+    "remove_task": [
+      "I want to remove items from my to do list"
+    ]
+  },
+  "FLOWPLANNER": {
+    "create_flow_plan": [
+      "Create a flow plan for tomorrow"
+    ]
+  },
+  "CGM": {
+    "find_company": [
+      "I need a company that...",
+      "I want a company that..."
+    ]
+  },
+  "TIFFANY": {
+    "process_dental_patient": [
+      "I want to process the patient visit for <patient-name>"
+    ]
+  },
+  "CAREMEDIC": {
+    "process_medical_patient": [
+      "I want to process the patient visit for <patient-name>"
+    ]
+  },
+  "DISCRETION": {
+    "retrieval_augmented_generation": [
+      "I need to retrieve information from my documents"
+    ]
+  }
+};
+
+// src/index.ts
+var extractorPromise = null;
+function getExtractor() {
+  if (!extractorPromise) {
+    extractorPromise = (async () => {
+      console.log("Loading local embedding model 'Xenova/all-MiniLM-L12-v2'...");
+      console.log("Note: On first run, this will download the model files (approx. 120MB).");
+      const extractor = await (0, import_transformers.pipeline)("feature-extraction", "Xenova/all-MiniLM-L12-v2");
+      console.log("Local embedding model loaded successfully.");
+      return extractor;
+    })();
+  }
+  return extractorPromise;
+}
+var cachedTaxonomyEmbeddingsPromise = null;
+async function getTaxonomyEmbeddings() {
+  if (cachedTaxonomyEmbeddingsPromise) {
+    return cachedTaxonomyEmbeddingsPromise;
+  }
+  cachedTaxonomyEmbeddingsPromise = (async () => {
+    const extractor = await getExtractor();
+    const items = [];
+    for (const [domain, intents] of Object.entries(INTENTS_TAXONOMY)) {
+      for (const [intent, utterances] of Object.entries(intents)) {
+        for (const utterance of utterances) {
+          if (utterance.trim()) {
+            items.push({ domain, intent, utterance });
+          }
+        }
+      }
+    }
+    const texts = items.map((item) => item.utterance);
+    const output = await extractor(texts, { pooling: "mean", normalize: true });
+    const dim = output.dims[1];
+    const data = output.data;
+    const embeddings = [];
+    for (let i = 0; i < items.length; i++) {
+      const startIndex = i * dim;
+      const vector = Array.from(data.subarray(startIndex, startIndex + dim));
+      embeddings.push({
+        ...items[i],
+        vector
+      });
+    }
+    return embeddings;
+  })();
+  return cachedTaxonomyEmbeddingsPromise;
+}
+function cosineSimilarity(v1, v2) {
+  let dotProduct = 0;
+  let norm1 = 0;
+  let norm2 = 0;
+  for (let i = 0; i < v1.length; i++) {
+    dotProduct += v1[i] * v2[i];
+    norm1 += v1[i] * v1[i];
+    norm2 += v2[i] * v2[i];
+  }
+  if (norm1 === 0 || norm2 === 0) return 0;
+  return dotProduct / (Math.sqrt(norm1) * Math.sqrt(norm2));
+}
 var Botcierge = class {
   baseUrl;
+  apiKey;
   constructor(config = {}) {
     this.baseUrl = config.baseUrl || "https://horribleprogram-intentapi.hf.space";
+    this.apiKey = config.apiKey;
+  }
+  getApiKey() {
+    if (this.apiKey) return this.apiKey;
+    if (typeof process !== "undefined" && process.env) {
+      return process.env.BOTCIERGE_API_KEY || process.env.OPENAI_API_KEY;
+    }
+    return void 0;
   }
   async query_intent(utterance) {
-    const response = await fetch(`${this.baseUrl}/classify`, {
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify({ utterance })
-    });
-    if (!response.ok) {
-      const error = await response.json().catch(() => ({ detail: "Unknown error" }));
-      throw new Error(`Botcierge API error: ${error.detail || response.statusText}`);
+    const apiKey = this.getApiKey();
+    if (apiKey) {
+      const headers = {
+        "Content-Type": "application/json",
+        "Authorization": `Bearer ${apiKey}`,
+        "X-API-Key": apiKey
+      };
+      const response = await fetch(`${this.baseUrl}/classify`, {
+        method: "POST",
+        headers,
+        body: JSON.stringify({ utterance })
+      });
+      if (!response.ok) {
+        const error = await response.json().catch(() => ({ detail: "Unknown error" }));
+        throw new Error(`Botcierge API error: ${error.detail || response.statusText}`);
+      }
+      return await response.json();
+    } else {
+      const extractor = await getExtractor();
+      const output = await extractor(utterance, { pooling: "mean", normalize: true });
+      const queryVector = Array.from(output.data);
+      const taxEmbeddings = await getTaxonomyEmbeddings();
+      let bestSim = -1;
+      let bestItem = null;
+      for (const item of taxEmbeddings) {
+        const sim = cosineSimilarity(queryVector, item.vector);
+        if (sim > bestSim) {
+          bestSim = sim;
+          bestItem = item;
+        }
+      }
+      if (!bestItem || bestSim < 0.35) {
+        return {
+          domain: "unknown",
+          intent: "unknown",
+          confidence: "low",
+          scores: {}
+        };
+      }
+      let confidence = "low";
+      if (bestSim >= 0.7) {
+        confidence = "high";
+      } else if (bestSim >= 0.5) {
+        confidence = "medium";
+      }
+      return {
+        domain: bestItem.domain,
+        intent: bestItem.intent,
+        confidence,
+        scores: {
+          [bestItem.intent]: bestSim
+        }
+      };
     }
-    return await response.json();
   }
 };
 var defaultClient = new Botcierge();

package/dist/index.d.cts

@@ -6,15 +6,17 @@ interface IntentResult {
 }
 interface BotciergeConfig {
     baseUrl?: string;
+    apiKey?: string;
 }
 declare class Botcierge {
     private baseUrl;
+    private apiKey?;
     constructor(config?: BotciergeConfig);
+    private getApiKey;
     query_intent(utterance: string): Promise<IntentResult>;
 }
 /**
  * Classifies an utterance into an intent.
- * By default, connects to http://localhost:8000
  */
 declare function query_intent(utterance: string): Promise<IntentResult>;
 

package/dist/index.d.ts

@@ -6,15 +6,17 @@ interface IntentResult {
 }
 interface BotciergeConfig {
     baseUrl?: string;
+    apiKey?: string;
 }
 declare class Botcierge {
     private baseUrl;
+    private apiKey?;
     constructor(config?: BotciergeConfig);
+    private getApiKey;
     query_intent(utterance: string): Promise<IntentResult>;
 }
 /**
  * Classifies an utterance into an intent.
- * By default, connects to http://localhost:8000
  */
 declare function query_intent(utterance: string): Promise<IntentResult>;
 

package/dist/index.js

@@ -1,7 +1,7 @@
 import {
   Botcierge,
   query_intent
-} from "./chunk-EWLGKE65.js";
+} from "./chunk-4DZAWXQM.js";
 export {
   Botcierge,
   query_intent

package/dist/main.cjs

@@ -26,24 +26,274 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 var readline = __toESM(require("readline"), 1);
 
 // src/index.ts
+var import_transformers = require("@xenova/transformers");
+
+// src/intents.ts
+var INTENTS_TAXONOMY = {
+  "GIFTMINDER": {
+    "view_events": [
+      "I want to see events coming up"
+    ],
+    "view_wishlists": [
+      "I want to see wishlists"
+    ],
+    "buy_gifts": [
+      "I want to buy gifts for my loved ones"
+    ]
+  },
+  "REMEMBOT": {
+    "remember": [
+      "I want to remember something"
+    ],
+    "recall": [
+      "I want to recall something"
+    ]
+  },
+  "RESCUEBIDS": {
+    "avoid_fee": [
+      "I want to avoid a bank fee"
+    ],
+    "request_loan": [
+      "I need a small loan"
+    ]
+  },
+  "FOODLINK": {
+    "request_food": [
+      "I'm hungry",
+      "I am hungry"
+    ],
+    "share_food": [
+      "I have food to share",
+      "I want to share some food"
+    ]
+  },
+  "BILLBRIDGE": {
+    "request_bill_help": [
+      "I need help with a bill"
+    ]
+  },
+  "BLOODSHARE": {
+    "request_blood": [
+      "I need blood"
+    ],
+    "share_blood": [
+      "I have blood to share"
+    ]
+  },
+  "MATH": {
+    "compute_expression": [
+      "compute xplusonesquared",
+      "compute xsquaredplusone",
+      "Compute x squared plus one",
+      "Compute x plus one squared"
+    ]
+  },
+  "SHOP_SAVVY": {
+    "add_item": [
+      "Add an item to my shopping list",
+      "I want to add items to my shopping list"
+    ],
+    "remove_item": [
+      "Remove an item from my shopping list"
+    ],
+    "view_list": [
+      "What's on my shopping list",
+      "I want to see my shopping list"
+    ],
+    "edit_list": [
+      "I want to change my shopping list"
+    ],
+    "mark_purchased": [
+      "I've purchased some items on my shopping list",
+      "I have purchased some items on my shopping list"
+    ]
+  },
+  "BOT_STORE": {
+    "add_package": [
+      "I want to buy a package",
+      "I want to add a package"
+    ],
+    "remove_package": [
+      "I want to stop using a package",
+      "I want to remove a package"
+    ]
+  },
+  "TASKMASTERAI": {
+    "add_task": [
+      "I want to add a to do list item to my list",
+      "Add a to do list item",
+      "I want to add a to do list item"
+    ],
+    "view_tasks": [
+      "I want to see my to do list",
+      "Show to do list",
+      "Show my to do list"
+    ],
+    "edit_tasks": [
+      "I want to change my to do list",
+      "I want to edit my to do list"
+    ],
+    "remove_task": [
+      "I want to remove items from my to do list"
+    ]
+  },
+  "FLOWPLANNER": {
+    "create_flow_plan": [
+      "Create a flow plan for tomorrow"
+    ]
+  },
+  "CGM": {
+    "find_company": [
+      "I need a company that...",
+      "I want a company that..."
+    ]
+  },
+  "TIFFANY": {
+    "process_dental_patient": [
+      "I want to process the patient visit for <patient-name>"
+    ]
+  },
+  "CAREMEDIC": {
+    "process_medical_patient": [
+      "I want to process the patient visit for <patient-name>"
+    ]
+  },
+  "DISCRETION": {
+    "retrieval_augmented_generation": [
+      "I need to retrieve information from my documents"
+    ]
+  }
+};
+
+// src/index.ts
+var extractorPromise = null;
+function getExtractor() {
+  if (!extractorPromise) {
+    extractorPromise = (async () => {
+      console.log("Loading local embedding model 'Xenova/all-MiniLM-L12-v2'...");
+      console.log("Note: On first run, this will download the model files (approx. 120MB).");
+      const extractor = await (0, import_transformers.pipeline)("feature-extraction", "Xenova/all-MiniLM-L12-v2");
+      console.log("Local embedding model loaded successfully.");
+      return extractor;
+    })();
+  }
+  return extractorPromise;
+}
+var cachedTaxonomyEmbeddingsPromise = null;
+async function getTaxonomyEmbeddings() {
+  if (cachedTaxonomyEmbeddingsPromise) {
+    return cachedTaxonomyEmbeddingsPromise;
+  }
+  cachedTaxonomyEmbeddingsPromise = (async () => {
+    const extractor = await getExtractor();
+    const items = [];
+    for (const [domain, intents] of Object.entries(INTENTS_TAXONOMY)) {
+      for (const [intent, utterances] of Object.entries(intents)) {
+        for (const utterance of utterances) {
+          if (utterance.trim()) {
+            items.push({ domain, intent, utterance });
+          }
+        }
+      }
+    }
+    const texts = items.map((item) => item.utterance);
+    const output = await extractor(texts, { pooling: "mean", normalize: true });
+    const dim = output.dims[1];
+    const data = output.data;
+    const embeddings = [];
+    for (let i = 0; i < items.length; i++) {
+      const startIndex = i * dim;
+      const vector = Array.from(data.subarray(startIndex, startIndex + dim));
+      embeddings.push({
+        ...items[i],
+        vector
+      });
+    }
+    return embeddings;
+  })();
+  return cachedTaxonomyEmbeddingsPromise;
+}
+function cosineSimilarity(v1, v2) {
+  let dotProduct = 0;
+  let norm1 = 0;
+  let norm2 = 0;
+  for (let i = 0; i < v1.length; i++) {
+    dotProduct += v1[i] * v2[i];
+    norm1 += v1[i] * v1[i];
+    norm2 += v2[i] * v2[i];
+  }
+  if (norm1 === 0 || norm2 === 0) return 0;
+  return dotProduct / (Math.sqrt(norm1) * Math.sqrt(norm2));
+}
 var Botcierge = class {
   baseUrl;
+  apiKey;
   constructor(config = {}) {
     this.baseUrl = config.baseUrl || "https://horribleprogram-intentapi.hf.space";
+    this.apiKey = config.apiKey;
+  }
+  getApiKey() {
+    if (this.apiKey) return this.apiKey;
+    if (typeof process !== "undefined" && process.env) {
+      return process.env.BOTCIERGE_API_KEY || process.env.OPENAI_API_KEY;
+    }
+    return void 0;
   }
   async query_intent(utterance) {
-    const response = await fetch(`${this.baseUrl}/classify`, {
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify({ utterance })
-    });
-    if (!response.ok) {
-      const error = await response.json().catch(() => ({ detail: "Unknown error" }));
-      throw new Error(`Botcierge API error: ${error.detail || response.statusText}`);
+    const apiKey = this.getApiKey();
+    if (apiKey) {
+      const headers = {
+        "Content-Type": "application/json",
+        "Authorization": `Bearer ${apiKey}`,
+        "X-API-Key": apiKey
+      };
+      const response = await fetch(`${this.baseUrl}/classify`, {
+        method: "POST",
+        headers,
+        body: JSON.stringify({ utterance })
+      });
+      if (!response.ok) {
+        const error = await response.json().catch(() => ({ detail: "Unknown error" }));
+        throw new Error(`Botcierge API error: ${error.detail || response.statusText}`);
+      }
+      return await response.json();
+    } else {
+      const extractor = await getExtractor();
+      const output = await extractor(utterance, { pooling: "mean", normalize: true });
+      const queryVector = Array.from(output.data);
+      const taxEmbeddings = await getTaxonomyEmbeddings();
+      let bestSim = -1;
+      let bestItem = null;
+      for (const item of taxEmbeddings) {
+        const sim = cosineSimilarity(queryVector, item.vector);
+        if (sim > bestSim) {
+          bestSim = sim;
+          bestItem = item;
+        }
+      }
+      if (!bestItem || bestSim < 0.35) {
+        return {
+          domain: "unknown",
+          intent: "unknown",
+          confidence: "low",
+          scores: {}
+        };
+      }
+      let confidence = "low";
+      if (bestSim >= 0.7) {
+        confidence = "high";
+      } else if (bestSim >= 0.5) {
+        confidence = "medium";
+      }
+      return {
+        domain: bestItem.domain,
+        intent: bestItem.intent,
+        confidence,
+        scores: {
+          [bestItem.intent]: bestSim
+        }
+      };
     }
-    return await response.json();
   }
 };
 var defaultClient = new Botcierge();

package/dist/main.js

@@ -1,6 +1,6 @@
 import {
   query_intent
-} from "./chunk-EWLGKE65.js";
+} from "./chunk-4DZAWXQM.js";
 
 // src/main.ts
 import * as readline from "readline";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@horribleprogram/sdk",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "SDK for Botcierge Intent Classification",
   "type": "module",
   "main": "./dist/index.js",
@@ -32,5 +32,7 @@
     "typescript": "^5.0.0",
     "vitest": "^4.1.6"
   },
-  "dependencies": {}
+  "dependencies": {
+    "@xenova/transformers": "^2.17.2"
+  }
 }

package/dist/chunk-EWLGKE65.js

@@ -1,30 +0,0 @@
-// src/index.ts
-var Botcierge = class {
-  baseUrl;
-  constructor(config = {}) {
-    this.baseUrl = config.baseUrl || "https://horribleprogram-intentapi.hf.space";
-  }
-  async query_intent(utterance) {
-    const response = await fetch(`${this.baseUrl}/classify`, {
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify({ utterance })
-    });
-    if (!response.ok) {
-      const error = await response.json().catch(() => ({ detail: "Unknown error" }));
-      throw new Error(`Botcierge API error: ${error.detail || response.statusText}`);
-    }
-    return await response.json();
-  }
-};
-var defaultClient = new Botcierge();
-async function query_intent(utterance) {
-  return defaultClient.query_intent(utterance);
-}
-
-export {
-  Botcierge,
-  query_intent
-};