mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-09 15:52:40 +02:00
6648409237
- Added `citations_enabled` and `qna_custom_instructions` fields to the SearchSpace model for better QnA configuration. - Updated the creation and update schemas to handle new fields with appropriate defaults. - Refactored QnA handling in the agent to utilize the new SearchSpace fields for improved response customization. - Adjusted UI components to include settings for managing QnA configurations. - Enhanced onboarding process to incorporate prompt setup as an optional step.
125 lines
4 KiB
Python
125 lines
4 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from sqlalchemy.future import select
|
|
|
|
from app.db import SearchSpace, User, get_async_session
|
|
from app.schemas import SearchSpaceCreate, SearchSpaceRead, SearchSpaceUpdate
|
|
from app.users import current_active_user
|
|
from app.utils.check_ownership import check_ownership
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post("/searchspaces", response_model=SearchSpaceRead)
|
|
async def create_search_space(
|
|
search_space: SearchSpaceCreate,
|
|
session: AsyncSession = Depends(get_async_session),
|
|
user: User = Depends(current_active_user),
|
|
):
|
|
try:
|
|
search_space_data = search_space.model_dump()
|
|
|
|
# citations_enabled defaults to True (handled by Pydantic schema)
|
|
# qna_custom_instructions defaults to None/empty (handled by DB)
|
|
|
|
db_search_space = SearchSpace(**search_space_data, user_id=user.id)
|
|
session.add(db_search_space)
|
|
await session.commit()
|
|
await session.refresh(db_search_space)
|
|
return db_search_space
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
await session.rollback()
|
|
raise HTTPException(
|
|
status_code=500, detail=f"Failed to create search space: {e!s}"
|
|
) from e
|
|
|
|
|
|
@router.get("/searchspaces", response_model=list[SearchSpaceRead])
|
|
async def read_search_spaces(
|
|
skip: int = 0,
|
|
limit: int = 200,
|
|
session: AsyncSession = Depends(get_async_session),
|
|
user: User = Depends(current_active_user),
|
|
):
|
|
try:
|
|
result = await session.execute(
|
|
select(SearchSpace)
|
|
.filter(SearchSpace.user_id == user.id)
|
|
.offset(skip)
|
|
.limit(limit)
|
|
)
|
|
return result.scalars().all()
|
|
except Exception as e:
|
|
raise HTTPException(
|
|
status_code=500, detail=f"Failed to fetch search spaces: {e!s}"
|
|
) from e
|
|
|
|
|
|
@router.get("/searchspaces/{search_space_id}", response_model=SearchSpaceRead)
|
|
async def read_search_space(
|
|
search_space_id: int,
|
|
session: AsyncSession = Depends(get_async_session),
|
|
user: User = Depends(current_active_user),
|
|
):
|
|
try:
|
|
search_space = await check_ownership(
|
|
session, SearchSpace, search_space_id, user
|
|
)
|
|
return search_space
|
|
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
raise HTTPException(
|
|
status_code=500, detail=f"Failed to fetch search space: {e!s}"
|
|
) from e
|
|
|
|
|
|
@router.put("/searchspaces/{search_space_id}", response_model=SearchSpaceRead)
|
|
async def update_search_space(
|
|
search_space_id: int,
|
|
search_space_update: SearchSpaceUpdate,
|
|
session: AsyncSession = Depends(get_async_session),
|
|
user: User = Depends(current_active_user),
|
|
):
|
|
try:
|
|
db_search_space = await check_ownership(
|
|
session, SearchSpace, search_space_id, user
|
|
)
|
|
update_data = search_space_update.model_dump(exclude_unset=True)
|
|
for key, value in update_data.items():
|
|
setattr(db_search_space, key, value)
|
|
await session.commit()
|
|
await session.refresh(db_search_space)
|
|
return db_search_space
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
await session.rollback()
|
|
raise HTTPException(
|
|
status_code=500, detail=f"Failed to update search space: {e!s}"
|
|
) from e
|
|
|
|
|
|
@router.delete("/searchspaces/{search_space_id}", response_model=dict)
|
|
async def delete_search_space(
|
|
search_space_id: int,
|
|
session: AsyncSession = Depends(get_async_session),
|
|
user: User = Depends(current_active_user),
|
|
):
|
|
try:
|
|
db_search_space = await check_ownership(
|
|
session, SearchSpace, search_space_id, user
|
|
)
|
|
await session.delete(db_search_space)
|
|
await session.commit()
|
|
return {"message": "Search space deleted successfully"}
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
await session.rollback()
|
|
raise HTTPException(
|
|
status_code=500, detail=f"Failed to delete search space: {e!s}"
|
|
) from e
|