Fix sqlite_vec config defaults

This commit is contained in:
Ashwin Bharambe 2025-02-20 17:50:24 -08:00
parent 832c535aaf
commit 35ae0e16a1
2 changed files with 2 additions and 15 deletions

View file

@ -4,26 +4,16 @@
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
# config.py
from typing import Any, Dict
from pydantic import BaseModel
from llama_stack.providers.utils.kvstore.config import (
KVStoreConfig,
SqliteKVStoreConfig,
)
class SQLiteVectorIOConfig(BaseModel):
db_path: str
kvstore: KVStoreConfig
@classmethod
def sample_run_config(cls, __distro_dir__: str) -> Dict[str, Any]:
return {
"kvstore": SqliteKVStoreConfig.sample_run_config(
__distro_dir__=__distro_dir__,
db_name="sqlite_vec.db",
)
"db_path": "${env.SQLITE_STORE_DIR:~/.llama/" + __distro_dir__ + "}/" + "sqlite_vec.db",
}

View file

@ -30,9 +30,6 @@ providers:
- provider_id: sqlite_vec
provider_type: inline::sqlite_vec
config:
kvstore:
type: sqlite
namespace: null
db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/ollama}/sqlite_vec.db
safety:
- provider_id: llama-guard