Skip to content

Commit 32c004a

Browse files
fix role page (#545)
* fix role page * Update pyproject.toml version --------- Co-authored-by: GitHub Actions <actions@github.com>
1 parent 0d6924a commit 32c004a

3 files changed

Lines changed: 16 additions & 15 deletions

File tree

playground/app/features/roles/state.py

Lines changed: 14 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ def _format_role(self, role: dict) -> Role:
2828
permissions_provide_models = True if "provide_models" in role["permissions"] else False
2929

3030
limits_dict = defaultdict(lambda: {"rpm": None, "rpd": None, "tpm": None, "tpd": None})
31+
3132
for limit in role["limits"]:
3233
router_name = router_dict_reverse[limit["router"]]
3334
limits_dict[router_name][limit["type"]] = limit["value"]
@@ -78,6 +79,19 @@ async def load_entities(self):
7879

7980
response = None
8081
try:
82+
async with httpx.AsyncClient() as client:
83+
response = await client.get(
84+
f"{self.opengatellm_url}/v1/admin/routers",
85+
headers={"Authorization": f"Bearer {self.api_key}"},
86+
timeout=configuration.settings.playground_opengatellm_timeout,
87+
)
88+
89+
response.raise_for_status()
90+
data = response.json()
91+
routers_data = data.get("data", [])
92+
self.routers_list = [{"id": router["id"], "name": router["name"]} for router in routers_data]
93+
self.routers_dict = {router["name"]: router["id"] for router in routers_data}
94+
8195
async with httpx.AsyncClient() as client:
8296
response = await client.get(
8397
f"{self.opengatellm_url}/v1/admin/roles",
@@ -97,19 +111,6 @@ async def load_entities(self):
97111
for role in data.get("data", []):
98112
self.entities.append(self._format_role(role))
99113

100-
async with httpx.AsyncClient() as client:
101-
response = await client.get(
102-
f"{self.opengatellm_url}/v1/admin/routers",
103-
headers={"Authorization": f"Bearer {self.api_key}"},
104-
timeout=configuration.settings.playground_opengatellm_timeout,
105-
)
106-
107-
response.raise_for_status()
108-
data = response.json()
109-
routers_data = data.get("data", [])
110-
self.routers_list = [{"id": router["id"], "name": router["name"]} for router in routers_data]
111-
self.routers_dict = {router["name"]: router["id"] for router in routers_data}
112-
113114
self.has_more_page = len(self.entities) == self.per_page
114115

115116
except Exception as e:

playground/app/shared/components/toasts.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,5 +13,5 @@ def httpx_error_toast(exception: Exception, response: Response | None = None) ->
1313
except Exception:
1414
message = response.text
1515
else:
16-
message = str(exception)
16+
message = type(exception).__name__ + ": " + str(exception)
1717
return rx.toast.error(message=message, position="bottom-right")

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[project]
22
name = "opengatellm"
3-
version = "0.3.1"
3+
version = "0.3.2"
44
description = "OpenGateLLM project"
55
requires-python = ">=3.12"
66
license = { text = "MIT" }

0 commit comments

Comments
 (0)