CarGDev/avante.nvim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: openai env (#1400)

Browse Source
This commit is contained in:
nzlov
2025-02-26 17:35:49 +08:00
committed by GitHub
parent 008e8c32b8
commit ef78c36e98
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lua/avante/rag_service.lua
View File

@@ -63,12 +63,13 @@ function M.launch_rag_service(cb)
Utils.debug(string.format("container %s not found, starting...", container_name)) Utils.debug(string.format("container %s not found, starting...", container_name))
end end
local cmd_ = string.format( local cmd_ = string.format(
"docker run -d -p %d:8000 --name %s -v %s:/data -v /:/host -e DATA_DIR=/data -e OPENAI_API_KEY=%s -e OPENAI_BASE_URL=%s %s", "docker run -d -p %d:8000 --name %s -v %s:/data -v /:/host -e DATA_DIR=/data -e OPENAI_API_KEY=%s -e OPENAI_API_BASE=%s -e OPENAI_EMBED_MODEL=%s %s",
port, port,
container_name, container_name,
data_path, data_path,
openai_api_key, openai_api_key,
openai_base_url, openai_base_url,
os.getenv("OPENAI_EMBED_MODEL"),
image image
) )
vim.fn.jobstart(cmd_, { vim.fn.jobstart(cmd_, {

3
py/rag-service/src/main.py
View File

@@ -315,6 +315,9 @@ chroma_collection = chroma_client.get_or_create_collection("documents")
vector_store = ChromaVectorStore(chroma_collection=chroma_collection) vector_store = ChromaVectorStore(chroma_collection=chroma_collection)
storage_context = StorageContext.from_defaults(vector_store=vector_store) storage_context = StorageContext.from_defaults(vector_store=vector_store)
embed_model = OpenAIEmbedding() embed_model = OpenAIEmbedding()
model = os.getenv("OPENAI_EMBED_MODEL", "")
if model:
embed_model = OpenAIEmbedding(model=model)
Settings.embed_model = embed_model Settings.embed_model = embed_model