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

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.3"
 def get_version() -> str:
     return VERSION

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

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

{quantalogic-0.2.2.dist-info → quantalogic-0.2.3.dist-info}/RECORD

@@ -5,7 +5,7 @@ quantalogic/coding_agent.py,sha256=ivHBn3hIDEt9V1LDXI4l-COeGXYVp_v_rrx9ng8sP2A,3
 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=drydBT0J_qpHkpmCE8_x46mY98FpK0zY8R1huMWdpA4,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.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+quantalogic-0.2.3.dist-info/METADATA,sha256=Vy2jq-b-nWZmFDYZFNjmemwaC9b2tEjKoftgCRV1H5k,38601
+quantalogic-0.2.3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+quantalogic-0.2.3.dist-info/entry_points.txt,sha256=wgSq5SRU98yvlRHGEZD1Xn7sS5CSjH2RfUtTa6Qy28Q,52
+quantalogic-0.2.3.dist-info/RECORD,,