hanzo 0.3.21__py3-none-any.whl → 0.3.23__py3-none-any.whl

hanzo/streaming.py

@@ -3,42 +3,44 @@ Streaming response handler for Hanzo Dev.
 Provides real-time feedback as AI generates responses.
 """
 
+import time
 import asyncio
-from typing import AsyncGenerator, Optional, Callable
-from rich.console import Console
+from typing import Callable, Optional, AsyncGenerator
+
 from rich.live import Live
 from rich.panel import Panel
+from rich.console import Console
 from rich.markdown import Markdown
-import time
 
 
 class StreamingHandler:
     """Handles streaming responses from AI models."""
-    
+
     def __init__(self, console: Console = None):
         """Initialize streaming handler."""
         self.console = console or Console()
         self.current_response = ""
         self.is_streaming = False
-        
+
     async def stream_openai(self, client, messages: list, model: str = "gpt-4") -> str:
         """Stream response from OpenAI API."""
         try:
             stream = await client.chat.completions.create(
-                model=model,
-                messages=messages,
-                stream=True,
-                max_tokens=1000
+                model=model, messages=messages, stream=True, max_tokens=1000
             )
-            
+
             self.current_response = ""
             self.is_streaming = True
-            
+
             with Live(
-                Panel("", title="[bold cyan]AI Response[/bold cyan]", 
-                      title_align="left", border_style="dim cyan"),
+                Panel(
+                    "",
+                    title="[bold cyan]AI Response[/bold cyan]",
+                    title_align="left",
+                    border_style="dim cyan",
+                ),
                 console=self.console,
-                refresh_per_second=10
+                refresh_per_second=10,
             ) as live:
                 async for chunk in stream:
                     if chunk.choices[0].delta.content:
@@ -49,34 +51,38 @@ class StreamingHandler:
                                 title="[bold cyan]AI Response[/bold cyan]",
                                 title_align="left",
                                 border_style="dim cyan",
-                                padding=(1, 2)
+                                padding=(1, 2),
                             )
                         )
-                        
+
             self.is_streaming = False
             return self.current_response
-            
+
         except Exception as e:
             self.console.print(f"[red]Streaming error: {e}[/red]")
             self.is_streaming = False
             return None
-    
-    async def stream_anthropic(self, client, messages: list, model: str = "claude-3-5-sonnet-20241022") -> str:
+
+    async def stream_anthropic(
+        self, client, messages: list, model: str = "claude-3-5-sonnet-20241022"
+    ) -> str:
         """Stream response from Anthropic API."""
         try:
             self.current_response = ""
             self.is_streaming = True
-            
+
             with Live(
-                Panel("", title="[bold cyan]AI Response[/bold cyan]",
-                      title_align="left", border_style="dim cyan"),
+                Panel(
+                    "",
+                    title="[bold cyan]AI Response[/bold cyan]",
+                    title_align="left",
+                    border_style="dim cyan",
+                ),
                 console=self.console,
-                refresh_per_second=10
+                refresh_per_second=10,
             ) as live:
                 async with client.messages.stream(
-                    model=model,
-                    messages=messages,
-                    max_tokens=1000
+                    model=model, messages=messages, max_tokens=1000
                 ) as stream:
                     async for text in stream.text_stream:
                         self.current_response += text
@@ -86,42 +92,47 @@ class StreamingHandler:
                                 title="[bold cyan]AI Response[/bold cyan]",
                                 title_align="left",
                                 border_style="dim cyan",
-                                padding=(1, 2)
+                                padding=(1, 2),
                             )
                         )
-            
+
             self.is_streaming = False
             return self.current_response
-            
+
         except Exception as e:
             self.console.print(f"[red]Streaming error: {e}[/red]")
             self.is_streaming = False
             return None
-    
+
     async def stream_ollama(self, message: str, model: str = "llama3.2") -> str:
         """Stream response from Ollama local model."""
         import httpx
-        
+
         try:
             self.current_response = ""
             self.is_streaming = True
-            
+
             with Live(
-                Panel("", title="[bold cyan]AI Response (Local)[/bold cyan]",
-                      title_align="left", border_style="dim cyan"),
+                Panel(
+                    "",
+                    title="[bold cyan]AI Response (Local)[/bold cyan]",
+                    title_align="left",
+                    border_style="dim cyan",
+                ),
                 console=self.console,
-                refresh_per_second=10
+                refresh_per_second=10,
             ) as live:
                 async with httpx.AsyncClient() as client:
                     async with client.stream(
                         "POST",
                         "http://localhost:11434/api/generate",
                         json={"model": model, "prompt": message, "stream": True},
-                        timeout=60.0
+                        timeout=60.0,
                     ) as response:
                         async for line in response.aiter_lines():
                             if line:
                                 import json
+
                                 data = json.loads(line)
                                 if "response" in data:
                                     self.current_response += data["response"]
@@ -131,52 +142,56 @@ class StreamingHandler:
                                             title="[bold cyan]AI Response (Local)[/bold cyan]",
                                             title_align="left",
                                             border_style="dim cyan",
-                                            padding=(1, 2)
+                                            padding=(1, 2),
                                         )
                                     )
                                 if data.get("done", False):
                                     break
-            
+
             self.is_streaming = False
             return self.current_response
-            
+
         except Exception as e:
             self.console.print(f"[red]Ollama streaming error: {e}[/red]")
             self.is_streaming = False
             return None
-    
+
     async def simulate_streaming(self, text: str, delay: float = 0.02) -> str:
         """Simulate streaming for non-streaming APIs."""
         self.current_response = ""
         self.is_streaming = True
-        
+
         words = text.split()
-        
+
         with Live(
-            Panel("", title="[bold cyan]AI Response[/bold cyan]",
-                  title_align="left", border_style="dim cyan"),
+            Panel(
+                "",
+                title="[bold cyan]AI Response[/bold cyan]",
+                title_align="left",
+                border_style="dim cyan",
+            ),
             console=self.console,
-            refresh_per_second=20
+            refresh_per_second=20,
         ) as live:
             for i, word in enumerate(words):
                 self.current_response += word
                 if i < len(words) - 1:
                     self.current_response += " "
-                
+
                 live.update(
                     Panel(
                         Markdown(self.current_response),
                         title="[bold cyan]AI Response[/bold cyan]",
                         title_align="left",
                         border_style="dim cyan",
-                        padding=(1, 2)
+                        padding=(1, 2),
                     )
                 )
                 await asyncio.sleep(delay)
-        
+
         self.is_streaming = False
         return self.current_response
-    
+
     def stop_streaming(self):
         """Stop current streaming operation."""
         self.is_streaming = False
@@ -186,34 +201,38 @@ class StreamingHandler:
 
 class TypewriterEffect:
     """Provides typewriter effect for text output."""
-    
+
     def __init__(self, console: Console = None):
         self.console = console or Console()
-    
+
     async def type_text(self, text: str, speed: float = 0.03):
         """Type text with typewriter effect."""
         for char in text:
             self.console.print(char, end="")
             await asyncio.sleep(speed)
         self.console.print()  # New line at end
-    
+
     async def type_code(self, code: str, language: str = "python", speed: float = 0.01):
         """Type code with syntax highlighting."""
         from rich.syntax import Syntax
-        
+
         # Build up code progressively
         current_code = ""
-        lines = code.split('\n')
-        
+        lines = code.split("\n")
+
         with Live(console=self.console, refresh_per_second=30) as live:
             for line in lines:
                 for char in line:
                     current_code += char
-                    syntax = Syntax(current_code, language, theme="monokai", line_numbers=True)
+                    syntax = Syntax(
+                        current_code, language, theme="monokai", line_numbers=True
+                    )
                     live.update(syntax)
                     await asyncio.sleep(speed)
-                current_code += '\n'
-                syntax = Syntax(current_code, language, theme="monokai", line_numbers=True)
+                current_code += "\n"
+                syntax = Syntax(
+                    current_code, language, theme="monokai", line_numbers=True
+                )
                 live.update(syntax)
 
 
@@ -222,50 +241,52 @@ async def stream_with_fallback(message: str, console: Console = None) -> Optiona
     Stream response with automatic fallback to available options.
     """
     import os
