diff --git a/questions.ts b/questions.ts
index d37096011ff94eb97615f4e5d761ce537c836aac..72c1e07583ec61ead0a4aefdea385858a4d0dea8 100644
--- a/questions.ts
+++ b/questions.ts
@@ -239,28 +239,24 @@ export const askQuestions = async (
     if (ciInfo.isCI) {
       program.vectorDb = getPrefOrDefault("vectorDb");
     } else {
-      if (program.framework === "fastapi") {
-        program.vectorDb = "none";
-      } else {
-        const { vectorDb } = await prompts(
-          {
-            type: "select",
-            name: "vectorDb",
-            message: "Would you like to use a vector database?",
-            choices: [
-              {
-                title: "No, just store the data in the file system",
-                value: "none",
-              },
-              { title: "MongoDB", value: "mongo" },
-            ],
-            initial: 0,
-          },
-          handlers,
-        );
-        program.vectorDb = vectorDb;
-        preferences.vectorDb = vectorDb;
-      }
+      const { vectorDb } = await prompts(
+        {
+          type: "select",
+          name: "vectorDb",
+          message: "Would you like to use a vector database?",
+          choices: [
+            {
+              title: "No, just store the data in the file system",
+              value: "none",
+            },
+            { title: "MongoDB", value: "mongo" },
+          ],
+          initial: 0,
+        },
+        handlers,
+      );
+      program.vectorDb = vectorDb;
+      preferences.vectorDb = vectorDb;
     }
   }
 
diff --git a/templates/index.ts b/templates/index.ts
index ff9ed0ae4af4d9ab237c970a91f6d2ea2a292cde..169dd2ec6b3d09ce8dc7d3bb85e0af43fc45b857 100644
--- a/templates/index.ts
+++ b/templates/index.ts
@@ -317,7 +317,11 @@ const installPythonTemplate = async ({
   template,
   framework,
   engine,
-}: Pick<InstallTemplateArgs, "root" | "framework" | "template" | "engine">) => {
+  vectorDb,
+}: Pick<
+  InstallTemplateArgs,
+  "root" | "framework" | "template" | "engine" | "vectorDb"
+>) => {
   console.log("\nInitializing Python project with template:", template, "\n");
   const templatePath = path.join(__dirname, "types", template, framework);
   await copy("**", root, {
@@ -342,7 +346,12 @@ const installPythonTemplate = async ({
 
   if (engine === "context") {
     const compPath = path.join(__dirname, "components");
-    const VectorDBPath = path.join(compPath, "vectordbs", "python", "none");
+    const VectorDBPath = path.join(
+      compPath,
+      "vectordbs",
+      "python",
+      vectorDb || "none",
+    );
     await copy("**", path.join(root, "app", "engine"), {
       parents: true,
       cwd: VectorDBPath,