File tree Expand file tree Collapse file tree 3 files changed +29
-2
lines changed
Expand file tree Collapse file tree 3 files changed +29
-2
lines changed Original file line number Diff line number Diff line change @@ -58,6 +58,8 @@ M._defaults = {
5858 enabled = false , -- Enables the RAG service
5959 host_mount = os.getenv (" HOME" ), -- Host mount path for the RAG service (Docker will mount this path)
6060 runner = " docker" , -- The runner for the RAG service (can use docker or nix)
61+ -- The image to use to run the rag service if runner is docker
62+ image = " quay.io/yetoneful/avante-rag-service:0.0.11" ,
6163 llm = { -- Configuration for the Language Model (LLM) used by the RAG service
6264 provider = " openai" , -- The LLM provider
6365 endpoint = " https://api.openai.com/v1" , -- The LLM API endpoint
Original file line number Diff line number Diff line change @@ -8,7 +8,13 @@ local M = {}
88local container_name = " avante-rag-service"
99local service_path = " /tmp/" .. container_name
1010
11- function M .get_rag_service_image () return " quay.io/yetoneful/avante-rag-service:0.0.11" end
11+ function M .get_rag_service_image ()
12+ if Config .rag_service and Config .rag_service .image then
13+ return Config .rag_service .image
14+ else
15+ return " quay.io/yetoneful/avante-rag-service:0.0.11"
16+ end
17+ end
1218
1319function M .get_rag_service_port () return 20250 end
1420
Original file line number Diff line number Diff line change @@ -9,6 +9,7 @@ attrs==25.1.0
99backoff == 2.2.1
1010bcrypt == 4.2.1
1111beautifulsoup4 == 4.13.3
12+ bleach == 6.3.0
1213build == 1.2.2.post1
1314cachetools == 5.5.1
1415certifi == 2024.12.14
@@ -20,6 +21,7 @@ coloredlogs==15.0.1
2021dashscope == 1.22.2
2122dataclasses-json == 0.6.7
2223decorator == 5.1.1
24+ defusedxml == 0.7.1
2325deprecated == 1.2.18
2426dirtyjson == 1.0.8
2527distro == 1.9.0
@@ -31,6 +33,7 @@ et-xmlfile==2.0.0
3133executing == 2.2.0
3234fastapi == 0.115.8
3335fastapi-cli == 0.0.7
36+ fastjsonschema == 2.21.2
3437filelock == 3.17.0
3538filetype == 1.2.0
3639flatbuffers == 25.1.24
@@ -55,6 +58,11 @@ jedi==0.19.2
5558jinja2 == 3.1.5
5659jiter == 0.8.2
5760joblib == 1.4.2
61+ jsonschema == 4.25.1
62+ jsonschema-specifications == 2025.9.1
63+ jupyter-client == 8.7.0
64+ jupyter-core == 5.9.1
65+ jupyterlab-pygments == 0.3.0
5866kubernetes == 32.0.0
5967llama-cloud == 0.1.11
6068llama-cloud-services == 0.6.0
@@ -84,11 +92,15 @@ markupsafe==3.0.2
8492marshmallow == 3.26.1
8593matplotlib-inline == 0.1.7
8694mdurl == 0.1.2
95+ mistune == 3.1.4
8796mmh3 == 5.1.0
8897monotonic == 1.6
8998mpmath == 1.3.0
9099multidict == 6.1.0
91100mypy-extensions == 1.0.0
101+ nbclient == 0.10.2
102+ nbconvert == 7.16.6
103+ nbformat == 5.10.4
92104nest-asyncio == 1.6.0
93105networkx == 3.4.2
94106nltk == 3.9.1
@@ -112,10 +124,12 @@ orjson==3.10.15
112124overrides == 7.7.0
113125packaging == 24.2
114126pandas == 2.2.3
127+ pandocfilters == 1.5.1
115128parso == 0.8.4
116129pathspec == 0.12.1
117130pexpect == 4.9.0
118131pillow == 11.1.0
132+ platformdirs == 4.5.1
119133posthog == 3.11.0
120134prompt-toolkit == 3.0.50
121135propcache == 0.2.1
@@ -135,11 +149,14 @@ python-dotenv==1.0.1
135149python-multipart == 0.0.20
136150pytz == 2025.1
137151pyyaml == 6.0.2
152+ pyzmq == 27.1.0
153+ referencing == 0.37.0
138154regex == 2024.11.6
139155requests == 2.32.3
140156requests-oauthlib == 2.0.0
141157rich == 13.9.4
142158rich-toolkit == 0.13.2
159+ rpds-py == 0.30.0
143160rsa == 4.9
144161safetensors == 0.5.3
145162shellingham == 1.5.4
@@ -153,7 +170,9 @@ striprtf==0.0.26
153170sympy == 1.13.3
154171tenacity == 9.0.0
155172tiktoken == 0.8.0
173+ tinycss2 == 1.4.0
156174tokenizers == 0.21.0
175+ tornado == 6.5.3
157176tqdm == 4.67.1
158177traitlets == 5.14.3
159178transformers == 4.51.3
@@ -172,9 +191,9 @@ uvloop==0.21.0
172191watchdog == 6.0.0
173192watchfiles == 1.0.4
174193wcwidth == 0.2.13
194+ webencodings == 0.5.1
175195websocket-client == 1.8.0
176196websockets == 14.2
177197wrapt == 1.17.2
178198yarl == 1.18.3
179199zipp == 3.21.0
180- nbconvert == 7.16.6
You can’t perform that action at this time.
0 commit comments