quantalogic 0.2.2__tar.gz → 0.2.4__tar.gz

{quantalogic-0.2.2 → quantalogic-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quantalogic
-Version: 0.2.2
+Version: 0.2.4
 Summary: QuantaLogic ReAct Agents
 Author: Raphaël MANSUY
 Author-email: raphael.mansuy@gmail.com

{quantalogic-0.2.2 → quantalogic-0.2.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "quantalogic"
-version = "0.2.2"
+version = "0.2.4"
 description = "QuantaLogic ReAct Agents"
 authors = ["Raphaël MANSUY <raphael.mansuy@gmail.com>"]
 readme = "README.md"

{quantalogic-0.2.2 → quantalogic-0.2.4}/quantalogic/agent.py

@@ -21,12 +21,6 @@ from quantalogic.utils.ask_user_validation import console_ask_for_user_validatio
 from quantalogic.xml_parser import ToleranceXMLParser
 from quantalogic.xml_tool_parser import ToolParser
 
-# Configure logger based on environment variable
-log_level = os.getenv("LOG_LEVEL", "ERROR")
-logger.remove()
-logger.add(sys.stderr, level=log_level)
-
-
 # Maximum ratio occupancy of the occupied memory
 MAX_OCCUPANCY = 90.0
 
@@ -81,7 +75,7 @@ class Agent(BaseModel):
 
     def __init__(
         self,
-        model_name: str = "ollama/qwen2.5-coder:14b",
+        model_name: str = "",
         memory: AgentMemory = AgentMemory(),
         tools: list[Tool] = [TaskCompleteTool()],
         ask_for_user_validation: Callable[[str], bool] = console_ask_for_user_validation,
@@ -91,6 +85,7 @@ class Agent(BaseModel):
     ):
         """Initialize the agent with model, memory, tools, and configurations."""
         try:
+            logger.debug("Initializing agent...")
             # Add TaskCompleteTool to the tools list if not already present
             if TaskCompleteTool() not in tools:
                 tools.append(TaskCompleteTool())
@@ -119,7 +114,7 @@ class Agent(BaseModel):
                 task_to_solve=task_to_solve,
                 specific_expertise=specific_expertise,
             )
-            logger.info("Agent initialized successfully.")
+            logger.debug("Agent initialized successfully.")
         except Exception as e:
             logger.error(f"Failed to initialize agent: {str(e)}")
             raise
@@ -135,6 +130,7 @@ class Agent(BaseModel):
         Returns:
             str: The final response after task completion.
         """
+        logger.debug(f"Solving task... {task}")
         self._reset_session(task_to_solve=task, max_iterations=max_iterations)
 
         # Add system prompt to memory
@@ -215,10 +211,13 @@ class Agent(BaseModel):
         # Emit event: Task Solve End
         self._emit_event("task_solve_end")
 
+        logger.debug(f"Task solved: {answer}")
+
         return answer
 
     def _reset_session(self, task_to_solve: str = "", max_iterations: int = 30):
         """Reset the agent's session."""
+        logger.debug("Resetting session...")
         self.task_to_solve = task_to_solve
         self.memory.reset()
         self.variable_store.reset()

{quantalogic-0.2.2 → quantalogic-0.2.4}/quantalogic/main.py

@@ -7,30 +7,37 @@ from typing import Optional
 
 # Third-party imports
 import click
-from rich.console import Console
-from rich.panel import Panel
-from rich.prompt import Confirm
+from loguru import logger
 
-from quantalogic.agent import Agent
+# Configure logger
+logger.remove()  # Remove default logger
+
+from rich.console import Console  # noqa: E402
+from rich.panel import Panel  # noqa: E402
+from rich.prompt import Confirm  # noqa: E402
+
+from quantalogic.agent import Agent  # noqa: E402
 
 # Local application imports
-from quantalogic.agent_config import (
+from quantalogic.agent_config import (  # noqa: E402
     MODEL_NAME,
     create_coding_agent,
     create_full_agent,
     create_interpreter_agent,
     create_orchestrator_agent,
 )
-from quantalogic.interactive_text_editor import get_multiline_input
-from quantalogic.print_event import console_print_events
-from quantalogic.version import get_version
+from quantalogic.interactive_text_editor import get_multiline_input  # noqa: E402
+from quantalogic.print_event import console_print_events  # noqa: E402
+from quantalogic.version import get_version  # noqa: E402
 
 AGENT_MODES = ["code", "basic", "interpreter", "full", "code-basic"]
 
 
 def create_agent_for_mode(mode: str, model_name: str) -> Agent:
     """Create an agent based on the specified mode."""
+    logger.debug(f"Creating agent for mode: {mode} with model: {model_name}")
     if mode == "code":
+        logger.debug("Creating code agent without basic mode")
         return create_coding_agent(model_name, basic=False)
     if mode == "code-basic":
         return create_coding_agent(model_name, basic=True)
@@ -47,6 +54,13 @@ def create_agent_for_mode(mode: str, model_name: str) -> Agent:
 def switch_verbose(verbose_mode: bool) -> None:
     import litellm
 
+    if verbose_mode:
+        logger.add(sys.stderr, level="DEBUG", format="{time} | {level} | {message}")
+        logger.info("Verbose mode enabled.")
+    else:
+        logger.info("Verbose mode disabled.")
+        logger.remove()
+
     litellm.set_verbose = verbose_mode
 
 
@@ -123,8 +137,11 @@ def task(file: Optional[str], model_name: str, verbose: bool, mode: str, task: O
                 task_content = task
             else:
                 display_welcome_message(console, model_name)
+                logger.info("Waiting for user input...")
                 task_content = get_multiline_input(console).strip()
+                logger.info(f"User input received. Task content: {task_content}")
                 if not task_content:
+                    logger.info("No task provided. Exiting...")
                     console.print("[yellow]No task provided. Exiting...[/yellow]")
                     sys.exit(2)
 
@@ -140,23 +157,31 @@ def task(file: Optional[str], model_name: str, verbose: bool, mode: str, task: O
                 console.print("[yellow]Task submission cancelled. Exiting...[/yellow]")
                 sys.exit(0)
 
+        logger.debug(f"Creating agent for mode: {mode} with model: {model_name}")
         agent = create_agent_for_mode(mode, model_name)
+        logger.debug(f"Created agent for mode: {mode} with model: {model_name}")
+
+        events = [
+            "task_start",
+            "task_think_start",
+            "task_think_end",
+            "task_complete",
+            "tool_execution_start",
+            "tool_execution_end",
+            "error_max_iterations_reached",
+            "memory_full",
+            "memory_compacted",
+            "memory_summary",
+        ]
         agent.event_emitter.on(
-            [
-                "task_complete",
-                "task_think_start",
-                "task_think_end",
-                "tool_execution_start",
-                "tool_execution_end",
-                "error_max_iterations_reached",
-                "memory_full",
-                "memory_compacted",
-                "memory_summary",
-            ],
-            console_print_events,
+            event=events,
+            listener=console_print_events,
         )
+        logger.debug("Registered event handlers for agent events with events: {events}")
 
+        logger.debug(f"Solving task with agent: {task_content}")
         result = agent.solve_task(task=task_content, max_iterations=300)
+        logger.debug(f"Task solved with result: {result}")
 
         console.print(
             Panel.fit(
@@ -166,6 +191,7 @@ def task(file: Optional[str], model_name: str, verbose: bool, mode: str, task: O
 
     except Exception as e:
         console.print(f"[red]{str(e)}[/red]")
+        logger.error(f"Error in task execution: {e}", exc_info=True)
         sys.exit(1)
 
 

{quantalogic-0.2.2 → quantalogic-0.2.4}/quantalogic/version.py

@@ -1,4 +1,4 @@
 
-VERSION = "0.2.2"
+VERSION = "0.2.4"
 def get_version() -> str:
     return VERSION