fix: add context rename functionality

- Add update_context function to backend CRUD
- Add PUT endpoint for contexts API
- Fixes 405 error when trying to rename contexts

Author: rb-x

backend/api/v1/contexts.py

@@ -3,7 +3,7 @@
 from uuid import UUID
 from neo4j import AsyncSession
 
-from schemas.context import Context, ContextCreate
+from schemas.context import Context, ContextCreate, ContextUpdate
 from crud import context as context_crud
 from api.dependencies import get_current_user, get_session
 from schemas.user import User
@@ -56,6 +56,21 @@ async def get_context(
         raise HTTPException(status_code=404, detail="Context not found.")
     return context
 
+@contexts_router.put("/{context_id}", response_model=Context)
+async def update_context(
+    project_id: UUID,
+    context_id: UUID,
+    context_in: ContextUpdate,
+    session: AsyncSession = Depends(get_session),
+    current_user: User = Depends(get_current_user),
+):
+    context = await context_crud.update_context(
+        session, context_id=context_id, context_in=context_in, project_id=project_id, owner_id=current_user.id
+    )
+    if not context:
+        raise HTTPException(status_code=404, detail="Context not found.")
+    return context
+
 @contexts_router.delete("/{context_id}", status_code=status.HTTP_204_NO_CONTENT)
 async def delete_context(
     project_id: UUID,

backend/crud/context.py

@@ -94,6 +94,35 @@ async def get_context_with_variables(
 
     return Context.model_validate(context_data)
 
+async def update_context(
+    session: AsyncSession, context_id: UUID, context_in: ContextUpdate, project_id: UUID, owner_id: UUID
+) -> Context | None:
+    props = context_in.model_dump(exclude_unset=True)
+    if not props:
+        # If no properties to update, just return the current context
+        return await get_context_with_variables(session, context_id, project_id, owner_id, include_sensitive=True)
+    
+    query = """
+    MATCH (user:User {id: $owner_id})-[:OWNS]->(project:Project {id: $project_id})-[:HAS_CONTEXT]->(context:Context {id: $context_id})
+    SET context += $props
+    RETURN context
+    """
+    result = await session.run(
+        query,
+        {
+            "owner_id": str(owner_id),
+            "project_id": str(project_id),
+            "context_id": str(context_id),
+            "props": props,
+        },
+    )
+    record = await result.single()
+    if not record:
+        return None
+    
+    # Return the updated context with variables
+    return await get_context_with_variables(session, context_id, project_id, owner_id, include_sensitive=True)
+
 async def delete_context(session: AsyncSession, context_id: UUID, project_id: UUID, owner_id: UUID) -> bool:
     # Deletes context and all its variables due to DETACH DELETE
     query = """