diff --git a/agent/agent.sh b/agent/agent.sh index 57d95ec..60818ea 100755 --- a/agent/agent.sh +++ b/agent/agent.sh @@ -48,11 +48,49 @@ if cfg and os.path.exists(cfg): default_name = dp.split('/')[0] if '/' in dp else dp for n,p in m.get('providers',{}).items(): if not isinstance(p,dict): continue + base = p.get('baseUrl','') + key = p.get('apiKey','') + api_type = p.get('api','') for mod in p.get('models',[]): - providers.append({'name':n,'model':mod.get('id',''),'api':p.get('api',''),'default':n==default_name}) + providers.append({'name':n,'model':mod.get('id',''),'api':api_type,'default':n==default_name, + '_base':base,'_key':key}) providers.sort(key=lambda x: (not x.get('default',False), x['name'])) except: pass +# Health check each provider +import urllib.request,urllib.error +def check_provider(p): + base,key,api = p.get('_base',''),p.get('_key',''),p.get('api','') + if not base or not key: return {'ok':False,'ms':0,'err':'no config'} + try: + t0 = time.time() + if 'anthropic' in api: + url = base.rstrip('/')+'/v1/messages' + data = json.dumps({"model":p['model'],"max_tokens":1,"messages":[{"role":"user","content":"hi"}]}).encode() + req = urllib.request.Request(url,data,{'Content-Type':'application/json','x-api-key':key,'anthropic-version':'2023-06-01'}) + else: + url = base.rstrip('/')+'/chat/completions' + data = json.dumps({"model":p['model'],"max_tokens":1,"messages":[{"role":"user","content":"hi"}]}).encode() + req = urllib.request.Request(url,data,{'Content-Type':'application/json','Authorization':'Bearer '+key}) + resp = urllib.request.urlopen(req,timeout=10) + ms = int((time.time()-t0)*1000) + return {'ok':True,'ms':ms,'err':''} + except urllib.error.HTTPError as e: + ms = int((time.time()-t0)*1000) + code = e.code + if code == 429: return {'ok':False,'ms':ms,'err':'限额'} + if code in (402,): return {'ok':False,'ms':ms,'err':'余额不足'} + return {'ok':True,'ms':ms,'err':''} # 400/401/403/422 means API is reachable + except Exception as e: + return {'ok':False,'ms':0,'err':'timeout'} + +for p in providers: + r = check_provider(p) + p['status'] = 'ok' if r['ok'] else 'err' + p['ms'] = r['ms'] + p['err'] = r['err'] + del p['_base'], p['_key'] + # CPU if mac: raw = run("ps -A -o %cpu | tail -n +2") diff --git a/public/index.html b/public/index.html index a23462c..b22a956 100644 --- a/public/index.html +++ b/public/index.html @@ -122,7 +122,7 @@ function renderNodes(){
${n.role}OC ${n.oc_version}${n.os}
${hbBars}
供应商
-${provs.sort((a,b)=>b.default-a.default||(a.name>b.name?1:-1)).map(p=>`
${p.default?'':''}${p.name} ${p.model}
`).join('')} +${provs.sort((a,b)=>b.default-a.default||(a.name>b.name?1:-1)).map(p=>`
${p.default?'':''}${p.name} ${p.model}
${p.status==='ok'?' '+p.ms+'ms':' '+(p.err||'离线')+''}
`).join('')}
${[['cpu',n.cpu],['mem',n.mem],['disk',n.disk],['swap',n.swap]].map(([l,v])=>`
${l}
${on?v+'%':'—'}
`).join('')}
@@ -143,7 +143,8 @@ function renderMatrix(){ let h='Provider'+ns.map(n=>''+n.name+'').join('')+''; for(const[prov,set]of allProvs){ h+=''+prov+''; - ns.forEach(n=>{h+=set.has(n.name)?'✓':'—';}); + ns.forEach(n=>{const ps=JSON.parse(n.providers||'[]');const f=ps.find(x=>x.name===prov.split(' (')[0]); + h+=f?(f.status==='ok'?`✓ ${f.ms}ms`:`✗ ${f.err||'离线'}`):'—';}); h+=''; } $('#matrixTable').innerHTML=h+'';