crazywoola 1 месяц назад
Родитель
Сommit
878420463c
Аккаунт пользователя с таким Email не найден
2 измененных файлов: 4 добавлений и 2 удалений
  1. 2
    0
      api/controllers/web/message.py
  2. 2
    2
      api/services/message_service.py

+ 2
- 0
api/controllers/web/message.py Просмотреть файл

@@ -261,6 +261,8 @@ class MessageSuggestedQuestionApi(WebApiResource):
questions = MessageService.get_suggested_questions_after_answer(
app_model=app_model, user=end_user, message_id=message_id, invoke_from=InvokeFrom.WEB_APP
)
# questions is a list of strings, not a list of Message objects
# so we can directly return it
except MessageNotExistsError:
raise NotFound("Message not found")
except ConversationNotExistsError:

+ 2
- 2
api/services/message_service.py Просмотреть файл

@@ -217,7 +217,7 @@ class MessageService:
@classmethod
def get_suggested_questions_after_answer(
cls, app_model: App, user: Union[Account, EndUser] | None, message_id: str, invoke_from: InvokeFrom
) -> list[Message]:
) -> list[str]:
if not user:
raise ValueError("user cannot be None")

@@ -288,7 +288,7 @@ class MessageService:
)

with measure_time() as timer:
questions: list[Message] = LLMGenerator.generate_suggested_questions_after_answer(
questions: list[str] = LLMGenerator.generate_suggested_questions_after_answer(
tenant_id=app_model.tenant_id, histories=histories
)


Загрузка…
Отмена
Сохранить