Skip to content

Commit a6cedd4

Browse files
authored
Merge pull request #406 from Quickchive/feat/SCRUM-188/change-recommend-model
feat: 카테고리 추천 모델 변경
2 parents 0d4d7ef + ba2b4ad commit a6cedd4

File tree

1 file changed

+3
-11
lines changed

1 file changed

+3
-11
lines changed

src/categories/category.service.ts

Lines changed: 3 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -15,23 +15,15 @@ import {
1515
UpdateCategoryBodyDto,
1616
UpdateCategoryOutput,
1717
} from './dtos/category.dto';
18-
import {
19-
LoadFrequentCategoriesOutput,
20-
LoadPersonalCategoriesOutput,
21-
} from './dtos/load-personal-categories.dto';
18+
import { LoadFrequentCategoriesOutput, LoadPersonalCategoriesOutput, } from './dtos/load-personal-categories.dto';
2219
import { Category } from './category.entity';
2320
import { Content } from '../contents/entities/content.entity';
2421
import { CategoryRepository } from './category.repository';
2522
import { ContentRepository } from '../contents/repository/content.repository';
2623
import { getLinkContent, getLinkInfo } from '../contents/util/content.util';
2724
import { User } from '../users/entities/user.entity';
2825
import { UserRepository } from '../users/repository/user.repository';
29-
import {
30-
generateCategoriesTree,
31-
generateSlug,
32-
loadLogs,
33-
makeCategoryListWithSaveCount,
34-
} from './utils/category.util';
26+
import { generateCategoriesTree, generateSlug, loadLogs, makeCategoryListWithSaveCount, } from './utils/category.util';
3527
import { Transactional } from '../common/aop/transactional';
3628
import { AiService } from '../ai/ai.service';
3729

@@ -477,7 +469,7 @@ Present your reply options in JSON format below.
477469

478470
try {
479471
const categoryStr = await this.aiService.chat({
480-
model: 'llama3-8b-8192',
472+
model: 'llama-3.1-8b-instant',
481473
messages: [{ role: 'user', content: question }],
482474
temperature: 0,
483475
responseType: 'json_object',

0 commit comments

Comments
 (0)