ExternalVendorCode/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-06-21 00:04:59 +00:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Wendy Liga
2023-07-27 23:41:53 +07:00
committed by GitHub
parent 78e36779cf
commit 9b307799ce
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/langchain-chroma/query.py
View File

@ -9,7 +9,7 @@ from langchain.vectorstores.base import VectorStoreRetriever
base_path = os.environ.get('OPENAI_API_BASE', 'http://localhost:8080/v1') base_path = os.environ.get('OPENAI_API_BASE', 'http://localhost:8080/v1')
# Load and process the text # Load and process the text
embedding = OpenAIEmbeddings() embedding = OpenAIEmbeddings(model="text-embedding-ada-002", openai_api_base=base_path)
persist_directory = 'db' persist_directory = 'db'
# Now we can load the persisted database from disk, and use it as normal. # Now we can load the persisted database from disk, and use it as normal.

4
examples/langchain-chroma/store.py
View File

@ -18,8 +18,8 @@ texts = text_splitter.split_documents(documents)
# Supplying a persist_directory will store the embeddings on disk # Supplying a persist_directory will store the embeddings on disk
persist_directory = 'db' 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 = Chroma.from_documents(documents=texts, embedding=embedding, persist_directory=persist_directory)
vectordb.persist() vectordb.persist()
vectordb = None vectordb = None