Sfoglia il codice sorgente

fix: Message => str (#25876)

tags/1.9.0
crazywoola 1 mese fa
parent
commit
878420463c
Nessun account collegato all'indirizzo email del committer
2 ha cambiato i file con 4 aggiunte e 2 eliminazioni
  1. 2
    0
      api/controllers/web/message.py
  2. 2
    2
      api/services/message_service.py

+ 2
- 0
api/controllers/web/message.py Vedi File

@@ -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 Vedi File

@@ -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
)


Loading…
Annulla
Salva