fix for valid oai keys with restricted permissions showing up as invalid

2026-05-10 18:39:04 -07:00 · 2024-08-30 00:39:15 +10:00
parent f37f07e350
commit 3387a9c573
1 changed files with 6 additions and 0 deletions
@@ -29,6 +29,9 @@ async def get_oai_model(key: APIKey, session, retries, org=None):
                        top_model = model["id"]
                key.model = top_model
                return True
+            elif response.status == 403:
+                key.model = "gpt-4"
+                return True
            elif response.status != 502:
                return
        await asyncio.sleep(0.5)
@@ -99,6 +102,9 @@ async def get_oai_org(key: APIKey, session, retries):
                            key.default_org = org["name"]
                        key.organizations.append(org["name"])
                return True
+            elif response.status == 403:
+                key.default_org = None
+                return True
            elif response.status != 502:
                return
        await asyncio.sleep(0.5)