+
     handler = StreamingHandler(console)
-    
+
     # Try OpenAI streaming
     if os.getenv("OPENAI_API_KEY"):
         try:
             from openai import AsyncOpenAI
+
             client = AsyncOpenAI()
             return await handler.stream_openai(
-                client,
-                [{"role": "user", "content": message}]
+                client, [{"role": "user", "content": message}]
             )
         except Exception as e:
             if console:
                 console.print(f"[yellow]OpenAI streaming failed: {e}[/yellow]")
-    
+
     # Try Anthropic streaming
     if os.getenv("ANTHROPIC_API_KEY"):
         try:
             from anthropic import AsyncAnthropic
+
             client = AsyncAnthropic()
             return await handler.stream_anthropic(
-                client,
-                [{"role": "user", "content": message}]
+                client, [{"role": "user", "content": message}]
             )
         except Exception as e:
             if console:
                 console.print(f"[yellow]Anthropic streaming failed: {e}[/yellow]")
-    
+
     # Try Ollama streaming
     try:
         return await handler.stream_ollama(message)
     except:
         pass
-    
+
     # Fallback to non-streaming with simulated effect
     if console:
         console.print("[yellow]Falling back to non-streaming mode[/yellow]")
-    
+
     # Get response from fallback handler
     from .fallback_handler import smart_chat
