removed extra spaces and fix collection_name

This commit is contained in:
shrinitgoyal 2024-11-28 22:06:49 +05:30
parent c2f3a7f87e
commit fdda9f3098

View file

@ -56,15 +56,15 @@ class SqliteKVStoreConfig(CommonConfig):
@classmethod @classmethod
def sample_run_config( def sample_run_config(
cls, __distro_dir__: str = "runtime", db_name: str = "kvstore.db" cls, __distro_dir__: str = "runtime", db_name: str = "kvstore.db"
): ):
return { return {
"type": "sqlite", "type": "sqlite",
"namespace": None, "namespace": None,
"db_path": "${env.SQLITE_STORE_DIR:~/.llama/" "db_path": "${env.SQLITE_STORE_DIR:~/.llama/"
+ __distro_dir__ + __distro_dir__
+ "}/" + "}/"
+ db_name, + db_name,
} }
@ -126,7 +126,7 @@ class MongoDBKVStoreConfig(CommonConfig):
"db": "${env.MONGODB_DB}", "db": "${env.MONGODB_DB}",
"user": "${env.MONGODB_USER}", "user": "${env.MONGODB_USER}",
"password": "${env.MONGODB_PASSWORD}", "password": "${env.MONGODB_PASSWORD}",
"table_name": "${env.MONGODB_COLLECTION_NAME:" + collection_name + "}", "collection_name": "${env.MONGODB_COLLECTION_NAME:" + collection_name + "}",
} }