trieve-ts-sdk

import { TrieveSDK } from "trieve-ts-sdk"; import type { SearchPayload } from "trieve-ts-sdk"; // Explicitly import a type for clarity // Initialize the Trieve SDK client with your API key and dataset ID. // It's recommended to load these from environment variables for production applications. const trieve = new TrieveSDK({ apiKey: process.env.TRIEVE_API_KEY ?? "", // Replace with your actual Trieve API Key datasetId: process.env.TRIEVE_DATASET_ID ?? "", // Replace with your actual Trieve Dataset ID }); async function performSearch(queryText: string) { if (!process.env.TRIEVE_API_KEY || !process.env.TRIEVE_DATASET_ID) { console.error("Missing TRIEVE_API_KEY or TRIEVE_DATASET_ID environment variables. Please set them."); return; } const searchPayload: SearchPayload = { query: queryText, search_type: "hybrid", // You can also use "semantic" or "fulltext" filters: {}, // Optional: Add filters to narrow down search results page: 1, page_size: 10, highlight_results: true, }; try { console.log(`Searching for: "${queryText}" in dataset "${trieve.datasetId}"...`); const data = await trieve.search(searchPayload); console.log("Search Results:", JSON.stringify(data, null, 2)); if (data.hits && data.hits.length > 0) { console.log(`Found ${data.hits.length} relevant results.`); data.hits.forEach((hit, index) => { console.log(`Result ${index + 1}: ${hit.chunk_html ?? hit.link ?? 'No content preview'}`); }); } else { console.log("No results found for your query."); } } catch (error) { console.error("Error during Trieve search:", error); } } // Example usage: performSearch("What are the key features of the Trieve API and how do I use them?");