Если вы использовали Trieve в приложении JavaScript, вы, вероятно, знаете, что вам нужно выполнять большую часть вызовов Trieve с помощью fetch. Хотя этот подход хорош, он не идеален, и мы хотим предоставить пользователям более простой способ использования наших API.

Ну, за кулисами мы работали над тем, чтобы Trieve было проще использовать в приложениях JavaScript, и это включает в себя создание нового JavaScript SDK, который значительно упрощает интеграцию Trieve в любое приложение.

Во-первых, вы можете установить новый trieve-ts-sdk с помощью вашего любимого менеджера пакетов:

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

А теперь давайте посмотрим, как это работает, и возьмем в качестве примера поисковый вызов.

Прежде чем вам нужно будет сделать что-то вроде:

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',
  }),
});

Хотя этот метод работает хорошо, это не самый чистый подход. Вам нужно будет открыть документацию рядом с вашим редактором кода, так как нет типов, которые могли бы помочь вам в вызовах функций. Теперь, с новым SDK, вы можете вызывать ее так:

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

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

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

С помощью экспортированных типов также гораздо проще создать гораздо более сложный поиск, включающий, например, фильтры:

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'],
      },
    ],
  },
});

И это не только методы для фрагментов, у нас есть функции для большинства наших API, которые вы можете использовать, хотите транслировать завершение RAG? У нас это получилось:

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

Мы также создали подробную документацию, чтобы вам было легко найти все эти функции независимо от того, используете ли вы TypeScript или нет.

Хорошо, последний шаг — установить его и приступить к созданию поиска и RAG в вашем приложении!