WEBDIENSTE/openswarm
1
0
Fork 0
mirror of https://github.com/openswarm-ai/openswarm.git synced 2026-09-30 05:24:50 +02:00
Code Issues Packages Projects Releases Wiki Activity

[Haik]: minor cleanup focussed around naming (especially the names starting with an underscore as this just nerfs all type checking/specing) also some logger removes, pulling out constants, etc. Now gonna pull out some helpers then maybe its singelton time again?

Browse Source
This commit is contained in:
haikdc
2026-04-05 20:33:40 -07:00
parent 961941bd38
commit aa250ce203
5 changed files with 67 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/subscriptions/NineRouter/NineRouterClient.py
+5 -25
View File
@@ -1,31 +1,11 @@
"""HTTP client for 9Router's REST API."""
import logging
import httpx
from typeguard import typechecked
from backend.apps.subscriptions.NineRouter.constants import NINE_ROUTER_API, NINE_ROUTER_V1
from backend.ports import NINE_ROUTER_PORT
logger = logging.getLogger(__name__)
NINE_ROUTER_URL: str = f"http://localhost:{NINE_ROUTER_PORT}"
NINE_ROUTER_API: str = f"{NINE_ROUTER_URL}/api"
NINE_ROUTER_V1: str = f"{NINE_ROUTER_URL}/v1"
@typechecked
async def get_usage_stats(period: str = "all") -> dict | None:
try:
async with httpx.AsyncClient(timeout=5.0) as client:
r = await client.get(f"{NINE_ROUTER_API}/usage/stats", params={"period": period})
if r.status_code == 200:
return r.json()
except Exception as e:
logger.debug(f"9Router usage stats fetch failed: {e}")
return None
@typechecked
async def get_providers() -> list[dict] | dict:
try:
@@ -34,7 +14,7 @@ async def get_providers() -> list[dict] | dict:
if r.status_code == 200:
return r.json()
except Exception as e:
logger.debug(f"9Router providers fetch failed: {e}")
print(f"9Router providers fetch failed: {e}")
return []
@@ -110,10 +90,10 @@ async def exchange_oauth(
"codeVerifier": code_verifier,
"state": state,
}
logger.info(f"exchange_oauth: provider={provider} redirect_uri={redirect_uri}")
print(f"exchange_oauth: provider={provider} redirect_uri={redirect_uri}")
async with httpx.AsyncClient(timeout=15.0) as client:
r = await client.post(f"{NINE_ROUTER_API}/oauth/{provider}/exchange", json=payload)
logger.info(f"exchange_oauth: status={r.status_code}")
print(f"exchange_oauth: status={r.status_code}")
r.raise_for_status()
return r.json()
@@ -136,5 +116,5 @@ async def get_models() -> list[dict]:
for m in models
]
except Exception as e:
logger.debug(f"9Router models fetch failed: {e}")
print(f"9Router models fetch failed: {e}")
return []