15427c96c0
- Change file mode from 644 to 755 for all project files - Add chain_id parameter to get_balance RPC endpoint with default "ait-devnet" - Rename Miner.extra_meta_data to extra_metadata for consistency
64 lines
1.8 KiB
Python
Executable File
64 lines
1.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
Test script for host GPU miner
|
|
"""
|
|
|
|
import subprocess
|
|
import httpx
|
|
|
|
# Test GPU
|
|
print("Testing GPU access...")
|
|
result = subprocess.run(['nvidia-smi', '--query-gpu=name', '--format=csv,noheader,nounits'],
|
|
capture_output=True, text=True)
|
|
if result.returncode == 0:
|
|
print(f"✅ GPU detected: {result.stdout.strip()}")
|
|
else:
|
|
print("❌ GPU not accessible")
|
|
|
|
# Test Ollama
|
|
print("\nTesting Ollama...")
|
|
try:
|
|
response = httpx.get("http://localhost:11434/api/tags", timeout=5)
|
|
if response.status_code == 200:
|
|
models = response.json().get('models', [])
|
|
print(f"✅ Ollama running with {len(models)} models")
|
|
for m in models[:3]: # Show first 3 models
|
|
print(f" - {m['name']}")
|
|
else:
|
|
print("❌ Ollama not responding")
|
|
except Exception as e:
|
|
print(f"❌ Ollama error: {e}")
|
|
|
|
# Test Coordinator
|
|
print("\nTesting Coordinator...")
|
|
try:
|
|
response = httpx.get("http://127.0.0.1:8000/v1/health", timeout=5)
|
|
if response.status_code == 200:
|
|
print("✅ Coordinator is accessible")
|
|
else:
|
|
print("❌ Coordinator not responding")
|
|
except Exception as e:
|
|
print(f"❌ Coordinator error: {e}")
|
|
|
|
# Test Ollama inference
|
|
print("\nTesting Ollama inference...")
|
|
try:
|
|
response = httpx.post(
|
|
"http://localhost:11434/api/generate",
|
|
json={
|
|
"model": "llama3.2:latest",
|
|
"prompt": "Say hello",
|
|
"stream": False
|
|
},
|
|
timeout=10
|
|
)
|
|
if response.status_code == 200:
|
|
result = response.json()
|
|
print(f"✅ Inference successful: {result.get('response', '')[:50]}...")
|
|
else:
|
|
print("❌ Inference failed")
|
|
except Exception as e:
|
|
print(f"❌ Inference error: {e}")
|
|
|
|
print("\n✅ All tests completed!")
|