File size: 1,218 Bytes
1e9fae3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
99bedac
1e9fae3
e07a544
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import chainlit as cl
from langchain.embeddings.openai import OpenAIEmbeddings
from langchain.document_loaders.csv_loader import CSVLoader
from langchain.embeddings import CacheBackedEmbeddings
from langchain.text_splitter import RecursiveCharacterTextSplitter
from langchain.vectorstores import FAISS
from langchain.chains import RetrievalQA
from langchain.chat_models import ChatOpenAI
from langchain.storage import LocalFileStore
from langchain.prompts.chat import (
    ChatPromptTemplate,
    SystemMessagePromptTemplate,
    HumanMessagePromptTemplate,
)
import chainlit as cl

import build_langchain_vector_store

build_langchain_vector_store

from langchain.vectorstores import Chroma
from langchain.embeddings import OpenAIEmbeddings
import openai
import os

openai.api_key = os.getenv("OPENAI_API_KEY")
openai.api_base = 'https://api.openai.com/v1' # default

embedding_model_name = "text-embedding-ada-002"
embedding_model = OpenAIEmbeddings(model=embedding_model_name)
read_vector_store = Chroma(
        persist_directory="langchain-chroma-pulze-docs", embedding_function=embedding_model
    )
query_results = read_vector_store.similarity_search("How do I use Pulze?")
print(query_results[0].page_content)