diff --git a/main.py b/main.py index 4aaa0db..c212533 100644 --- a/main.py +++ b/main.py @@ -30,8 +30,7 @@ from pytchat.processors.default.processor import Chat from aques import Aques from common_module import CommonModule from nizika_ai.config import DB -from nizika_ai.consts import (AnswerType, Character, GPTModel, Platform, - QueryType) +from nizika_ai.consts import Character, GPTModel, Platform, QueryType from nizika_ai.models import Answer, AnsweredFlag, Query, User pygame.init () @@ -71,7 +70,7 @@ def main ( if answer_flags: answer_flag = random.choice (answer_flags) answer = Answer.find (answer_flag.answer_id) - if answer.answer_type == AnswerType.YOUTUBE_REPLY.value: + if answer.query_ref.query_type == QueryType.YOUTUBE_COMMENT.value: query = Query.find (answer.query_id) deerjika.talk (query.content, answer.content) answer_flag.answered = True diff --git a/nizika_ai b/nizika_ai index 9e136a7..3be6d90 160000 --- a/nizika_ai +++ b/nizika_ai @@ -1 +1 @@ -Subproject commit 9e136a7cb3baf97b384a07c4af31205d745e7200 +Subproject commit 3be6d9063c987deaceee24a1d16296d21319778c