quantalogic 0.2.5__tar.gz → 0.2.6__tar.gz

{quantalogic-0.2.5 → quantalogic-0.2.6}/PKG-INFO

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

{quantalogic-0.2.5 → quantalogic-0.2.6}/pyproject.toml

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

{quantalogic-0.2.5 → quantalogic-0.2.6}/quantalogic/main.py

@@ -51,17 +51,27 @@ def create_agent_for_mode(mode: str, model_name: str) -> Agent:
         raise ValueError(f"Unknown agent mode: {mode}")
 
 
-def switch_verbose(verbose_mode: bool) -> None:
+def configure_logger(log_level: str) -> None:
+    """Configure the logger with the specified log level and format."""
+    logger.remove()
+    logger.add(sys.stderr, level=log_level.upper(), format="{time} | {level} | {message}")
+    logger.info(f"Log level set to: {log_level}")
+
+
+def set_litellm_verbose(verbose_mode: bool) -> None:
+    """Set the verbosity of the litellm library."""
     import litellm
+    litellm.set_verbose = verbose_mode
 
-    if verbose_mode:
-        logger.add(sys.stderr, level="DEBUG", format="{time} | {level} | {message}")
-        logger.info("Verbose mode enabled.")
+
+def switch_verbose(verbose_mode: bool, log_level: str = "info") -> None:
+    """Switch verbose mode and configure logger and litellm verbosity."""
+    if log_level == "debug":
+        configure_logger("DEBUG")
     else:
-        logger.info("Verbose mode disabled.")
-        logger.remove()
+        configure_logger(log_level)
 
-    litellm.set_verbose = verbose_mode
+    set_litellm_verbose(verbose_mode)
 
 
 def get_task_from_file(file_path: str) -> str:
@@ -103,15 +113,23 @@ def display_welcome_message(console: Console, model_name: str) -> None:
 
 @click.group(invoke_without_command=True)
 @click.option("--version", is_flag=True, help="Show version information.")
+@click.option(
+    "--model-name",
+    default=MODEL_NAME,
+    help='Specify the model to use (litellm format, e.g. "openrouter/deepseek-chat").',
+)
+@click.option("--log", type=click.Choice(["info", "debug", "warning"]), default="info", help="Set logging level (info/debug/warning).")
+@click.option("--verbose", is_flag=True, help="Enable verbose output.")
+@click.option("--mode", type=click.Choice(AGENT_MODES), default="code", help="Agent mode (code/search/full).")
 @click.pass_context
-def cli(ctx: click.Context, version: bool) -> None:
+def cli(ctx: click.Context, version: bool, model_name: str, verbose: bool, mode: str, log: str) -> None:
     """QuantaLogic AI Assistant - A powerful AI tool for various tasks."""
     if version:
         console = Console()
         console.print(f"QuantaLogic version: {get_version()}")
         sys.exit(0)
     if ctx.invoked_subcommand is None:
-        ctx.invoke(task)
+        ctx.invoke(task, model_name=model_name, verbose=verbose, mode=mode, log=log)
 
 
 @cli.command()
@@ -124,10 +142,10 @@ def cli(ctx: click.Context, version: bool) -> None:
 @click.option("--verbose", is_flag=True, help="Enable verbose output.")
 @click.option("--mode", type=click.Choice(AGENT_MODES), default="code", help="Agent mode (code/search/full).")
 @click.argument("task", required=False)
-def task(file: Optional[str], model_name: str, verbose: bool, mode: str, task: Optional[str]) -> None:
+def task(file: Optional[str], model_name: str, verbose: bool, mode: str, task: Optional[str], log: str) -> None:
     """Execute a task with the QuantaLogic AI Assistant."""
     console = Console()
-    switch_verbose(verbose)
+    switch_verbose(verbose, log)
 
     try:
         if file:

{quantalogic-0.2.5 → quantalogic-0.2.6}/quantalogic/version.py

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