Si ha utilizado Trieve en una aplicación JavaScript, probablemente sepa que necesita realizar la mayoría de sus llamadas a Trieve utilizando fetch. Si bien este enfoque es bueno, no es ideal y queremos ofrecer a los usuarios una forma más sencilla de utilizar nuestras API.

Bueno, detrás de escena hemos estado trabajando para hacer que Trieve sea más fácil de usar que nunca en aplicaciones JavaScript y eso incluye crear un nuevo SDK de JavaScript que simplifica mucho la integración de Trieve en cualquier aplicación.

Lo primero es lo primero, puedes instalar el nuevo trieve-ts-sdk con tu administrador de paquetes favorito:

yarn add trieve-ts-sdk
# or
npm install trieve-ts-sdk
# or
pnpm install trieve-ts-sdk

Y ahora veamos cómo funciona y tomemos una llamada de búsqueda como ejemplo.

Antes tendrías que hacer algo como:

fetch('https://api.trieve.ai/api/chunk/search', {
  method: 'POST',
  headers: {
    'TR-Dataset': 'dc6f3b0d-cf21-412b-9d16-fb7ade090365',
    Authorization: 'tr-********************************',
  },
  body: JSON.stringify({
    query: 'Sonic the Hedgehog',
  }),
});

Si bien este método funciona bien, no es el enfoque más limpio. Necesitará tener la documentación abierta junto a su editor de código, ya que no hay tipos que lo ayuden a realizar sus llamadas a funciones. Ahora, con el nuevo SDK puede llamarlo así:

import { TrieveSDK } from 'trieve-ts-sdk';

export const trieve = new TrieveSDK({
  apiKey: '',
  datasetId: '',
});

const results = await trieve.search({
  query: 'Sonic the Hedgehog',
});

Con la ayuda de los tipos exportados también es mucho más fácil crear una búsqueda mucho más complicada que incluya, por ejemplo, filtros:

import { TrieveSDK } from 'trieve-ts-sdk';

const results = await trieve.search({
  query: 'Sonic the Hedgehog',
  search_type: 'hybrid',
  filters: {
    must: [
      {
        field: 'meta.rating',
        range: {
          gt: 80,
        },
      },
    ],
    must_not: [
      {
        field: 'metadata.console',
        match: ['gba', 'wii'],
      },
    ],
  },
});

Y no se trata solo de métodos para fragmentos, tenemos funciones para la mayoría de nuestra API que puedes usar. ¿Quieres transmitir una finalización de RAG? Lo tenemos:

const reader = await trieve.createMessageReader({
  topic_id: id || currentTopic,
  new_message_content: currentQuestion,
  llm_options: {
    completion_first: true,
  },
});
handleReader(reader);

También creamos documentos completos para que todas estas funciones sean fáciles de encontrar, ya sea que uses TypeScript o no.

¡Bien, el último paso es instalarlo y comenzar a crear búsqueda y RAG en tu aplicación!