chore(fern-bot): try/catch cohere calls (#1506)

fern-api · Sep 19, 2024 · 935fb87 · 935fb87
1 parent d57cadf
commit 935fb87
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 6 deletions.
diff --git a/servers/fern-bot/src/functions/oas-cron/shared/updateSpecInternal.ts b/servers/fern-bot/src/functions/oas-cron/shared/updateSpecInternal.ts
@@ -29,7 +29,7 @@ export async function updateSpecInternal(
         // Add + commit files
         const commitDiff = await git.diff();
         await git.add(["-A"]);
-        await git.commit(await generateCommitMessage(commitDiff));
+        await git.commit(await generateCommitMessage(commitDiff, "chore: update API specification"));
 
         // Push the changes
         await git.push([
@@ -45,7 +45,7 @@ export async function updateSpecInternal(
             {
                 title: ":herb: :sparkles: [Scheduled] Update API Spec",
                 base: "main",
-                body: await generateChangelog(fullDiff),
+                body: await generateChangelog(fullDiff, "This PR updates your API Definition to the latest version."),
             },
             repository.full_name,
             repository.full_name,

diff --git a/servers/fern-bot/src/libs/cohere.ts b/servers/fern-bot/src/libs/cohere.ts
@@ -13,7 +13,7 @@ async function coChat(prompt: string): Promise<string> {
     return response.text;
 }
 
-export async function generateCommitMessage(diff: string): Promise<string> {
+export async function generateCommitMessage(diff: string, fallbackMessage: string): Promise<string> {
     if (diff === "") {
         return DEFAULT_GITHUB_MESSAGE;
     }
@@ -25,11 +25,18 @@ export async function generateCommitMessage(diff: string): Promise<string> {
     ${diff}
     \`\`\`
     `;
+    try {
+        return await coChat(prompt);
+    } catch (error) {
+        console.error(
+            `Call to Cohere failed generating commit message, with error: ${(error as Error).message}, using fallback message: ${fallbackMessage}`,
+        );
 
-    return coChat(prompt);
+        return fallbackMessage;
+    }
 }
 
-export async function generateChangelog(diff: string): Promise<string> {
+export async function generateChangelog(diff: string, fallbackMessage: string): Promise<string> {
     if (diff === "") {
         return DEFAULT_GITHUB_MESSAGE;
     }
@@ -216,5 +223,13 @@ export async function generateChangelog(diff: string): Promise<string> {
         \`\`\`
     `;
 
-    return coChat(prompt);
+    try {
+        return await coChat(prompt);
+    } catch (error) {
+        console.error(
+            `Call to Cohere failed writing the PR body, with error: ${(error as Error).message}, using fallback message: ${fallbackMessage}`,
+        );
+
+        return fallbackMessage;
+    }
 }