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:
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user