fix(db): release connection before embedding in knowledge /{id}/update (#20574)
Remove Depends(get_session) from POST /{id}/update endpoint to prevent database connections from being held during embedding API calls (1-5+ seconds).
All database operations (get_knowledge_by_id, has_access, has_permission, update_knowledge_by_id, get_file_metadatas_by_id) manage their own short-lived sessions internally, releasing connections before and after the slow embed_knowledge_base_metadata() call.
This commit is contained in:
@@ -419,9 +419,12 @@ async def update_knowledge_by_id(
|
|||||||
id: str,
|
id: str,
|
||||||
form_data: KnowledgeForm,
|
form_data: KnowledgeForm,
|
||||||
user=Depends(get_verified_user),
|
user=Depends(get_verified_user),
|
||||||
db: Session = Depends(get_session),
|
|
||||||
):
|
):
|
||||||
knowledge = Knowledges.get_knowledge_by_id(id=id, db=db)
|
# NOTE: We intentionally do NOT use Depends(get_session) here.
|
||||||
|
# Database operations manage their own short-lived sessions internally.
|
||||||
|
# This prevents holding a connection during embed_knowledge_base_metadata()
|
||||||
|
# which makes external embedding API calls (1-5+ seconds).
|
||||||
|
knowledge = Knowledges.get_knowledge_by_id(id=id)
|
||||||
if not knowledge:
|
if not knowledge:
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
status_code=status.HTTP_400_BAD_REQUEST,
|
status_code=status.HTTP_400_BAD_REQUEST,
|
||||||
@@ -430,7 +433,7 @@ async def update_knowledge_by_id(
|
|||||||
# Is the user the original creator, in a group with write access, or an admin
|
# Is the user the original creator, in a group with write access, or an admin
|
||||||
if (
|
if (
|
||||||
knowledge.user_id != user.id
|
knowledge.user_id != user.id
|
||||||
and not has_access(user.id, "write", knowledge.access_control, db=db)
|
and not has_access(user.id, "write", knowledge.access_control)
|
||||||
and user.role != "admin"
|
and user.role != "admin"
|
||||||
):
|
):
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
@@ -446,12 +449,11 @@ async def update_knowledge_by_id(
|
|||||||
user.id,
|
user.id,
|
||||||
"sharing.public_knowledge",
|
"sharing.public_knowledge",
|
||||||
request.app.state.config.USER_PERMISSIONS,
|
request.app.state.config.USER_PERMISSIONS,
|
||||||
db=db,
|
|
||||||
)
|
)
|
||||||
):
|
):
|
||||||
form_data.access_control = {}
|
form_data.access_control = {}
|
||||||
|
|
||||||
knowledge = Knowledges.update_knowledge_by_id(id=id, form_data=form_data, db=db)
|
knowledge = Knowledges.update_knowledge_by_id(id=id, form_data=form_data)
|
||||||
if knowledge:
|
if knowledge:
|
||||||
# Re-embed knowledge base for semantic search
|
# Re-embed knowledge base for semantic search
|
||||||
await embed_knowledge_base_metadata(
|
await embed_knowledge_base_metadata(
|
||||||
@@ -462,7 +464,7 @@ async def update_knowledge_by_id(
|
|||||||
)
|
)
|
||||||
return KnowledgeFilesResponse(
|
return KnowledgeFilesResponse(
|
||||||
**knowledge.model_dump(),
|
**knowledge.model_dump(),
|
||||||
files=Knowledges.get_file_metadatas_by_id(knowledge.id, db=db),
|
files=Knowledges.get_file_metadatas_by_id(knowledge.id),
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
|
|||||||
Reference in New Issue
Block a user