fix missing openai_api_base on langchain-chroma example (#818)

This commit is contained in:
Wendy Liga 2023-07-27 23:41:53 +07:00 committed by GitHub
parent 78e36779cf
commit 9b307799ce
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions

View file

@ -18,8 +18,8 @@ texts = text_splitter.split_documents(documents)
# Supplying a persist_directory will store the embeddings on disk
persist_directory = 'db'
embedding = OpenAIEmbeddings(model="text-embedding-ada-002")
embedding = OpenAIEmbeddings(model="text-embedding-ada-002", openai_api_base=base_path)
vectordb = Chroma.from_documents(documents=texts, embedding=embedding, persist_directory=persist_directory)
vectordb.persist()
vectordb = None
vectordb = None