mirror of
https://github.com/hanzoai/python-sdk.git
synced 2026-10-23 20:23:00 +00:00
zeekay
bff668fbf6
- Added comprehensive anti-stub test suite that detects: - TODO/FIXME/STUB/FAKE/UNFINISHED patterns - Empty functions with only pass/ellipsis - NotImplementedError raises - Mock implementations in production - Debug prints in production - Skipped tests - Created strict GitHub Actions workflow that: - BLOCKS deployment if ANY forbidden patterns found - Requires ALL tests to pass (no skips allowed) - Runs security scans - Verifies all functions are implemented - Only allows PyPI publish after ALL checks pass - Added pre-commit hooks to catch issues locally before push - Updated Makefile with 'make check' that runs all quality gates - Fixed stub function issue in tools/__init__.py This ensures we NEVER deploy incomplete or stub code to production again!
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
"""Test streaming responses."""
|
|
|
|
import sys
|
|
import asyncio
|
|
from pathlib import Path
|
|
|
|
from rich.console import Console
|
|
|
|
# Add src to path
|
|
sys.path.insert(0, str(Path(__file__).parent / "pkg" / "hanzo" / "src"))
|
|
|
|
from hanzo.streaming import StreamingHandler, TypewriterEffect, stream_with_fallback
|
|
|
|
|
|
async def test_streaming():
|
|
"""Test streaming functionality."""
|
|
console = Console()
|
|
|
|
console.print("\n[bold cyan]Testing Streaming Responses[/bold cyan]\n")
|
|
|
|
# Test typewriter effect
|
|
console.print("[bold]Testing typewriter effect:[/bold]")
|
|
typewriter = TypewriterEffect(console)
|
|
|
|
await typewriter.type_text(
|
|
"This is a typewriter effect demonstration...", speed=0.02
|
|
)
|
|
|
|
# Test code typing
|
|
console.print("\n[bold]Testing code typing:[/bold]")
|
|
code = """def fibonacci(n: int) -> int:
|
|
if n <= 1:
|
|
return n
|
|
return fibonacci(n-1) + fibonacci(n-2)"""
|
|
|
|
await typewriter.type_code(code, language="python", speed=0.01)
|
|
|
|
# Test simulated streaming
|
|
console.print("\n[bold]Testing simulated streaming:[/bold]")
|
|
handler = StreamingHandler(console)
|
|
|
|
sample_text = "This is a simulated streaming response. It will appear word by word as if being generated in real-time. This creates a better user experience!"
|
|
|
|
await handler.simulate_streaming(sample_text, delay=0.03)
|
|
|
|
# Test real streaming with fallback
|
|
console.print("\n[bold]Testing streaming with fallback:[/bold]")
|
|
|
|
test_message = "What is 2 + 2? Reply with just the number."
|
|
|
|
response = await stream_with_fallback(test_message, console)
|
|
|
|
if response:
|
|
console.print(f"\n[green]✅ Streaming test successful![/green]")
|
|
console.print(f"Response: {response}")
|
|
else:
|
|
console.print("\n[yellow]⚠️ No streaming available (no API keys)[/yellow]")
|
|
|
|
return True
|
|
|
|
|
|
if __name__ == "__main__":
|
|
success = asyncio.run(test_streaming())
|
|
sys.exit(0 if success else 1)
|