Spaces:
Paused
Paused
File size: 1,544 Bytes
a5784e9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | """
API utility function module
Contains utility functions for SSE generation, stream processing, token statistics, and request validation
(Refactored: logic moved to api_utils.utils_ext submodules)
"""
from .sse import generate_sse_stop_chunk
from .utils_ext import (
_extension_for_mime,
calculate_usage_stats,
clear_stream_queue,
collect_and_validate_attachments,
estimate_tokens,
extract_data_url_to_local,
extract_json_from_text,
get_latest_user_text,
maybe_execute_tools,
prepare_combined_prompt,
save_blob_to_local,
use_helper_get_response,
use_stream_response,
validate_chat_request,
)
# For backward compatibility with existing code that might import these private functions
_extract_json_from_text = extract_json_from_text
_get_latest_user_text = get_latest_user_text
def generate_sse_stop_chunk_with_usage(req_id: str, model: str, usage_stats: dict, reason: str = "stop") -> str:
"""Generate SSE stop chunk with usage statistics"""
return generate_sse_stop_chunk(req_id, model, reason, usage_stats)
__all__ = [
"generate_sse_stop_chunk_with_usage",
"extract_data_url_to_local",
"save_blob_to_local",
"collect_and_validate_attachments",
"prepare_combined_prompt",
"maybe_execute_tools",
"extract_json_from_text",
"get_latest_user_text",
"use_stream_response",
"clear_stream_queue",
"use_helper_get_response",
"validate_chat_request",
"estimate_tokens",
"calculate_usage_stats",
"_extension_for_mime",
]
|