Merge pull request #179 from boostcampwm2023/hotfix/sentiment-save-error

[Hotfix] 감정 분석 결과 저장 오류 수정
boostcampwm2023 · Nov 30, 2023 · 073163d · 073163d
2 parents f659250 + b252422
commit 073163d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/BE/src/diaries/diaries.repository.ts b/BE/src/diaries/diaries.repository.ts
@@ -28,7 +28,7 @@ export class DiariesRepository {
     const positiveRatio = sentimentResult.positiveRatio;
     const negativeRatio = sentimentResult.negativeRatio;
     const neutralRatio = sentimentResult.neutralRatio;
-    const sentiment = sentimentStatus[sentimentResult.sentiment];
+    const sentiment = sentimentResult.sentiment;
 
     const newDiary = Diary.create({
       title,

diff --git a/BE/src/diaries/diaries.service.ts b/BE/src/diaries/diaries.service.ts
@@ -31,8 +31,9 @@ export class DiariesService {
     const shape = await this.shapesRepository.getShapeByUuid(shapeUuid);
     const encryptedContent = this.getEncryptedContent(content);
     const tagEntities = await this.getTags(tags);
-    const sentimentResult = await this.getSentiment(content);
+    const sentimentResult: SentimentDto = await this.getSentiment(content);
 
+    console.log(sentimentResult);
     const diary = await this.diariesRepository.createDiary(
       createDiaryDto,
       encryptedContent,
@@ -167,11 +168,12 @@ export class DiariesService {
     const sentiment = (await sentimentResponse).data.document.sentiment;
 
     const result: SentimentDto = {
-      positiveRatio,
-      neutralRatio,
-      negativeRatio,
-      sentiment,
+      positiveRatio: positiveRatio,
+      neutralRatio: neutralRatio,
+      negativeRatio: negativeRatio,
+      sentiment: sentimentStatus[sentiment.toUpperCase()],
     };
+
     return result;
   }