|
|
""" |
|
|
Quick test of SPARKNET core functionality |
|
|
""" |
|
|
|
|
|
import asyncio |
|
|
import sys |
|
|
from pathlib import Path |
|
|
|
|
|
sys.path.insert(0, str(Path(__file__).parent)) |
|
|
|
|
|
from src.llm.ollama_client import OllamaClient |
|
|
from src.utils.gpu_manager import get_gpu_manager |
|
|
from src.tools.gpu_tools import GPUMonitorTool |
|
|
from loguru import logger |
|
|
|
|
|
|
|
|
async def test_ollama(): |
|
|
"""Test Ollama client""" |
|
|
print("\n=== Testing Ollama Client ===") |
|
|
|
|
|
client = OllamaClient(default_model="gemma2:2b") |
|
|
|
|
|
|
|
|
response = client.generate( |
|
|
prompt="Say 'Hello from SPARKNET!' and nothing else.", |
|
|
temperature=0.1, |
|
|
) |
|
|
print(f"Response: {response[:100]}") |
|
|
|
|
|
return True |
|
|
|
|
|
|
|
|
async def test_gpu(): |
|
|
"""Test GPU manager""" |
|
|
print("\n=== Testing GPU Manager ===") |
|
|
|
|
|
gpu_manager = get_gpu_manager() |
|
|
print(gpu_manager.monitor()) |
|
|
|
|
|
return True |
|
|
|
|
|
|
|
|
async def test_tools(): |
|
|
"""Test tools""" |
|
|
print("\n=== Testing Tools ===") |
|
|
|
|
|
gpu_tool = GPUMonitorTool() |
|
|
result = await gpu_tool.execute() |
|
|
|
|
|
print(f"Tool Success: {result.success}") |
|
|
print(f"Output Preview: {result.output[:200] if result.output else 'None'}") |
|
|
|
|
|
return True |
|
|
|
|
|
|
|
|
async def main(): |
|
|
"""Run all tests""" |
|
|
print("="*60) |
|
|
print("SPARKNET Basic Functionality Test") |
|
|
print("="*60) |
|
|
|
|
|
try: |
|
|
|
|
|
await test_gpu() |
|
|
|
|
|
|
|
|
await test_ollama() |
|
|
|
|
|
|
|
|
await test_tools() |
|
|
|
|
|
print("\n" + "="*60) |
|
|
print("✓ All tests passed!") |
|
|
print("="*60) |
|
|
|
|
|
except Exception as e: |
|
|
print(f"\n✗ Test failed: {e}") |
|
|
import traceback |
|
|
traceback.print_exc() |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
asyncio.run(main()) |
|
|
|
