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

fix: use host network for rag service container (#1649)

Browse Source
This commit is contained in:
yetone
2025-03-20 03:45:44 +08:00
committed by GitHub
parent 8c9cd1a7b2
commit 008fe29de8
3 changed files with 5 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Makefile
View File

@@ -17,7 +17,7 @@ BUILD_DIR := build
BUILD_FROM_SOURCE ?= false BUILD_FROM_SOURCE ?= false
TARGET_LIBRARY ?= all TARGET_LIBRARY ?= all
RAG_SERVICE_VERSION ?= 0.0.8 RAG_SERVICE_VERSION ?= 0.0.9
RAG_SERVICE_IMAGE := quay.io/yetoneful/avante-rag-service:$(RAG_SERVICE_VERSION) RAG_SERVICE_IMAGE := quay.io/yetoneful/avante-rag-service:$(RAG_SERVICE_VERSION)
all: luajit all: luajit

5
lua/avante/rag_service.lua
View File

@@ -8,7 +8,7 @@ local M = {}
local container_name = "avante-rag-service" local container_name = "avante-rag-service"
local service_path = "/tmp/" .. container_name local service_path = "/tmp/" .. container_name
function M.get_rag_service_image() return "quay.io/yetoneful/avante-rag-service:0.0.8" end function M.get_rag_service_image() return "quay.io/yetoneful/avante-rag-service:0.0.9" end
function M.get_rag_service_port() return 20250 end function M.get_rag_service_port() return 20250 end
@@ -74,8 +74,7 @@ function M.launch_rag_service(cb)
M.stop_rag_service() M.stop_rag_service()
end end
local cmd_ = string.format( local cmd_ = string.format(
"docker run -d -p %d:8000 --name %s -v %s:/data -v %s:/host:ro -e ALLOW_RESET=TRUE -e DATA_DIR=/data -e RAG_PROVIDER=%s -e %s_API_KEY=%s -e %s_API_BASE=%s -e RAG_LLM_MODEL=%s -e RAG_EMBED_MODEL=%s %s %s", "docker run -d --network=host --name %s -v %s:/data -v %s:/host:ro -e ALLOW_RESET=TRUE -e DATA_DIR=/data -e RAG_PROVIDER=%s -e %s_API_KEY=%s -e %s_API_BASE=%s -e RAG_LLM_MODEL=%s -e RAG_EMBED_MODEL=%s %s %s",
port,
container_name, container_name,
data_path, data_path,
Config.rag_service.host_mount, Config.rag_service.host_mount,

7
py/rag-service/Dockerfile
View File

@@ -15,11 +15,8 @@ RUN uv venv
RUN uv pip install -r requirements.txt RUN uv pip install -r requirements.txt
ENV PYTHONUNBUFFERED=1 \ ENV PYTHONUNBUFFERED=1 \
PYTHONDONTWRITEBYTECODE=1 \ PYTHONDONTWRITEBYTECODE=1
PORT=8000
EXPOSE ${PORT}
COPY . . COPY . .
CMD ["uv", "run", "fastapi", "run", "src/main.py", "--workers", "3"] CMD ["uv", "run", "fastapi", "run", "src/main.py", "--workers", "3", "--port", "20250"]