Merge pull request #682 from githru/feat/579

[engine/AI] AI 요약 기능 고도화
githru · Aug 31, 2024 · 886be7a · 886be7a
2 parents ed73632 + d73a815
commit 886be7a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/packages/analysis-engine/.gitignore b/packages/analysis-engine/.gitignore
@@ -1,3 +1,4 @@
 node_modules
 dist
 coverage
+.env
diff --git a/packages/analysis-engine/src/index.ts b/packages/analysis-engine/src/index.ts
@@ -75,7 +75,8 @@ export class AnalysisEngine {
     if (this.isDebugMode) console.log("stemDict: ", stemDict);
     const csmDict = buildCSMDict(commitDict, stemDict, this.baseBranchName, pullRequests);
     if (this.isDebugMode) console.log("csmDict: ", csmDict);
-    const geminiCommitSummary = await getSummary(commitRaws?.slice(-10) ?? []);
+    const nodes = stemDict.get(this.baseBranchName)?.nodes?.map(({commit}) => commit);
+    const geminiCommitSummary = await getSummary(nodes ? nodes?.slice(-10) : []);
     if (this.isDebugMode) console.log("GeminiCommitSummary: ", geminiCommitSummary);
 
     return {

diff --git a/packages/analysis-engine/src/summary.ts b/packages/analysis-engine/src/summary.ts
@@ -4,9 +4,8 @@ const apiKey = process.env.GEMENI_API_KEY || '';
 const apiUrl = "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash-latest:generateContent?key=";
 
 export async function getSummary(csmNodes: CommitRaw[]) {
-  const commitMessages = csmNodes.map((csmNode) => csmNode.message).join(', ');
+  const commitMessages = csmNodes.map((csmNode) => csmNode.message.split('\n')[0]).join(', ');
 
-  console.log(commitMessages, 'apiKey');
   try {
     const response = await fetch(apiUrl + apiKey, {
       method: "POST",