Skip to content

Commit 9b221b1

Browse files
authored
Merge pull request #11 from bmahabirbu/add-rag-embed
chore: updated llama-stack and dependencies with uv and added RAG sup…
2 parents 8d6a4a9 + ac181ac commit 9b221b1

File tree

7 files changed

+4339
-41
lines changed

7 files changed

+4339
-41
lines changed

.github/workflows/build-stack.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,8 @@ jobs:
3131

3232
- name: build package
3333
run: |
34-
pip install --upgrade build
35-
python -m build
34+
pip install --upgrade uv
35+
uv build
3636
3737
- name: Install qemu dependency
3838
run: |

build/Containerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ WORKDIR /opt/app-root/src
44

55
COPY dist dist
66

7-
RUN pip install ./dist/podman_ai_lab_stack-0.12.0.tar.gz
7+
RUN pip install --extra-index-url https://download.pytorch.org/whl/cpu/ ./dist/podman_ai_lab_stack-0.12.0.tar.gz
88
RUN \
99
mkdir -p $HOME/.llama/providers.d/remote/inference && \
1010
cp /opt/app-root/lib64/python3.11/site-packages/podman_ai_lab_stack/providers.d/remote/inference/podman-ai-lab.yaml \

pyproject.toml

Lines changed: 30 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,26 @@ name = "podman-ai-lab-stack"
77
version = "0.12.0"
88
description = "Podman AI Lab provider for Llama Stack"
99
requires-python = ">=3.10"
10-
dynamic = ["dependencies"]
10+
dependencies = [
11+
"aiosqlite>=0.21.0",
12+
"autoevals>=0.0.130",
13+
"blobfile>=3.1.0",
14+
"chardet>=5.2.0",
15+
"datasets>=4.1.1",
16+
"faiss-cpu>=1.12.0",
17+
"fastapi>=0.118.0",
18+
"greenlet>=3.2.4",
19+
"llama-stack>=0.2.9",
20+
"mcp>=1.15.0",
21+
"ollama>=0.6.0",
22+
"opentelemetry-api>=1.37.0",
23+
"opentelemetry-exporter-otlp>=1.37.0",
24+
"opentelemetry-sdk>=1.37.0",
25+
"pydantic>=2.11.9",
26+
"sentence-transformers>=5.1.1",
27+
"sqlalchemy>=2.0.43",
28+
"torch>=2.7.0",
29+
]
1130

1231
[tool.setuptools]
1332
package-dir = { "" = "src" }
@@ -21,3 +40,13 @@ dependencies = { file = ["requirements.txt"] }
2140

2241
[tool.ruff]
2342
extend-exclude = ["*.ipynb"]
43+
44+
[[tool.uv.index]]
45+
name = "pytorch-cpu"
46+
url = "https://download.pytorch.org/whl/cpu"
47+
explicit = true
48+
49+
[tool.uv.sources]
50+
torch = [
51+
{ index = "pytorch-cpu" },
52+
]

0 commit comments

Comments
 (0)