Spaces:
Sleeping
Sleeping
File size: 1,921 Bytes
93c9664 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
"""
Simple V4 test with short text.
"""
import requests
import json
# Simple test text
payload = {
"text": "Artificial intelligence is transforming healthcare. AI algorithms can analyze medical images faster than human doctors. Machine learning helps predict patient outcomes. This technology will revolutionize medical diagnosis.",
"style": "executive",
"max_tokens": 256
}
print("Testing V4 API with short text...\n")
try:
response = requests.post(
"http://localhost:7860/api/v4/scrape-and-summarize/stream",
json=payload,
stream=True,
timeout=600
)
print(f"Status: {response.status_code}\n")
if response.status_code != 200:
print(f"Error: {response.text}")
else:
print("=== STREAMING OUTPUT ===\n")
for line in response.iter_lines():
if line:
line_str = line.decode('utf-8')
if line_str.startswith('data: '):
try:
event = json.loads(line_str[6:])
# Print metadata
if event.get('type') == 'metadata':
print(f"Metadata: {json.dumps(event['data'], indent=2)}\n")
# Print content
elif 'content' in event and not event.get('done'):
print(event['content'], end='', flush=True)
# Print done event
elif event.get('done'):
print(f"\n\n=== DONE ===")
print(f"Tokens: {event.get('tokens_used', 0)}")
print(f"Latency: {event.get('latency_ms', 0):.2f}ms")
except json.JSONDecodeError as e:
print(f"\nJSON Error: {e}")
print(f"Raw: {line_str}")
except Exception as e:
print(f"Error: {e}")
|