+
     response = await smart_chat(message, console)
-    
+
     if response:
         # Simulate streaming
         return await handler.simulate_streaming(response)
-    
-    return None
+
+    return None

{hanzo-0.3.21.dist-info → hanzo-0.3.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hanzo
-Version: 0.3.21
+Version: 0.3.23
 Summary: Hanzo AI - Complete AI Infrastructure Platform with CLI, Router, MCP, and Agent Runtime
 Project-URL: Homepage, https://hanzo.ai
 Project-URL: Repository, https://github.com/hanzoai/python-sdk

{hanzo-0.3.21.dist-info → hanzo-0.3.23.dist-info}/RECORD

@@ -1,14 +1,17 @@
 hanzo/__init__.py,sha256=f6N_RcJZ0F9ADrROlvPi1OrgwjF8cWQm34cml8hb1zk,169
 hanzo/__main__.py,sha256=F3Vz0Ty3bdAj_8oxyETMIqxlmNRnJOAFB1XPxbyfouI,105
-hanzo/cli.py,sha256=uxlJ30ZBwr_q8-iTipatwCXUA8cY3FZFMnVySRxehes,18586
-hanzo/dev.py,sha256=OcRGLjH9m_HSFhVRm7i8L4TtgB-9XseXLA0RexBHUH0,102736
-hanzo/fallback_handler.py,sha256=UJOzfGbf_5rg168PyzC9BQlB6yJ-lEW7B8MkHkVSiK8,10148
+hanzo/base_agent.py,sha256=GIKnJU0rCC7Jgmahi0ynxBKAfscb6CQBP261_aXDl1g,15325
+hanzo/batch_orchestrator.py,sha256=dCuWyEA9ulM5SWfz5Ecep-IK615AO56pXmLOCz3fv6c,35817
+hanzo/cli.py,sha256=5e_ee4_dOuiVWf--_ycsyBETkMOnrn0a30kORwwTTcY,18586
+hanzo/dev.py,sha256=56ug80N_UDaOuXkq9-QwjZkqn4nc2h-NNrLBqwFSFYc,105807
+hanzo/fallback_handler.py,sha256=avGujRF5a_x2Q-Q5ebQG87f54DcYfPFwlMMFhG2V45k,10181
 hanzo/mcp_server.py,sha256=XVygFNn-9CVdu8c95sP7fQjIRtA8K7nsGpgQNe44BRg,460
-hanzo/memory_manager.py,sha256=BjnHN0Fu6Lqg3aKGTVPKfQxDBjtengjwG3ac8zoirrQ,15101
+hanzo/memory_manager.py,sha256=eoaMS7f7rb2n9QjwcsCOz4EyyYx-RZ1gJi5Yzwc-aWU,14883
+hanzo/model_registry.py,sha256=q5i9ARhdn2io_5FvDsA8wsvpZ3_KdyaPVYQN__Nsui4,11921
 hanzo/orchestrator_config.py,sha256=JV7DS8aVZwBJ9XzgkQronFwV_A50QyXG3MH_pKwmCB8,11006
-hanzo/rate_limiter.py,sha256=wDC_dwx1pg87YnfNwaELnW2zYRAYjmrXb0_LhsJbz5c,11442
+hanzo/rate_limiter.py,sha256=jK166q_IAukHMyT-k0BQVVO2JxmkVpnJjTHqR17Q-7A,10858
 hanzo/repl.py,sha256=sW1quuqGkJ_AqgjN2vLNdtWgKDlXIkXiO9Bo1QQI0G4,1089
-hanzo/streaming.py,sha256=ZcFGD0k-RjURoamqCeq1Ripxs6I-ousaeQUFIs52Sic,10188
+hanzo/streaming.py,sha256=McLFkJZ0EEcnGKJU0jE3aJj-JO0sfCGS6Q44TcPV1LI,10209
 hanzo/commands/__init__.py,sha256=7rh94TPNhdq4gJBJS0Ayf0fGNChQYCQCJcJPmYYehiQ,182
 hanzo/commands/agent.py,sha256=DXCfuxHfmC90IoIOL6BJyp7h2yNUo-VIxrfl4OMh8CU,3480
 hanzo/commands/auth.py,sha256=JrM-EV4XDHzNDJeGJMjAr69T0Rxez53HEzlNo0jQ8nE,11187
@@ -18,7 +21,7 @@ hanzo/commands/config.py,sha256=xAzM6n9GhdVIqtn7JrHfLRzj1sshmxCujo7iet2hHqE,7490
 hanzo/commands/mcp.py,sha256=u1uEKDY6gUIa7VymEnRzy0ZphdIKYoNwPSeffZaiKnk,7418
 hanzo/commands/miner.py,sha256=_mZT9nQcT2QSSxI0rDDKuSBVdsg_uE_N_j3PXOHoj-Q,11677
 hanzo/commands/network.py,sha256=wJDxGIxJqc6FzQhbAn0Mw-WGCPUeCOsxmdU6GCmOhgM,11408
-hanzo/commands/repl.py,sha256=FwbBmjNDBZCUQlwKbMZl33r3f899rzI2a2-ZRoBe6kE,5977
+hanzo/commands/repl.py,sha256=Frc7tVTThFz0M_vwyA-m4pmfGDUr0WZGsxdu6g0ylNU,6081
 hanzo/commands/tools.py,sha256=fG27wRweVmaFJowBpmwp5PgkRUtIF8bIlu_hGWr69Ss,10393
 hanzo/interactive/__init__.py,sha256=ENHkGOqu-JYI05lqoOKDczJGl96oq6nM476EPhflAbI,74
 hanzo/interactive/dashboard.py,sha256=XB5H_PMlReriCip-wW9iuUiJQOAtSATFG8EyhhFhItU,3842
@@ -28,7 +31,7 @@ hanzo/utils/__init__.py,sha256=5RRwKI852vp8smr4xCRgeKfn7dLEnHbdXGfVYTZ5jDQ,69
 hanzo/utils/config.py,sha256=FD_LoBpcoF5dgJ7WL4o6LDp2pdOy8kS-dJ6iRO2GcGM,4728
 hanzo/utils/net_check.py,sha256=YFbJ65SzfDYHkHLZe3n51VhId1VI3zhyx8p6BM-l6jE,3017
 hanzo/utils/output.py,sha256=W0j3psF07vJiX4s02gbN4zYWfbKNsb8TSIoagBSf5vA,2704
-hanzo-0.3.21.dist-info/METADATA,sha256=OTh520I_RktXqn_gjGoIdbOD0sZS7LHeeeZwDMNMHSM,4279
-hanzo-0.3.21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hanzo-0.3.21.dist-info/entry_points.txt,sha256=pQLPMdqOXU_2BfTcMDhkqTCDNk_H6ApvYuSaWcuQOOw,171
-hanzo-0.3.21.dist-info/RECORD,,
+hanzo-0.3.23.dist-info/METADATA,sha256=eGqS7GX-lV1HYIJpSPq7KdKkOEph9tAYyE57JJfQWgM,4279
+hanzo-0.3.23.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hanzo-0.3.23.dist-info/entry_points.txt,sha256=pQLPMdqOXU_2BfTcMDhkqTCDNk_H6ApvYuSaWcuQOOw,171
+hanzo-0.3.23.dist-info/RECORD,,