add blog to simple use cases

run-llama · Jan 20, 2025 · 578173f · 578173f
1 parent d5eac5d
commit 578173f
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 3 deletions.
diff --git a/helpers/datasources.ts b/helpers/datasources.ts
@@ -42,6 +42,16 @@ export const EXAMPLE_GDPR: TemplateDataSource = {
   },
 };
 
+export const AI_REPORTS: TemplateDataSource = {
+  type: "file",
+  config: {
+    url: new URL(
+      "https://www.europarl.europa.eu/RegData/etudes/ATAG/2024/760392/EPRS_ATA(2024)760392_EN.pdf",
+    ),
+    filename: "EPRS_ATA_2024_760392_EN.pdf",
+  },
+};
+
 export function getDataSources(
   files?: string,
   exampleFile?: boolean,

diff --git a/helpers/types.ts b/helpers/types.ts
@@ -54,7 +54,8 @@ export type TemplateUseCase =
   | "blog"
   | "form_filling"
   | "extractor"
-  | "contract_review";
+  | "contract_review"
+  | "blog";
 // Config for both file and folder
 export type FileSourceConfig =
   | {

diff --git a/questions/simple.ts b/questions/simple.ts
@@ -1,5 +1,6 @@
 import prompts from "prompts";
 import {
+  AI_REPORTS,
   EXAMPLE_10K_SEC_FILES,
   EXAMPLE_FILE,
   EXAMPLE_GDPR,
@@ -17,7 +18,8 @@ type AppType =
   | "form_filling"
   | "extractor"
   | "contract_review"
-  | "data_scientist";
+  | "data_scientist"
+  | "blog";
 
 type SimpleAnswers = {
   appType: AppType;
@@ -51,6 +53,7 @@ export const askSimpleQuestions = async (
           title: "Contract Review (using Workflows)",
           value: "contract_review",
         },
+        { title: "Blog Writer (using Workflows)", value: "blog" },
       ],
     },
     questionHandlers,
@@ -60,7 +63,11 @@ export const askSimpleQuestions = async (
   let llamaCloudKey = args.llamaCloudKey;
   let useLlamaCloud = false;
 
-  if (appType !== "extractor" && appType !== "contract_review") {
+  if (
+    appType !== "extractor" &&
+    appType !== "contract_review" &&
+    appType !== "blog"
+  ) {
     const { language: newLanguage } = await prompts(
       {
         type: "select",
@@ -188,6 +195,13 @@ const convertAnswers = async (
       frontend: false,
       dataSources: [EXAMPLE_GDPR],
     },
+    blog: {
+      template: "multiagent",
+      useCase: "blog",
+      tools: [],
+      frontend: true,
+      dataSources: [AI_REPORTS],
+    },
   };
   const results = lookup[answers.appType];
   return {