Added support for mongoDB KV store

llama_stack/providers/utils/kvstore/config.py

@@ -17,7 +17,8 @@ from llama_stack.distribution.utils.config_dirs import RUNTIME_BASE_DIR
 class KVStoreType(Enum):
     redis = "redis"
     sqlite = "sqlite"
-    postgres = "postgres"
+    postgres = "postgres",
+    mongodb = "mongodb"
 
 
 class CommonConfig(BaseModel):
@@ -55,15 +56,15 @@ class SqliteKVStoreConfig(CommonConfig):
 
     @classmethod
     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 {
             "type": "sqlite",
             "namespace": None,
             "db_path": "${env.SQLITE_STORE_DIR:~/.llama/"
-            + __distro_dir__
-            + "}/"
-            + db_name,
+                       + __distro_dir__
+                       + "}/"
+                       + db_name,
         }
 
 
@@ -106,7 +107,30 @@ class PostgresKVStoreConfig(CommonConfig):
         return v
 
 
+class MongoDBKVStoreConfig(CommonConfig):
+    type: Literal[KVStoreType.mongodb.value] = KVStoreType.mongodb.value
+    host: str = "localhost"
+    port: int = 5432
+    db: str = "llamastack"
+    user: str = None
+    password: Optional[str] = None
+    collection_name: str = "llamastack_kvstore"
+
+    @classmethod
+    def sample_run_config(cls, collection_name: str = "llamastack_kvstore"):
+        return {
+            "type": "mongodb",
+            "namespace": None,
+            "host": "${env.MONGODB_HOST:localhost}",
+            "port": "${env.MONGODB_PORT:5432}",
+            "db": "${env.MONGODB_DB}",
+            "user": "${env.MONGODB_USER}",
+            "password": "${env.MONGODB_PASSWORD}",
+            "table_name": "${env.MONGODB_COLLECTION_NAME:" + collection_name + "}",
+        }
+
+
 KVStoreConfig = Annotated[
-    Union[RedisKVStoreConfig, SqliteKVStoreConfig, PostgresKVStoreConfig],
+    Union[RedisKVStoreConfig, SqliteKVStoreConfig, PostgresKVStoreConfig, MongoDBKVStoreConfig],
     Field(discriminator="type", default=KVStoreType.sqlite.value),
 ]