Closed bricefotzo closed 3 weeks ago
All modified and coverable lines are covered by tests :white_check_mark:
Please upload report for BASE (
main@ead0f79
). Learn more about missing BASE report.:exclamation: Current head 3ab8f36 differs from pull request most recent head fb7259c
Please upload reports for the commit fb7259c to get more accurate results.
:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.
Added a new class OllamaContainer with few methods to handle the Ollama container.
The
_check_and_add_gpu_capabilities
method checks if the host has GPUs and adds the necessary capabilities to the container.The
commit_to_image
allows to save somehow the state of a container into an image so that we can reuse it, especially for the ones having some models pulled.Added tests to check the functionality of the new class.
Fixes #617