mirror of
https://github.com/zulip/zulip.git
synced 2025-11-07 07:23:22 +00:00
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
import warnings
|
|
|
|
warnings.filterwarnings("ignore", category=UserWarning, module="pydantic")
|
|
|
|
|
|
from django.conf import settings
|
|
from django.http import HttpRequest, HttpResponse
|
|
from django.utils.translation import gettext as _
|
|
from pydantic import Json
|
|
|
|
from zerver.actions.message_summary import do_summarize_narrow
|
|
from zerver.lib.exceptions import JsonableError
|
|
from zerver.lib.narrow import NarrowParameter
|
|
from zerver.lib.response import json_success
|
|
from zerver.lib.typed_endpoint import typed_endpoint
|
|
from zerver.models import UserProfile
|
|
|
|
|
|
@typed_endpoint
|
|
def get_messages_summary(
|
|
request: HttpRequest,
|
|
user_profile: UserProfile,
|
|
*,
|
|
narrow: Json[list[NarrowParameter] | None] = None,
|
|
) -> HttpResponse:
|
|
if settings.TOPIC_SUMMARIZATION_MODEL is None: # nocoverage
|
|
raise JsonableError(_("AI features are not enabled on this server."))
|
|
|
|
if not (user_profile.is_moderator or user_profile.is_realm_admin): # nocoverage
|
|
return json_success(request, {"summary": "Feature limited to moderators for now."})
|
|
|
|
summary = do_summarize_narrow(user_profile, narrow)
|
|
if summary is None: # nocoverage
|
|
return json_success(request, {"summary": "No messages in conversation to summarize"})
|
|
|
|
return json_success(request, {"summary": summary})
|