mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-04 05:12:38 +02:00
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from fastapi.responses import StreamingResponse
|
|
from pydantic import BaseModel, Field
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.db import User, get_async_session
|
|
from app.services.new_streaming_service import VercelStreamingService
|
|
from app.services.vision_autocomplete_service import stream_vision_autocomplete
|
|
from app.users import current_active_user
|
|
from app.utils.rbac import check_search_space_access
|
|
|
|
router = APIRouter(prefix="/autocomplete", tags=["autocomplete"])
|
|
|
|
MAX_SCREENSHOT_SIZE = 20 * 1024 * 1024 # 20 MB base64 ceiling
|
|
|
|
|
|
class VisionAutocompleteRequest(BaseModel):
|
|
screenshot: str = Field(..., max_length=MAX_SCREENSHOT_SIZE)
|
|
search_space_id: int
|
|
app_name: str = ""
|
|
window_title: str = ""
|
|
|
|
|
|
@router.post("/vision/stream")
|
|
async def vision_autocomplete_stream(
|
|
body: VisionAutocompleteRequest,
|
|
user: User = Depends(current_active_user),
|
|
session: AsyncSession = Depends(get_async_session),
|
|
):
|
|
await check_search_space_access(session, user, body.search_space_id)
|
|
|
|
return StreamingResponse(
|
|
stream_vision_autocomplete(
|
|
body.screenshot, body.search_space_id, session,
|
|
app_name=body.app_name, window_title=body.window_title,
|
|
),
|
|
media_type="text/event-stream",
|
|
headers={
|
|
**VercelStreamingService.get_response_headers(),
|
|
"X-Accel-Buffering": "no",
|
|
},
|
|
)
|