This commit is contained in:
Raghotham Murthy 2025-05-28 17:09:10 -07:00
parent 5a422e236c
commit fb47cc0931
4 changed files with 32 additions and 18 deletions

View file

@ -432,11 +432,9 @@
"ollama": [ "ollama": [
"aiohttp", "aiohttp",
"aiosqlite", "aiosqlite",
"autoevals",
"blobfile", "blobfile",
"chardet", "chardet",
"chromadb-client", "chromadb-client",
"datasets",
"emoji", "emoji",
"faiss-cpu", "faiss-cpu",
"fastapi", "fastapi",
@ -446,27 +444,19 @@
"matplotlib", "matplotlib",
"mcp", "mcp",
"nltk", "nltk",
"numpy",
"ollama", "ollama",
"openai", "openai",
"opentelemetry-exporter-otlp-proto-http", "opentelemetry-exporter-otlp-proto-http",
"opentelemetry-sdk", "opentelemetry-sdk",
"pandas",
"peft", "peft",
"pillow", "pillow",
"psycopg2-binary", "psycopg2-binary",
"pymongo", "pymongo",
"pypdf", "pypdf",
"pythainlp", "pythainlp",
"redis",
"requests", "requests",
"scikit-learn",
"scipy",
"sentencepiece",
"sqlalchemy[asyncio]", "sqlalchemy[asyncio]",
"torch",
"tqdm", "tqdm",
"transformers",
"tree_sitter", "tree_sitter",
"trl", "trl",
"uvicorn" "uvicorn"

View file

@ -5,9 +5,7 @@ distribution_spec:
inference: inference:
- remote::ollama - remote::ollama
vector_io: vector_io:
- inline::faiss
- remote::chromadb - remote::chromadb
- remote::pgvector
safety: safety:
- inline::llama-guard - inline::llama-guard
agents: agents:
@ -17,21 +15,33 @@ distribution_spec:
eval: eval:
- inline::meta-reference - inline::meta-reference
datasetio: datasetio:
- remote::huggingface
- inline::localfs - inline::localfs
scoring: scoring:
- inline::basic - inline::basic
- inline::llm-as-judge - inline::llm-as-judge
- inline::braintrust
post_training:
- inline::huggingface
tool_runtime: tool_runtime:
- remote::brave-search - remote::brave-search
- remote::tavily-search - remote::tavily-search
- inline::rag-runtime - inline::rag-runtime
- remote::model-context-protocol - remote::model-context-protocol
- remote::wolfram-alpha - remote::wolfram-alpha
image_type: conda metadata_store:
type: sqlite
db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/ollama}/registry.db
inference_store:
type: sqlite
db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/ollama}/inference_store.db
models:
- metadata: {}
model_id: ${env.INFERENCE_MODEL}
provider_id: remote::ollama
model_type: llm
- metadata:
embedding_dimension: 384
model_id: all-MiniLM-L6-v2
provider_id: remote::ollama
provider_model_id: all-minilm:latest
model_type: embedding
image_type: container
additional_pip_packages: additional_pip_packages:
- sqlalchemy[asyncio] - sqlalchemy[asyncio]
- sqlalchemy[asyncio]

View file

@ -138,6 +138,7 @@ explicit = true
[tool.uv.sources] [tool.uv.sources]
torch = [{ index = "pytorch-cpu" }] torch = [{ index = "pytorch-cpu" }]
torchvision = [{ index = "pytorch-cpu" }] torchvision = [{ index = "pytorch-cpu" }]
llama-stack = { workspace = true }
[tool.ruff] [tool.ruff]
line-length = 120 line-length = 120
@ -331,5 +332,10 @@ init_forbid_extra = true
init_typed = true init_typed = true
warn_required_dynamic_aliases = true warn_required_dynamic_aliases = true
[dependency-groups]
dev = [
"llama-stack",
]
[tool.ruff.lint.pep8-naming] [tool.ruff.lint.pep8-naming]
classmethod-decorators = ["classmethod", "pydantic.field_validator"] classmethod-decorators = ["classmethod", "pydantic.field_validator"]

8
uv.lock generated
View file

@ -1553,6 +1553,11 @@ unit = [
{ name = "sqlite-vec" }, { name = "sqlite-vec" },
] ]
[package.dev-dependencies]
dev = [
{ name = "llama-stack" },
]
[package.metadata] [package.metadata]
requires-dist = [ requires-dist = [
{ name = "aiohttp", marker = "extra == 'test'" }, { name = "aiohttp", marker = "extra == 'test'" },
@ -1639,6 +1644,9 @@ requires-dist = [
] ]
provides-extras = ["dev", "unit", "test", "docs", "codegen", "ui"] provides-extras = ["dev", "unit", "test", "docs", "codegen", "ui"]
[package.metadata.requires-dev]
dev = [{ name = "llama-stack", editable = "." }]
[[package]] [[package]]
name = "llama-stack-client" name = "llama-stack-client"
version = "0.2.7" version = "0.2.7"