LLM plugin providing access to models running on a local or remote Ollama server.
Install this plugin in the same environment as LLM.
llm install llm-ollama
First, ensure that your Ollama server is running and that you have pulled some models. You can use ollama list
to check what is locally available.
The plugin will query the Ollama server for the list of models. You can use llm ollama list-models
to see the list; it should be the same as output by ollama list
. All these models will be automatically registered with LLM and made available for prompting and chatting.
Assuming you have llama2:latest
available, you can run a prompt using:
llm -m llama2:latest 'How much is 2+2?'
The plugin automatically creates a short alias for models that have :latest
in the name, so the previous command is equivalent to running:
llm -m llama2 'How much is 2+2?'
To start an interactive chat session:
llm chat -m llama2
Chatting with llama2:latest
Type 'exit' or 'quit' to exit
Type '!multi' to enter multiple lines, then '!end' to finish
>
If your Ollama server is remote ensure you define the OLLAMA_HOST
environment variable so it is used instead of localhost
.
export OLLAMA_HOST=https://192.168.1.13:11434
The same Ollama model may be referred by several names with different tags. For example, in the following list, there is a single unique model with three different names:
ollama list
NAME ID SIZE MODIFIED
stable-code:3b aa5ab8afb862 1.6 GB 9 hours ago
stable-code:code aa5ab8afb862 1.6 GB 9 seconds ago
stable-code:latest aa5ab8afb862 1.6 GB 14 seconds ago
In such cases, the plugin will register a single model and create additional aliases. Continuing the previous example, this is what LLM will have:
llm models
...
Ollama: stable-code:3b (aliases: stable-code:code, stable-code:latest, stable-code)
All models accept Ollama modelfile parameters as options. Use the -o name value
syntax to specify them, for example:
-o temperature 0.8
: set the temperature of the model-o num_ctx 256000
: set the size of the context window used to generate the next token
See the referenced page for the complete list with descriptions and default values.
Additionally, the -o json_object 1
option can be used to force the model to reply with a valid JSON object. Note that your prompt must mention JSON for this to work.
If your Ollama server is not hosted at the default localhost:11434
address, you can use OLLAMA_HOST
environment variable to point the plugin to it.
To set up this plugin locally, first checkout the code. Then create a new virtual environment:
cd llm-ollama
python3 -m venv venv
source venv/bin/activate
Now install the dependencies and test dependencies:
pip install -e '.[test]'
To run the tests:
pytest