use GPT4o model for data scientist and code artifact

Author: marcusschiesser

questions/simple.ts

@@ -18,7 +18,6 @@ type SimpleAnswers = {
   language: TemplateFramework;
   useLlamaCloud: boolean;
   llamaCloudKey?: string;
-  modelConfig: ModelConfig;
 };
 
 export const askSimpleQuestions = async (
@@ -87,28 +86,36 @@ export const askSimpleQuestions = async (
     }
   }
 
-  const modelConfig = await askModelConfig({
-    openAiKey: args.openAiKey,
-    askModels: args.askModels ?? false,
-    framework: language,
-  });
-
-  const results = convertAnswers({
+  const results = await convertAnswers(args, {
     appType,
     language,
     useLlamaCloud,
     llamaCloudKey,
-    modelConfig,
   });
 
   results.postInstallAction = await askPostInstallAction(results);
   return results;
 };
 
-const convertAnswers = (answers: SimpleAnswers): QuestionResults => {
+const convertAnswers = async (
+  args: PureQuestionArgs,
+  answers: SimpleAnswers,
+): Promise<QuestionResults> => {
+  const MODEL_GPT4o: ModelConfig = {
+    provider: "openai",
+    apiKey: args.openAiKey,
+    model: "gpt-4o",
+    embeddingModel: "text-embedding-3-large",
+    dimensions: 1536,
+    isConfigured(): boolean {
+      return !!args.openAiKey;
+    },
+  };
   const lookup: Record<
     AppType,
-    Pick<QuestionResults, "template" | "tools" | "frontend" | "dataSources">
+    Pick<QuestionResults, "template" | "tools" | "frontend" | "dataSources"> & {
+      modelConfig?: ModelConfig;
+    }
   > = {
     rag: {
       template: "streaming",
@@ -121,12 +128,14 @@ const convertAnswers = (answers: SimpleAnswers): QuestionResults => {
       tools: getTools(["interpreter", "document_generator"]),
       frontend: true,
       dataSources: [],
+      modelConfig: MODEL_GPT4o,
     },
     code_artifact: {
       template: "streaming",
       tools: getTools(["artifact"]),
       frontend: true,
       dataSources: [],
+      modelConfig: MODEL_GPT4o,
     },
     multiagent: {
       template: "multiagent",
@@ -153,11 +162,16 @@ const convertAnswers = (answers: SimpleAnswers): QuestionResults => {
     llamaCloudKey: answers.llamaCloudKey,
     useLlamaParse: answers.useLlamaCloud,
     llamapack: "",
-    postInstallAction: "none",
     vectorDb: answers.useLlamaCloud ? "llamacloud" : "none",
-    modelConfig: answers.modelConfig,
     observability: "none",
     ...results,
+    modelConfig:
+      results.modelConfig ??
+      (await askModelConfig({
+        openAiKey: args.openAiKey,
+        askModels: args.askModels ?? false,
+        framework: answers.language,
+      })),
     frontend: answers.language === "nextjs" ? false : results.frontend,
   };
 };