fix: Corrected response_model usage (#1036)

2024-02-21 16:38:07 -05:00
parent d64178655d
commit c9349acecf
1 changed files with 1 additions and 1 deletions
--- a/memgpt/server/rest_api/admin/users.py
+++ b/memgpt/server/rest_api/admin/users.py
@@ -141,7 +141,7 @@ def setup_admin_router(server: SyncServer, interface: QueuingInterface):
            raise
        except Exception as e:
            raise HTTPException(status_code=500, detail=f"{e}")
-        return CreateAPIKeyResponse(api_key=processed_tokens)
+        return GetAPIKeyResponse(api_key=processed_tokens)

    @router.delete("/users/keys", tags=["admin"], response_model=DeleteAPIKeyResponse)
    def delete_api_key(