Added nbconvert needed for rag indexing jupyter notebooks (#2862)

* Added nbconvert needed for rag indexing jupyter notebooks

* Allow rag service image to be configured

* [pre-commit.ci lite] apply automatic fixes

---------

Co-authored-by: pre-commit-ci-lite[bot] <117423508+pre-commit-ci-lite[bot]@users.noreply.github.com>
This commit is contained in:
edmundhighcock
2025-12-30 04:59:52 +01:00
committed by GitHub
parent 7608dce00e
commit f663865186
3 changed files with 29 additions and 2 deletions

View File

@@ -58,6 +58,8 @@ M._defaults = {
enabled = false, -- Enables the RAG service
host_mount = os.getenv("HOME"), -- Host mount path for the RAG service (Docker will mount this path)
runner = "docker", -- The runner for the RAG service (can use docker or nix)
-- The image to use to run the rag service if runner is docker
image = "quay.io/yetoneful/avante-rag-service:0.0.11",
llm = { -- Configuration for the Language Model (LLM) used by the RAG service
provider = "openai", -- The LLM provider
endpoint = "https://api.openai.com/v1", -- The LLM API endpoint

View File

@@ -8,7 +8,13 @@ local M = {}
local container_name = "avante-rag-service"
local service_path = "/tmp/" .. container_name
function M.get_rag_service_image() return "quay.io/yetoneful/avante-rag-service:0.0.11" end
function M.get_rag_service_image()
if Config.rag_service and Config.rag_service.image then
return Config.rag_service.image
else
return "quay.io/yetoneful/avante-rag-service:0.0.11"
end
end
function M.get_rag_service_port() return 20250 end