cleaner

src/proxy_lite/logger.py

@@ -1,7 +1,6 @@
 import asyncio
 import logging
 import sys
-import time  # Added for time.sleep
 from typing import Literal
 from uuid import uuid4
 
@@ -9,29 +8,18 @@ from rich.logging import RichHandler
 
 
 class StructuredLogger(logging.Logger):
-    def _stream_message_sync(self, message: str) -> None:
-        """Synchronous version of stream message to run in separate thread."""
+    async def stream_message(self, message: str) -> None:
+        """Streams the message character by character asynchronously."""
         try:
             sys.stdout.write("\r")  # Overwrite current line
             for char in message:
                 sys.stdout.write(char)
                 sys.stdout.flush()
-                time.sleep(0.002)  # Using regular sleep in thread
+                await asyncio.sleep(0.002)
             sys.stdout.write("\n")
         except Exception:
             pass
 
-    async def stream_message(self, message: str) -> None:
-        """Streams the message character by character in a separate thread."""
-        return await asyncio.to_thread(self._stream_message_sync, message)
-
-    # def info_stream(self, msg: str) -> Coroutine[Any, Any, None]:
-    #     """Special method for streaming messages that returns a coroutine."""
-    #     # Log the message normally first
-    #     # self.info(msg)
-    #     # Return the streaming coroutine
-    #     return self.stream_message(msg)
-
     def _log(
         self,
         level,
@@ -102,5 +90,3 @@ logging.setLoggerClass(StructuredLogger)
 
 # Initialize logger
 logger = create_logger(__name__, level="INFO")
-
-stream_logger = create_logger(__name__, level="INFO", detailed_name=False)

src/proxy_lite/solvers/simple_solver.py

@@ -2,7 +2,7 @@
 import json
 import re
 from functools import cached_property
-from typing import Any, Coroutine, Literal, Optional
+from typing import Literal, Optional
 
 from proxy_lite.agents import AgentConfigTypes, Agents, BaseAgent
 from proxy_lite.environments.environment_base import Action, Observation
@@ -27,7 +27,6 @@ class SimpleSolverConfig(BaseSolverConfig):
 class SimpleSolver(BaseSolver):
     task: Optional[str] = None
     complete: bool = False
-    coroutines: list[Coroutine[Any, Any, None]] = []
 
     @cached_property
     def tools(self) -> list[Tool]:
@@ -84,7 +83,7 @@ class SimpleSolver(BaseSolver):
 
         self.logger.info("🌐 [bold blue]Observation:[/]")
         # await self.logger.stream_message(observation_content)
-        self.logger._stream_message_sync(observation_content)
+        await self.logger.stream_message(observation_content)
 
         # Extract text between thinking tags if present
         thinking_match = re.search(r"<thinking>(.*?)</thinking>", text_content, re.DOTALL)
@@ -92,7 +91,7 @@ class SimpleSolver(BaseSolver):
 
         self.logger.info("🧠 [bold purple]Thinking:[/]")
         # await self.logger.stream_message(thinking_content)
-        self.logger._stream_message_sync(thinking_content)
+        await self.logger.stream_message(thinking_content)
 
         return Action(tool_calls=message.tool_calls, text=text_content)