Semantic search on embeddings no vendor can retire

import { SIEClient } from '@superlinked/sie-sdk';

const client = new SIEClient('https://api.superlinked.com', {
  apiKey: 'sk-sie-…',
});

const query = "What happens if a monthly depositor racks up $100,000 of payroll tax in a single day?";
const documents = [
  "If you accumulate $100,000 or more in taxes on any day during a monthly or semiweekly deposit period (see Deposit period , earlier in this section), you must deposit the tax by the next business day, whether you’re a monthly or semiweekly schedule depositor. The $100,000 tax liability threshold requiring a next-day deposit is determined before you consider any reduction of your liability for nonrefundable credits.",
  "For purposes of the $100,000 rule, don’t continue accumulating a tax liability after the end of a deposit period. For example, if a semiweekly schedule depositor has accumulated a liability of $95,000 on a Tuesday (of a Saturday-through-Tuesday deposit period) and accumulated a $10,000 liability on Wednesday, the $100,000 next-day deposit rule doesn’t apply because the $10,000 is accumulated in the next deposit period. Thus, $95,000 must be deposited by Friday and $10,000 must be deposited by the following Wednesday.",
  "If you’re a monthly schedule depositor and accumulate a $100,000 tax liability on any day during the deposit period, you become a semiweekly schedule depositor on the next day and remain so for at least the rest of the calendar year and for the following calendar year. Example.",
  "However, once you accumulate at least $100,000 in a deposit period, stop accumulating at the end of that day and begin to accumulate anew on the next day. For example, Fir Co. is a semiweekly schedule depositor. On Monday, Fir Co. accumulates taxes of $110,000 and must deposit this amount on Tuesday, the next business day. On Tuesday, Fir Co. accumulates additional taxes of $30,000. Because the $30,000 isn’t added to the previous $110,000 and is less than $100,000, Fir Co. must deposit the $30,000 by Friday (following the semiweekly deposit schedule).",
];

const items = [{ text: query }, ...documents.map((text) => ({ text }))];
const vecs = await client.encode('Snowflake/snowflake-arctic-embed-l-v2.0', items);
const [q, ...d] = vecs.map((v) => v.dense);
const cosine = (a: Float32Array, b: Float32Array) => {
  const dot = a.reduce((s, x, i) => s + x * b[i], 0);
  return dot / (Math.hypot(...a) * Math.hypot(...b));
};
const ranked = documents
  .map((doc, i) => ({ doc, score: cosine(q!, d[i]!) }))
  .sort((a, b) => b.score - a.score);
console.log(ranked);

# Encode the query + candidates in one batch, then rank by cosine.
curl https://api.superlinked.com/v1/encode/Snowflake%2Fsnowflake-arctic-embed-l-v2.0 \
  -H "Authorization: Bearer sk-sie-…" \
  -H "Content-Type: application/json" \
  -d "{\"items\":[{\"text\":\"What happens if a monthly depositor racks up \$100,000 of payroll tax in a single day?\"},{\"text\":\"If you accumulate \$100,000 or more in taxes on any day during a monthly or semiweekly deposit period (see Deposit period , earlier in this section), you must deposit the tax by the next business day, whether you’re a monthly or semiweekly schedule depositor. The \$100,000 tax liability threshold requiring a next-day deposit is determined before you consider any reduction of your liability for nonrefundable credits.\"},{\"text\":\"For purposes of the \$100,000 rule, don’t continue accumulating a tax liability after the end of a deposit period. For example, if a semiweekly schedule depositor has accumulated a liability of \$95,000 on a Tuesday (of a Saturday-through-Tuesday deposit period) and accumulated a \$10,000 liability on Wednesday, the \$100,000 next-day deposit rule doesn’t apply because the \$10,000 is accumulated in the next deposit period. Thus, \$95,000 must be deposited by Friday and \$10,000 must be deposited by the following Wednesday.\"},{\"text\":\"If you’re a monthly schedule depositor and accumulate a \$100,000 tax liability on any day during the deposit period, you become a semiweekly schedule depositor on the next day and remain so for at least the rest of the calendar year and for the following calendar year. Example.\"},{\"text\":\"However, once you accumulate at least \$100,000 in a deposit period, stop accumulating at the end of that day and begin to accumulate anew on the next day. For example, Fir Co. is a semiweekly schedule depositor. On Monday, Fir Co. accumulates taxes of \$110,000 and must deposit this amount on Tuesday, the next business day. On Tuesday, Fir Co. accumulates additional taxes of \$30,000. Because the \$30,000 isn’t added to the previous \$110,000 and is less than \$100,000, Fir Co. must deposit the \$30,000 by Friday (following the semiweekly deposit schedule).\"}],\"params\":{\"output_types\":[\"dense\"]}}"