temp: /tool/auth

cohere-ai · Jul 11, 2024 · 81eeb01 · 81eeb01
1 parent e0cf5ff
commit 81eeb01
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/backend/tools/google_drive/auth.py b/src/backend/tools/google_drive/auth.py
@@ -124,6 +124,17 @@ def process_auth_token(cls, request: Request, session: DBSessionDep) -> str:
         if res.status_code != 200:
             logger.error(f"Error in google drive auth: {res_body}")
             return res_body
+
+        try:
+            tool_auth_crud.get_tool_auth(
+                db=session, tool_id=GoogleDrive.NAME, user_id=state["user_id"]
+            )
+            tool_auth_crud.delete_tool_auth(
+                db=session, user_id=state["user_id"], tool_id=GoogleDrive.NAME
+            )
+        except Exception as _e:
+            pass
+
         tool_auth_crud.create_tool_auth(
             session,
             ToolAuth(