fix: openai env (#1400)

lua/avante/rag_service.lua

@@ -63,12 +63,13 @@ function M.launch_rag_service(cb)
     Utils.debug(string.format("container %s not found, starting...", container_name))
   end
   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,
     container_name,
     data_path,
     openai_api_key,
     openai_base_url,
+    os.getenv("OPENAI_EMBED_MODEL"),
     image
   )
   vim.fn.jobstart(cmd_, {

py/rag-service/src/main.py

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