12345678910111213141516171819202122232425262728293031323334 |
- import { supabaseClient } from '@/utils/supabase-client';
- import { SupabaseVectorStore } from 'langchain/vectorstores';
- import { OpenAIEmbeddings } from 'langchain/embeddings';
- import { VectorDBQAChain } from 'langchain/chains';
- import { openai } from '@/utils/openai-client';
- const query = 'How do i create a notion database?';
- const model = openai;
- async function searchForDocs() {
- const vectorStore = await SupabaseVectorStore.fromExistingIndex(
- supabaseClient,
- new OpenAIEmbeddings(),
- );
- /*uncomment below to test similarity search */
- // const results = await vectorStore.similaritySearch(query, 2);
- // console.log("results", results);
- const chain = VectorDBQAChain.fromLLM(model, vectorStore);
- //Ask a question
- const response = await chain.call({
- query: query,
- });
- console.log('response', response);
- }
- (async () => {
- await searchForDocs();
- })();
|