Skip to content

Commit f663865

Browse files
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>
1 parent 7608dce commit f663865

File tree

3 files changed

+29
-2
lines changed

3 files changed

+29
-2
lines changed

lua/avante/config.lua

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff 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

lua/avante/rag_service.lua

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,13 @@ local M = {}
88
local container_name = "avante-rag-service"
99
local 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

1319
function M.get_rag_service_port() return 20250 end
1420

py/rag-service/requirements.txt

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ attrs==25.1.0
99
backoff==2.2.1
1010
bcrypt==4.2.1
1111
beautifulsoup4==4.13.3
12+
bleach==6.3.0
1213
build==1.2.2.post1
1314
cachetools==5.5.1
1415
certifi==2024.12.14
@@ -20,6 +21,7 @@ coloredlogs==15.0.1
2021
dashscope==1.22.2
2122
dataclasses-json==0.6.7
2223
decorator==5.1.1
24+
defusedxml==0.7.1
2325
deprecated==1.2.18
2426
dirtyjson==1.0.8
2527
distro==1.9.0
@@ -31,6 +33,7 @@ et-xmlfile==2.0.0
3133
executing==2.2.0
3234
fastapi==0.115.8
3335
fastapi-cli==0.0.7
36+
fastjsonschema==2.21.2
3437
filelock==3.17.0
3538
filetype==1.2.0
3639
flatbuffers==25.1.24
@@ -55,6 +58,11 @@ jedi==0.19.2
5558
jinja2==3.1.5
5659
jiter==0.8.2
5760
joblib==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
5866
kubernetes==32.0.0
5967
llama-cloud==0.1.11
6068
llama-cloud-services==0.6.0
@@ -84,11 +92,15 @@ markupsafe==3.0.2
8492
marshmallow==3.26.1
8593
matplotlib-inline==0.1.7
8694
mdurl==0.1.2
95+
mistune==3.1.4
8796
mmh3==5.1.0
8897
monotonic==1.6
8998
mpmath==1.3.0
9099
multidict==6.1.0
91100
mypy-extensions==1.0.0
101+
nbclient==0.10.2
102+
nbconvert==7.16.6
103+
nbformat==5.10.4
92104
nest-asyncio==1.6.0
93105
networkx==3.4.2
94106
nltk==3.9.1
@@ -112,10 +124,12 @@ orjson==3.10.15
112124
overrides==7.7.0
113125
packaging==24.2
114126
pandas==2.2.3
127+
pandocfilters==1.5.1
115128
parso==0.8.4
116129
pathspec==0.12.1
117130
pexpect==4.9.0
118131
pillow==11.1.0
132+
platformdirs==4.5.1
119133
posthog==3.11.0
120134
prompt-toolkit==3.0.50
121135
propcache==0.2.1
@@ -135,11 +149,14 @@ python-dotenv==1.0.1
135149
python-multipart==0.0.20
136150
pytz==2025.1
137151
pyyaml==6.0.2
152+
pyzmq==27.1.0
153+
referencing==0.37.0
138154
regex==2024.11.6
139155
requests==2.32.3
140156
requests-oauthlib==2.0.0
141157
rich==13.9.4
142158
rich-toolkit==0.13.2
159+
rpds-py==0.30.0
143160
rsa==4.9
144161
safetensors==0.5.3
145162
shellingham==1.5.4
@@ -153,7 +170,9 @@ striprtf==0.0.26
153170
sympy==1.13.3
154171
tenacity==9.0.0
155172
tiktoken==0.8.0
173+
tinycss2==1.4.0
156174
tokenizers==0.21.0
175+
tornado==6.5.3
157176
tqdm==4.67.1
158177
traitlets==5.14.3
159178
transformers==4.51.3
@@ -172,9 +191,9 @@ uvloop==0.21.0
172191
watchdog==6.0.0
173192
watchfiles==1.0.4
174193
wcwidth==0.2.13
194+
webencodings==0.5.1
175195
websocket-client==1.8.0
176196
websockets==14.2
177197
wrapt==1.17.2
178198
yarl==1.18.3
179199
zipp==3.21.0
180-
nbconvert==7.16.6

0 commit comments

Comments
 (0)