quantalogic 0.2.2__py3-none-any.whl → 0.2.4__py3-none-any.whl

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/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/version.py

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

{quantalogic-0.2.2.dist-info → quantalogic-0.2.4.dist-info}/METADATA

@@ -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.dist-info → quantalogic-0.2.4.dist-info}/RECORD

@@ -1,11 +1,11 @@
 quantalogic/__init__.py,sha256=HFk7_19UzHzYwvPzb9QTQ4w_lPwTTPda61AYb8qggZY,686
-quantalogic/agent.py,sha256=TzSnVWatyAcfIPo-hESubB-0nlhr82-7fIneajaIG3M,25179
+quantalogic/agent.py,sha256=pvXwJepJMRvkn5DX01b2R77mBSpfQI57mi81s4SN0xY,25196
 quantalogic/agent_config.py,sha256=sNuZ0Y02h3ra2hcjAE3-8Tt-NE8QSarRsTB0auTLgn0,3595
 quantalogic/coding_agent.py,sha256=ivHBn3hIDEt9V1LDXI4l-COeGXYVp_v_rrx9ng8sP2A,3214
 quantalogic/event_emitter.py,sha256=jqot2g4JRXc88K6PW837Oqxbf7shZfO-xdPaUWmzupk,7901
 quantalogic/generative_model.py,sha256=JkZz7YBd-v9n5EURidKUddkCLNvNZaqMzo5PgE_ugmc,8398
 quantalogic/interactive_text_editor.py,sha256=kYeTA2qej5kxtPvAUHy_Dr2MhrGQAyenLFpW9mU9Rmw,6855
-quantalogic/main.py,sha256=lN2lOQeZSIFBZOkgJ8CmHA_TWrGj6OM_dB8OqNrIkhY,6317
+quantalogic/main.py,sha256=PvSf_V-BSQpFSCKSgDIB6DaUCab8ECDd0QWgnWbw0Js,7564
 quantalogic/memory.py,sha256=zbtRuM05jaS2lJll-92dt5JfYVLERnF_m_9xqp2x-k0,6304
 quantalogic/model_names.py,sha256=UZlz25zG9B2dpfwdw_e1Gw5qFsKQ7iME9FJh9Ts4u6s,938
 quantalogic/print_event.py,sha256=-4qZmFI2BTkXuGE9DoKm6Vs-GzK1F9WJGt9GqpRQlQQ,2175
@@ -59,11 +59,11 @@ quantalogic/utils/get_quantalogic_rules_content.py,sha256=fnEFTyClXzpI0MLaM-gB9R
 quantalogic/utils/git_ls.py,sha256=_aXg2TwqYv9CoOrhQ1gqHCqu1j8wOVigQNWbGncSDlM,4361
 quantalogic/utils/read_file.py,sha256=tSRVHk8dIP4nNLL89v5kRki4hOTjVyjbmuEb2zwvwCY,2077
 quantalogic/utils/read_http_text_content.py,sha256=1nRLQ9DHP_fKrm0rIEJBF0ROmB78e4lct2hUzD2PAUk,4408
-quantalogic/version.py,sha256=KCMp_VO1Gh1okgz_ysyGPoR0Sw0FpFSFimW1hvAms6s,64
+quantalogic/version.py,sha256=H_M9VsTe7G2HeD8XedH6FDu8L351L1RYdBHbbXQpNL8,64
 quantalogic/xml_parser.py,sha256=cTRorr5sVfkIzH72M0C-GQ9ROGPiz2FTT66U9ndjzhE,9538
 quantalogic/xml_tool_parser.py,sha256=lsVzClZBrZan7wjCuCKnGHWzksXI3VMy_vWthxu2_bo,3738
-quantalogic-0.2.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-quantalogic-0.2.2.dist-info/METADATA,sha256=nK98IudCDiahREJnwwdiLI-B7flWMhYrpHtVTphpFYY,38601
-quantalogic-0.2.2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-quantalogic-0.2.2.dist-info/entry_points.txt,sha256=wgSq5SRU98yvlRHGEZD1Xn7sS5CSjH2RfUtTa6Qy28Q,52
-quantalogic-0.2.2.dist-info/RECORD,,
+quantalogic-0.2.4.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+quantalogic-0.2.4.dist-info/METADATA,sha256=d93jsS31zo9A5kTwe3PsSLyvLvwRw_pnIEwC7wziHV8,38601
+quantalogic-0.2.4.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+quantalogic-0.2.4.dist-info/entry_points.txt,sha256=wgSq5SRU98yvlRHGEZD1Xn7sS5CSjH2RfUtTa6Qy28Q,52
+quantalogic-0.2.4.dist-info/RECORD,,