Skip to content

Commit 8fdf640

Browse files
feat: use llamaparse if using llamacloud
1 parent 8328240 commit 8fdf640

File tree

1 file changed

+28
-22
lines changed

1 file changed

+28
-22
lines changed

questions.ts

Lines changed: 28 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -611,39 +611,45 @@ export const askQuestions = async (
611611
type: "llamacloud",
612612
config: {},
613613
});
614+
program.dataSources.push(EXAMPLE_FILE);
614615
break;
615616
}
616617
}
617618
}
618619
}
619620
}
620621

621-
// Asking for LlamaParse if user selected file data source
622-
if (program.dataSources.some((ds) => ds.type === "file")) {
623-
if (ciInfo.isCI) {
624-
program.useLlamaParse = getPrefOrDefault("useLlamaParse");
625-
} else {
626-
const { useLlamaParse } = await prompts(
627-
{
628-
type: "toggle",
629-
name: "useLlamaParse",
630-
message:
631-
"Would you like to use LlamaParse (improved parser for RAG - requires API key)?",
632-
initial: false,
633-
active: "yes",
634-
inactive: "no",
635-
},
636-
questionHandlers,
637-
);
638-
program.useLlamaParse = useLlamaParse;
639-
preferences.useLlamaParse = useLlamaParse;
640-
}
641-
}
642-
643622
const isUsingLlamaCloud = program.dataSources.some(
644623
(ds) => ds.type === "llamacloud",
645624
);
646625

626+
// Asking for LlamaParse if user selected file data source
627+
if (isUsingLlamaCloud) {
628+
// default to use LlamaParse if using LlamaCloud
629+
program.useLlamaParse = preferences.useLlamaParse = true;
630+
} else {
631+
if (program.dataSources.some((ds) => ds.type === "file")) {
632+
if (ciInfo.isCI) {
633+
program.useLlamaParse = getPrefOrDefault("useLlamaParse");
634+
} else {
635+
const { useLlamaParse } = await prompts(
636+
{
637+
type: "toggle",
638+
name: "useLlamaParse",
639+
message:
640+
"Would you like to use LlamaParse (improved parser for RAG - requires API key)?",
641+
initial: false,
642+
active: "yes",
643+
inactive: "no",
644+
},
645+
questionHandlers,
646+
);
647+
program.useLlamaParse = useLlamaParse;
648+
preferences.useLlamaParse = useLlamaParse;
649+
}
650+
}
651+
}
652+
647653
// Ask for LlamaCloud API key when using a LlamaCloud index or LlamaParse
648654
if (isUsingLlamaCloud || program.useLlamaParse) {
649655
if (ciInfo.isCI) {

0 commit comments

Comments
 (0)