50 lines
2.5 KiB
Python
50 lines
2.5 KiB
Python
"""Explicit, bounded connection smoke against an already configured local Provider.
|
|
|
|
Defaults to a plan. --execute performs one test request, never reads credentials.
|
|
The output deliberately keeps untested protocol scenarios pending.
|
|
"""
|
|
import argparse
|
|
from datetime import datetime, timezone
|
|
import json
|
|
from pathlib import Path
|
|
from urllib.error import HTTPError, URLError
|
|
from urllib.parse import urlparse
|
|
from urllib.request import Request, urlopen
|
|
|
|
SCENARIOS = ['model_discovery', 'tool_roundtrip', 'stream_reasoning_and_content',
|
|
'stream_cancel', 'cache_hit_and_miss', 'context_limit', 'context_compression']
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--base-url', default='http://127.0.0.1:8000')
|
|
parser.add_argument('--provider', required=True)
|
|
parser.add_argument('--model', required=True)
|
|
parser.add_argument('--output', required=True, type=Path)
|
|
parser.add_argument('--execute', action='store_true', help='Perform one provider connection test; may incur provider charges')
|
|
args = parser.parse_args()
|
|
target = urlparse(args.base_url)
|
|
if target.scheme != 'http' or target.hostname not in ('127.0.0.1', 'localhost', '::1') or target.username or target.password or target.query or target.fragment:
|
|
parser.error('Use a local HTTP AI Core address without credentials or query parameters')
|
|
result = {'date': datetime.now(timezone.utc).isoformat(), 'provider': args.provider, 'model': args.model,
|
|
'max_test_requests': 1, 'connection': 'pending',
|
|
'scenarios': {name: 'pending' for name in SCENARIOS}, 'overall': 'not_accepted'}
|
|
if args.execute:
|
|
body = json.dumps({'provider_id': args.provider, 'model': args.model}).encode()
|
|
request = Request(args.base_url.rstrip('/') + '/api/providers/test', data=body, headers={'Content-Type': 'application/json'}, method='POST')
|
|
try:
|
|
with urlopen(request, timeout=60) as response:
|
|
payload = json.load(response)
|
|
result['connection'] = 'passed' if payload.get('success') is True else 'failed'
|
|
result['latency_ms'] = payload.get('latency_ms')
|
|
except HTTPError as error:
|
|
result['connection'] = 'failed'
|
|
result['http_status'] = error.code # Do not persist remote error bodies or headers.
|
|
except (URLError, TimeoutError, ValueError):
|
|
result['connection'] = 'unavailable'
|
|
args.output.write_text(json.dumps(result, ensure_ascii=False, indent=2), encoding='utf-8')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|