disable multiple data sources for ts version

run-llama · Mar 25, 2024 · 3df53d6 · 3df53d6
1 parent 8b8f479
commit 3df53d6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/questions.ts b/questions.ts
@@ -659,7 +659,8 @@ export const askQuestions = async (
       program.dataSources = getPrefOrDefault("dataSources");
     } else {
       program.dataSources = [];
-      for (let i = 0; i < 2; i++) {
+      const numberDataSources = program.framework === "fastapi" ? 2 : 1;
+      for (let i = 0; i < numberDataSources; i++) {
         const { selectedSource } = await prompts(
           {
             type: "select",