coffre ObsiGate Phase 4 (issue #22) : app/routers/agent.py

2026-10-27 23:48:20 -04:00
parent cd0835116d
commit f827e9fc7d
1 changed files with 10 additions and 1 deletions
@@ -673,6 +673,8 @@ def create_connectors_route(request: Request, body: dict = Body(default={})):
            body.get("name") or "", body.get("url") or "", str(body.get("secret") or ""),
            kind=str(body.get("kind") or "custom"),
            auth=str(body.get("auth") or "bearer"),
+            workspace_id=body.get("workspace_id"),
+            coffre_reference=body.get("coffre_reference"),
        )
    except ValueError as exc:
        raise HTTPException(status_code=400, detail=str(exc)) from exc
@@ -682,11 +684,17 @@ def create_connectors_route(request: Request, body: dict = Body(default={})):
 def update_connectors_route(request: Request, connector_id: int, body: dict = Body(default={})):
    _current_user_id(request)
    try:
+        kwargs = {}
+        if "workspace_id" in body:
+            kwargs["workspace_id"] = body.get("workspace_id")
+        if "coffre_reference" in body:
+            kwargs["coffre_reference"] = body.get("coffre_reference")
        row = connectors.update_connector(
            connector_id,
            name=body.get("name"),
            enabled=body.get("enabled"),
            secret=body.get("secret"),
+            **kwargs,
        )
    except ValueError as exc:
        raise HTTPException(status_code=400, detail=str(exc)) from exc
@@ -711,7 +719,8 @@ async def probe_connectors_route(request: Request, body: dict = Body(default={})
        raise HTTPException(status_code=400, detail="connector est requis")
    user_id = _current_user_id(request)
    try:
-        return await connectors.probe(target, user_id)
+        return await connectors.probe(target, user_id,
+                                    workspace_id=body.get("workspace_id"))
    except ValueError as exc:
        raise HTTPException(status_code=404, detail=str(exc)) from exc