feat: support new model Grok 4.20 Beta

# Conflicts:
#	app/services/grok/services/chat.py
This commit is contained in:
Chenyme authored and XianYuDaXian committed 2026-02-18 23:29:48 +08:00
1 parent bac0ed508c
commit b6b753b97b
8 files changed
+263 -5

No files matched your search

+62
View File
@@ -0,0 +1,62 @@
"""
Simple usage probe to inspect /rest/rate-limits response.
Usage:
python scripts/test_usage_response.py
(optional) TOKEN_POOL=ssoBasic|ssoSuper
"""
import asyncio
import os
import sys
from pathlib import Path
ROOT = Path(__file__).resolve().parents[1]
if str(ROOT) not in sys.path:
sys.path.insert(0, str(ROOT))
from curl_cffi.requests import AsyncSession
from app.core.config import config
from app.services.reverse.rate_limits import RateLimitsReverse
from app.services.token import get_token_manager
async def main() -> int:
await config.load()
token = None
pool = os.getenv("TOKEN_POOL")
manager = await get_token_manager()
await manager.reload_if_stale()
if pool:
token = manager.get_token(pool_name=pool)
else:
token = manager.get_token(pool_name="ssoBasic") or manager.get_token(
pool_name="ssoSuper"
)
if not token:
token = os.getenv("GROK_TOKEN") or os.getenv("SSO_TOKEN") or os.getenv("TOKEN")
if not token:
print("Missing token. Ensure token pool is configured or set GROK_TOKEN.")
return 2
async with AsyncSession() as session:
response = await RateLimitsReverse.request(session, token)
try:
data = response.json()
except Exception as exc:
print(f"Failed to parse JSON: {exc}")
raw = getattr(response, "text", "")
if raw:
print(raw)
return 3
print(data)
return 0
if __name__ == "__main__":
raise SystemExit(asyncio.run(main()))