fix: Fix header check

lobehub · Dec 15, 2023 · 5235c97 · 5235c97
1 parent cfcc085
commit 5235c97
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/scripts/autoSubmit.ts b/scripts/autoSubmit.ts
@@ -8,7 +8,7 @@ import { resolve } from 'node:path';
 
 import { formatAgentJSON } from './check';
 import { agentsDir, githubHomepage } from './const';
-import { getLocaleAgentFileName, writeJSON } from './utils';
+import {checkHeader, getLocaleAgentFileName, writeJSON} from './utils';
 
 const GENERATE_LABEL = '🤖 Agent PR';
 const SUCCESS_LABEL = '✅ Auto Check Pass';
@@ -206,7 +206,7 @@ class AutoSubmit {
     let currentValue = '';
 
     for (const line of lines) {
-      if (line.startsWith('###')) {
+      if (checkHeader(line)) {
         if (currentKey && currentValue) {
           json[currentKey] = currentValue.trim();
           currentValue = '';

diff --git a/scripts/utils.ts b/scripts/utils.ts
@@ -54,3 +54,20 @@ export const updateAwesomeReadme = (md: string, prompts: string): string => {
 
   return mds.join(readmeSplit);
 };
+
+export const checkHeader = (line: string) => {
+  const header = [
+    '### systemRole',
+    '### identifier',
+    '### avatar',
+    '### title',
+    '### description',
+    '### tags',
+    '### locale',
+  ]
+  let check = false;
+  header.forEach((item) => {
+    if (line.startsWith(item)) check = true;
+  })
+  return check;
+}