lit_ollama.store.local
[docs]
module
lit_ollama.store.local
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49 | from __future__ import annotations
from pathlib import Path
from lit_ollama.server.schema.base import TagModel
from lit_ollama.store.blobs import BlobStore
from lit_ollama.store.models import ModelStore
class LocalStore:
"""Local filesystem-backed storage used by the API.
This is intentionally a thin façade that composes the independent blob/model
stores so the API only has to wire up a single dependency.
"""
def __init__(self, *, blobs_root: str | Path = "blobs") -> None:
self._models = ModelStore()
self._blobs = BlobStore(root=blobs_root)
@property
def models(self) -> dict[str, TagModel]:
return self._models.models
@property
def blobs(self) -> BlobStore:
return self._blobs
# ModelStore passthroughs
def load_models(self) -> None:
self._models.load_models()
def get_file_hash(self, file_path: Path) -> str:
return self._models.get_file_hash(file_path)
def get_folder_hash(self, folder_path: Path) -> str:
return self._models.get_folder_hash(folder_path)
def get_model(self, model_name: str) -> dict:
return self._models.get_model(model_name)
def copy_model(self, model_name: str, new_model_name: str) -> None:
self._models.copy_model(model_name, new_model_name)
def delete_model(self, model_name: str) -> None:
self._models.delete_model(model_name)
def pull_model(self, model_name: str) -> None:
self._models.pull_model(model_name)
|