Skip to content

Commit

Permalink
Add model context protocol tools with ollama provider
Browse files Browse the repository at this point in the history
Signed-off-by: Shreyanand <[email protected]>
  • Loading branch information
Shreyanand committed Feb 26, 2025
1 parent c8a20b8 commit c1294c5
Show file tree
Hide file tree
Showing 6 changed files with 10 additions and 1 deletion.
1 change: 1 addition & 0 deletions distributions/dependencies.json
Original file line number Diff line number Diff line change
Expand Up @@ -431,6 +431,7 @@
"fire",
"httpx",
"matplotlib",
"mcp",
"nltk",
"numpy",
"ollama",
Expand Down
2 changes: 1 addition & 1 deletion docs/source/distributions/self_hosted_distro/ollama.md
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ The `llamastack/distribution-ollama` distribution consists of the following prov
| safety | `inline::llama-guard` |
| scoring | `inline::basic`, `inline::llm-as-judge`, `inline::braintrust` |
| telemetry | `inline::meta-reference` |
| tool_runtime | `remote::brave-search`, `remote::tavily-search`, `inline::code-interpreter`, `inline::rag-runtime` |
| tool_runtime | `remote::brave-search`, `remote::tavily-search`, `remote::model-context-protocol`, `inline::code-interpreter`, `inline::rag-runtime` |
| vector_io | `inline::sqlite-vec`, `remote::chromadb`, `remote::pgvector` |


Expand Down
1 change: 1 addition & 0 deletions llama_stack/templates/ollama/build.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -28,4 +28,5 @@ distribution_spec:
- remote::tavily-search
- inline::code-interpreter
- inline::rag-runtime
- remote::model-context-protocol
image_type: conda
1 change: 1 addition & 0 deletions llama_stack/templates/ollama/ollama.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@ def get_distribution_template() -> DistributionTemplate:
"remote::tavily-search",
"inline::code-interpreter",
"inline::rag-runtime",
"remote::model-context-protocol"
],
}
name = "ollama"
Expand Down
3 changes: 3 additions & 0 deletions llama_stack/templates/ollama/run-with-safety.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -82,6 +82,9 @@ providers:
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}
- provider_id: model-context-protocol
provider_type: remote::model-context-protocol
config: {}
metadata_store:
type: sqlite
db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/ollama}/registry.db
Expand Down
3 changes: 3 additions & 0 deletions llama_stack/templates/ollama/run.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -79,6 +79,9 @@ providers:
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}
- provider_id: model-context-protocol
provider_type: remote::model-context-protocol
config: {}
metadata_store:
type: sqlite
db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/ollama}/registry.db
Expand Down

0 comments on commit c1294c5

Please sign in to comment.