Skip to content

Commit 4238994

Browse files
authored
fix: playground token display in account page (#432)
1 parent 820ede0 commit 4238994

3 files changed

Lines changed: 4 additions & 17 deletions

File tree

api/endpoints/admin/tokens.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ async def create_token(
4141

4242
@router.delete(
4343
path=ENDPOINT__ADMIN_TOKENS + "/{token:path}",
44-
dependencies=[Security(dependency=AccessController())],
44+
dependencies=[Security(dependency=AccessController(permissions=[PermissionType.ADMIN]))],
4545
status_code=204,
4646
)
4747
async def delete_token(
@@ -61,7 +61,7 @@ async def delete_token(
6161

6262
@router.get(
6363
path=ENDPOINT__ADMIN_TOKENS + "/{token:path}",
64-
dependencies=[Security(dependency=AccessController())],
64+
dependencies=[Security(dependency=AccessController(permissions=[PermissionType.ADMIN]))],
6565
status_code=200,
6666
response_model=Token,
6767
)
@@ -81,7 +81,7 @@ async def get_token(
8181

8282
@router.get(
8383
path=ENDPOINT__ADMIN_TOKENS,
84-
dependencies=[Security(dependency=AccessController())],
84+
dependencies=[Security(dependency=AccessController(permissions=[PermissionType.ADMIN]))],
8585
status_code=200,
8686
response_model=Tokens,
8787
)

playground/backend/admin.py

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -80,19 +80,6 @@ def create_user(email: str, password: str, role: int, name: Optional[str] = None
8080
headers={"Authorization": f"Bearer {st.session_state["user"].api_key}"},
8181
)
8282

83-
if response.status_code != 201:
84-
st.toast(response.json()["detail"], icon="❌")
85-
return
86-
87-
user_id = response.json()["id"]
88-
89-
# create playground token
90-
response = requests.post(
91-
url=f"{configuration.playground.api_url}/v1/admin/tokens",
92-
json={"user": user_id, "name": "playground"},
93-
headers={"Authorization": f"Bearer {st.session_state["user"].api_key}"},
94-
)
95-
9683
if response.status_code != 201:
9784
st.toast(response.json()["detail"], icon="❌")
9885
return

playground/backend/common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def get_documents(collection_id: int, offset: int = 0, limit: int = 10) -> dict:
6262

6363
def get_tokens(offset: int = 0, limit: int = 10) -> list:
6464
response = requests.get(
65-
url=f"{configuration.playground.api_url}/v1/admin/tokens?offset={offset}&limit={limit}",
65+
url=f"{configuration.playground.api_url}/v1/me/keys?offset={offset}&limit={limit}",
6666
headers={"Authorization": f"Bearer {st.session_state["user"].api_key}"},
6767
)
6868

0 commit comments

Comments
 (0)