quantalogic 0.33.4__py3-none-any.whl → 0.40.0__py3-none-any.whl

quantalogic/interactive_text_editor.py

@@ -24,54 +24,65 @@ class InputHistoryManager:
             return True
         return False
 
+
 class CommandRegistry:
     def __init__(self):
         self.commands = {}
-        
+
     def register(self, name: str, help_text: str = "") -> callable:
         def decorator(func: callable) -> callable:
-            self.commands[name] = {
-                "handler": func,
-                "help": func.__doc__ or help_text
-            }
+            self.commands[name] = {"handler": func, "help": func.__doc__ or help_text}
             return func
+
         return decorator
 
+
 registry = CommandRegistry()
 
+
 @registry.register("/help", "Show available commands")
-def handle_help_command(lines: List[str], args: List[str], console: Console, 
-                      session: PromptSession, history_manager: InputHistoryManager) -> None:
+def handle_help_command(
+    lines: List[str], args: List[str], console: Console, session: PromptSession, history_manager: InputHistoryManager
+) -> None:
     """Display auto-generated help from registered commands."""
     help_content = "\n".join([f"  {name}: {cmd['help']}" for name, cmd in registry.commands.items()])
     console.print(Panel(f"Available commands:\n{help_content}", title="Help Menu", border_style="green"))
 
+
 @registry.register("/date", "Show current date/time")
-def handle_date_command(lines: List[str], args: List[str], console: Console,
-                      session: PromptSession, history_manager: InputHistoryManager) -> None:
+def handle_date_command(
+    lines: List[str], args: List[str], console: Console, session: PromptSession, history_manager: InputHistoryManager
+) -> None:
     """Display current date and time."""
     from datetime import datetime
+
     console.print(f"[bold #ffaa00]Current datetime: {datetime.now().isoformat()}[/bold #ffaa00]")
 
+
 @registry.register("/edit", "Edit specific line: /edit <line_number>")
-def handle_edit_command(lines: List[str], args: List[str], console: Console,
-                      session: PromptSession, history_manager: InputHistoryManager) -> None:
+def handle_edit_command(
+    lines: List[str], args: List[str], console: Console, session: PromptSession, history_manager: InputHistoryManager
+) -> None:
     """Edit a specific line in the input buffer."""
     try:
         edit_line_num = int(args[0]) - 1
         if 0 <= edit_line_num < len(lines):
-            console.print(f"[bold #1d3557]Editing Line {edit_line_num + 1}:[/bold #1d3557] {lines[edit_line_num]}")  # Dark blue
+            console.print(
+                f"[bold #1d3557]Editing Line {edit_line_num + 1}:[/bold #1d3557] {lines[edit_line_num]}"
+            )  # Dark blue
             new_line = session.prompt("New content: ")
             history_manager.push_state(lines)
             lines[edit_line_num] = new_line
         else:
             console.print("[bold #ff4444]Invalid line number.[/bold #ff4444]")
     except (ValueError, IndexError):
-            console.print("[bold #ff4444]Invalid edit command. Usage: /edit <line_number>[/bold #ff4444]")
+        console.print("[bold #ff4444]Invalid edit command. Usage: /edit <line_number>[/bold #ff4444]")
+
 
 @registry.register("/delete", "Delete specific line: /delete <line_number>")
-def handle_delete_command(lines: List[str], args: List[str], console: Console,
-                        session: PromptSession, history_manager: InputHistoryManager) -> None:
+def handle_delete_command(
+    lines: List[str], args: List[str], console: Console, session: PromptSession, history_manager: InputHistoryManager
+) -> None:
     """Delete a specific line from the input buffer."""
     try:
         delete_line_num = int(args[0]) - 1
@@ -82,11 +93,13 @@ def handle_delete_command(lines: List[str], args: List[str], console: Console,
         else:
             console.print("[red]Invalid line number.[/red]")
     except (ValueError, IndexError):
-            console.print("[bold #ff4444]Invalid delete command. Usage: /delete <line_number>[/bold #ff4444]")
+        console.print("[bold #ff4444]Invalid delete command. Usage: /delete <line_number>[/bold #ff4444]")
+
 
 @registry.register("/replace", "Search and replace: /replace <search> <replace>")
-def handle_replace_command(lines: List[str], args: List[str], console: Console,
-    session: PromptSession, history_manager: InputHistoryManager) -> None:
+def handle_replace_command(
+    lines: List[str], args: List[str], console: Console, session: PromptSession, history_manager: InputHistoryManager
+) -> None:
     try:
         search_str = args[0]
         replace_str = args[1]
@@ -95,12 +108,17 @@ def handle_replace_command(lines: List[str], args: List[str], console: Console,
             lines[i] = lines[i].replace(search_str, replace_str)
             console.print("[bold #00cc66]Search and replace completed.[/bold #00cc66]")
     except (ValueError, IndexError):
-            console.print("[bold #ff4444]Invalid replace command. Usage: /replace <search_str> <replace_str>[/bold #ff4444]")
+        console.print(
+            "[bold #ff4444]Invalid replace command. Usage: /replace <search_str> <replace_str>[/bold #ff4444]"
+        )
+
 
-@registry.register("/model", "Show current AI model") 
-def handle_model_command(lines: List[str], args: List[str], console: Console,
-    session: PromptSession, history_manager: InputHistoryManager) -> None:
+@registry.register("/model", "Show current AI model")
+def handle_model_command(
+    lines: List[str], args: List[str], console: Console, session: PromptSession, history_manager: InputHistoryManager
+) -> None:
     from quantalogic.agent_factory import AgentRegistry
+
     try:
         current_agent = AgentRegistry.get_agent("main_agent")
         if current_agent:
@@ -108,17 +126,20 @@ def handle_model_command(lines: List[str], args: List[str], console: Console,
         else:
             console.print("[bold #ffaa00]No active agent found.[/bold #ffaa00]")
     except ValueError as e:
-            console.print(f"[bold #ff4444]Error: {str(e)}[/bold #ff4444]")
+        console.print(f"[bold #ff4444]Error: {str(e)}[/bold #ff4444]")
+
 
 @registry.register("/setmodel", "Set AI model name: /setmodel <name>")
-def handle_set_model_command(lines: List[str], args: List[str], console: Console,
-    session: PromptSession, history_manager: InputHistoryManager) -> None:
+def handle_set_model_command(
+    lines: List[str], args: List[str], console: Console, session: PromptSession, history_manager: InputHistoryManager
+) -> None:
     from quantalogic.agent_factory import AgentRegistry
+
     try:
         if len(args) < 1:
             console.print("[bold #ff4444]Error: Model name required. Usage: /setmodel <name>[/bold #ff4444]")
             return
-            
+
         model_name = args[0]
         current_agent = AgentRegistry.get_agent("main_agent")
         if current_agent:
@@ -129,29 +150,32 @@ def handle_set_model_command(lines: List[str], args: List[str], console: Console
     except ValueError as e:
         console.print(f"[red]Error: {str(e)}[/red]")
 
+
 @registry.register("/models", "List all available AI models")
-def handle_models_command(lines: List[str], args: List[str], console: Console,
-    session: PromptSession, history_manager: InputHistoryManager) -> None:
+def handle_models_command(
+    lines: List[str], args: List[str], console: Console, session: PromptSession, history_manager: InputHistoryManager
+) -> None:
     """Display all available AI models supported by the system."""
     from quantalogic.utils.get_all_models import get_all_models
+
     try:
         models = get_all_models()
         if models:
             # Group models by provider
             provider_groups = {}
             for model in models:
-                provider = model.split('/')[0] if '/' in model else 'default'
+                provider = model.split("/")[0] if "/" in model else "default"
                 if provider not in provider_groups:
                     provider_groups[provider] = []
                 provider_groups[provider].append(model)
-            
+
             # Create formatted output
             output = "[bold #00cc66]Available AI Models:[/bold #00cc66]\n"
             for provider, model_list in provider_groups.items():
                 output += f"\n[bold #ffaa00]{provider.upper()}[/bold #ffaa00]\n"
                 for model in sorted(model_list):
                     output += f"  • {model}\n"
-            
+
             console.print(Panel(output, border_style="green"))
         else:
             console.print("[yellow]No models available.[/yellow]")
@@ -163,8 +187,7 @@ def get_multiline_input(console: Console) -> str:
     """Get multiline input with slash command support."""
     console.print(
         Panel(
-            "Enter your task. Press [bold]Enter[/bold] twice to submit.\n"
-            "Type /help for available commands",
+            "Enter your task. Press [bold]Enter[/bold] twice to submit.\n" "Type /help for available commands",
             title="Multi-line Input",
             border_style="blue",
         )
@@ -182,16 +205,15 @@ def get_multiline_input(console: Console) -> str:
         if history_manager.undo(lines):
             console.print("[bold #00cc66]Undo successful.[/bold #00cc66]")
 
-
     from prompt_toolkit.completion import Completer, Completion
     from prompt_toolkit.styles import Style
-    
+
     class CommandCompleter(Completer):
         def get_completions(self, document, complete_event):
             text = document.text_before_cursor
             line_parts = text.split()
-            
-            if len(line_parts) == 0 or (len(line_parts) == 1 and text.endswith(' ')):
+
+            if len(line_parts) == 0 or (len(line_parts) == 1 and text.endswith(" ")):
                 for cmd, details in registry.commands.items():
                     yield Completion(
                         cmd,
@@ -203,9 +225,8 @@ def get_multiline_input(console: Console) -> str:
                 cmd = registry.commands[line_parts[0]]
                 arg_index = len(line_parts) - 1
                 if arg_index == 1:
-                    doc = cmd['handler'].__doc__ or ""
-                    args_hint = next((line.split(':')[1].strip() for line in doc.split('\n') 
-                                    if 'Args:' in line), "")
+                    doc = cmd["handler"].__doc__ or ""
+                    args_hint = next((line.split(":")[1].strip() for line in doc.split("\n") if "Args:" in line), "")
                     if args_hint:
                         yield Completion(
                             f"<{args_hint}>",
@@ -214,42 +235,43 @@ def get_multiline_input(console: Console) -> str:
                             display=f"Expected argument: {args_hint}",
                         )
             else:
-                if text.startswith('/'):
-                    partial = text[1:].lstrip('/')
+                if text.startswith("/"):
+                    partial = text[1:].lstrip("/")
                     exact_matches = []
                     prefix_matches = []
-                    
+
                     for cmd in registry.commands:
                         cmd_without_slash = cmd[1:]
                         if cmd_without_slash.lower() == partial.lower():
                             exact_matches.append(cmd)
                         elif cmd_without_slash.lower().startswith(partial.lower()):
                             prefix_matches.append(cmd)
-                    
+
                     # Prioritize exact matches first
                     for match in exact_matches:
-                        remaining = match[len('/' + partial):]
+                        remaining = match[len("/" + partial) :]
                         yield Completion(
                             remaining,
                             start_position=0,  # Corrected from -len(partial)
                             display=f"{match} - {registry.commands[match]['help']}",
                             style="fg:ansiyellow bold",
                         )
-                    
+
                     # Then prefix matches
                     for match in prefix_matches:
-                        remaining = match[len('/' + partial):]
+                        remaining = match[len("/" + partial) :]
                         yield Completion(
                             remaining,
                             start_position=0,  # Corrected from -len(partial)
                             display=f"{match} - {registry.commands[match]['help']}",
                             style="fg:ansiyellow bold",
                         )
+
             def get_completions(self, document, complete_event):
                 text = document.text_before_cursor
                 line_parts = text.split()
-                
-                if len(line_parts) == 0 or (len(line_parts) == 1 and text.endswith(' ')):
+
+                if len(line_parts) == 0 or (len(line_parts) == 1 and text.endswith(" ")):
                     for cmd, details in registry.commands.items():
                         yield Completion(
                             cmd,
@@ -261,9 +283,10 @@ def get_multiline_input(console: Console) -> str:
                     cmd = registry.commands[line_parts[0]]
                     arg_index = len(line_parts) - 1
                     if arg_index == 1:
-                        doc = cmd['handler'].__doc__ or ""
-                        args_hint = next((line.split(':')[1].strip() for line in doc.split('\n') 
-                                        if 'Args:' in line), "")
+                        doc = cmd["handler"].__doc__ or ""
+                        args_hint = next(
+                            (line.split(":")[1].strip() for line in doc.split("\n") if "Args:" in line), ""
+                        )
                         if args_hint:
                             yield Completion(
                                 f"<{args_hint}>",
@@ -272,29 +295,29 @@ def get_multiline_input(console: Console) -> str:
                                 display=f"Expected argument: {args_hint}",
                             )
                 else:
-                    if text.startswith('/'):
-                        partial = text[1:].lstrip('/')
+                    if text.startswith("/"):
+                        partial = text[1:].lstrip("/")
                         exact_matches = []
                         prefix_matches = []
-                        
+
                         for cmd in registry.commands:
                             cmd_without_slash = cmd[1:]
                             if cmd_without_slash.lower() == partial.lower():
                                 exact_matches.append(cmd)
                             elif cmd_without_slash.lower().startswith(partial.lower()):
                                 prefix_matches.append(cmd)
-                        
+
                         for match in exact_matches:
-                            remaining = match[len('/' + partial):]
+                            remaining = match[len("/" + partial) :]
                             yield Completion(
                                 remaining,
                                 start_position=-len(partial),
                                 display=f"{match} - {registry.commands[match]['help']}",
                                 style="fg:ansiyellow bold",
                             )
-                        
+
                         for match in prefix_matches:
-                            remaining = match[len('/' + partial):]
+                            remaining = match[len("/" + partial) :]
                             yield Completion(
                                 remaining,
                                 start_position=-len(partial),
@@ -303,20 +326,22 @@ def get_multiline_input(console: Console) -> str:
                             )
 
     command_completer = CommandCompleter()
-    
+
     def get_command_help(cmd_name: str) -> str:
         """Get formatted help text for a command"""
         if cmd := registry.commands.get(cmd_name):
             return f"[bold]{cmd_name}[/bold]: {cmd['help']}\n{cmd['handler'].__doc__ or ''}"
         return ""
 
-    custom_style = Style.from_dict({
-        'completion-menu.completion': 'bg:#005577 #ffffff',
-        'completion-menu.completion.current': 'bg:#007799 #ffffff bold',
-        'scrollbar.background': 'bg:#6699aa',
-        'scrollbar.button': 'bg:#444444',
-        'documentation': 'bg:#003366 #ffffff',
-    })
+    custom_style = Style.from_dict(
+        {
+            "completion-menu.completion": "bg:#005577 #ffffff",
+            "completion-menu.completion.current": "bg:#007799 #ffffff bold",
+            "scrollbar.background": "bg:#6699aa",
+            "scrollbar.button": "bg:#444444",
+            "documentation": "bg:#003366 #ffffff",
+        }
+    )
 
     session = PromptSession(
         history=InMemoryHistory(),
@@ -325,9 +350,11 @@ def get_multiline_input(console: Console) -> str:
         completer=command_completer,
         complete_while_typing=True,
         style=custom_style,
-        bottom_toolbar=lambda: get_command_help(session.default_buffer.document.text.split()[0][1:] 
-                                              if session.default_buffer.document.text.startswith('/') 
-                                              else ""),
+        bottom_toolbar=lambda: get_command_help(
+            session.default_buffer.document.text.split()[0][1:]
+            if session.default_buffer.document.text.startswith("/")
+            else ""
+        ),
     )
 
     try:
@@ -335,11 +362,11 @@ def get_multiline_input(console: Console) -> str:
             prompt_text = f"{line_number:>3}: "
             line = session.prompt(prompt_text, rprompt="Press Enter twice to submit")
 
-            if line.strip().startswith('/'):
+            if line.strip().startswith("/"):
                 cmd_parts = line.strip().split()
                 cmd_name = cmd_parts[0].lower()
                 args = cmd_parts[1:]
-                
+
                 if cmd_handler := registry.commands.get(cmd_name):
                     try:
                         cmd_handler["handler"](lines, args, console, session, history_manager)
@@ -365,4 +392,4 @@ def get_multiline_input(console: Console) -> str:
         console.print("\n[bold]Input cancelled by user.[/bold]")
         return ""
 
-    return "\n".join(lines)
+    return "\n".join(lines)

quantalogic/main.py

@@ -33,7 +33,7 @@ from quantalogic.utils.get_all_models import get_all_models  # noqa: E402
 
 # Platform-specific imports
 try:
-    if sys.platform == 'win32':
+    if sys.platform == "win32":
         import msvcrt  # Built-in Windows module
     else:
         import termios
@@ -50,7 +50,7 @@ AGENT_MODES = ["code", "basic", "interpreter", "full", "code-basic", "search", "
 
 def setup_terminal():
     """Configure terminal settings based on platform."""
-    if sys.platform == 'win32':
+    if sys.platform == "win32":
         if msvcrt:
             return None  # Windows terminal is already configured
         logger.warning("msvcrt module not available on Windows")
@@ -70,7 +70,7 @@ def setup_terminal():
 
 def restore_terminal(old_settings):
     """Restore terminal settings based on platform."""
-    if sys.platform != 'win32' and termios and old_settings:
+    if sys.platform != "win32" and termios and old_settings:
         try:
             termios.tcsetattr(sys.stdin.fileno(), termios.TCSADRAIN, old_settings)
         except (termios.error, AttributeError) as e:
@@ -88,14 +88,14 @@ def restore_terminal(old_settings):
 @click.option(
     "--model-name",
     default=MODEL_NAME,
-    help='Specify the model to use (litellm format). Examples:\n'
-         '  - openai/gpt-4o-mini\n'
-         '  - openai/gpt-4o\n'
-         '  - anthropic/claude-3.5-sonnet\n'
-         '  - deepseek/deepseek-chat\n'
-         '  - deepseek/deepseek-reasoner\n'
-         '  - openrouter/deepseek/deepseek-r1\n'
-         '  - openrouter/openai/gpt-4o',
+    help="Specify the model to use (litellm format). Examples:\n"
+    "  - openai/gpt-4o-mini\n"
+    "  - openai/gpt-4o\n"
+    "  - anthropic/claude-3.5-sonnet\n"
+    "  - deepseek/deepseek-chat\n"
+    "  - deepseek/deepseek-reasoner\n"
+    "  - openrouter/deepseek/deepseek-r1\n"
+    "  - openrouter/openai/gpt-4o",
 )
 @click.option(
     "--log",
@@ -122,6 +122,12 @@ def restore_terminal(old_settings):
     default=None,
     help="Set the maximum number of tokens allowed in the working memory.",
 )
+@click.option(
+    "--thinking-model",
+    type=str,
+    default="default",
+    help="The thinking model to use",
+)
 @click.pass_context
 def cli(
     ctx: click.Context,
@@ -134,6 +140,7 @@ def cli(
     max_iterations: int,
     compact_every_n_iteration: int | None,
     max_tokens_working_memory: int | None,
+    thinking_model: str,
 ) -> None:
     """QuantaLogic AI Assistant - A powerful AI tool for various tasks.
 
@@ -161,7 +168,8 @@ def cli(
             max_iterations=max_iterations,
             compact_every_n_iteration=compact_every_n_iteration,
             max_tokens_working_memory=max_tokens_working_memory,
-            no_stream=False  # Default value for backward compatibility
+            no_stream=False,  # Default value for backward compatibility
+            thinking_model_name=thinking_model,
         )
         ctx.invoke(
             task,
@@ -173,6 +181,7 @@ def cli(
             max_iterations=config.max_iterations,
             compact_every_n_iteration=config.compact_every_n_iteration,
             max_tokens_working_memory=config.max_tokens_working_memory,
+            thinking_model=thinking_model,
         )
 
 
@@ -219,6 +228,12 @@ def cli(
     is_flag=True,
     help="Disable streaming output (default: streaming enabled).",
 )
+@click.option(
+    "--thinking-model",
+    type=str,
+    default="default",
+    help="The thinking model to use",
+)
 @click.argument("task", required=False)
 def task(
     file: Optional[str],
@@ -232,6 +247,7 @@ def task(
     compact_every_n_iteration: int | None,
     max_tokens_working_memory: int | None,
     no_stream: bool,
+    thinking_model: str,
 ) -> None:
     console = Console()
 
@@ -246,8 +262,9 @@ def task(
             compact_every_n_iteration=compact_every_n_iteration,
             max_tokens_working_memory=max_tokens_working_memory,
             no_stream=no_stream,
+            thinking_model_name=thinking_model,
         )
-        
+
         task_runner(
             console,
             file,
@@ -264,29 +281,25 @@ def task(
 @click.option("--search", type=str, help="Fuzzy search for models containing the given string.")
 def list_models(search: Optional[str] = None):
     """List supported LiteLLM models with optional fuzzy search.
-    
+
     If a search term is provided, it will return models that closely match the term.
     """
     console = Console()
     all_models = get_all_models()
-    
+
     if search:
         # Perform fuzzy matching
         matched_models = process.extractBests(search, all_models, limit=None, score_cutoff=70)
         models = [model for model, score in matched_models]
     else:
         models = all_models
-    
-    console.print(Panel(
-        f"Total Models: {len(models)} "
-        f"({len(all_models)} total)",
-        title="Supported LiteLLM Models"
-    ))
-    
+
+    console.print(Panel(f"Total Models: {len(models)} " f"({len(all_models)} total)", title="Supported LiteLLM Models"))
+
     if not models:
         console.print(f"[yellow]No models found matching '[bold]{search}[/bold]'[/yellow]")
         return
-    
+
     for model in sorted(models):
         console.print(f"- {model}")
 

quantalogic/model_info_list.py

@@ -57,4 +57,16 @@ model_info = {
         max_output_tokens=8 * 1024,
         max_input_tokens=1024 * 64,
     ),
+    "gemini/gemini-2.0-flash": ModelInfo(
+        model_name="gemini/gemini-2.0-flash",
+        max_input_tokens=1000000,
+        max_output_tokens=8 * 1024,
+        input_cost_per_token=0.0000001,
+    ),
+    "openrouter/google/gemini-2.0-flash-001": ModelInfo(
+        model_name="openrouter/google/gemini-2.0-flash-001",
+        max_input_tokens=1000000,
+        max_output_tokens=8 * 1024,
+        input_cost_per_token=0.0000001,
+    ),
 }

quantalogic/model_info_litellm.py

@@ -6,18 +6,18 @@ import litellm
 @functools.lru_cache(maxsize=32)
 def litellm_get_model_info(model_name: str) -> dict | None:
     """Get model information with prefix fallback logic using only litellm.
-    
+
     Args:
         model_name: The model identifier to get information for
-        
+
     Returns:
         Dictionary containing model information
-        
+
     Raises:
         ValueError: If model info cannot be found after prefix fallbacks
     """
     tried_models = [model_name]
-    
+
     while True:
         try:
             # Attempt to get model info through litellm
@@ -26,45 +26,45 @@ def litellm_get_model_info(model_name: str) -> dict | None:
                 return info
         except Exception:
             pass
-        
+
         # Try removing one prefix level
         parts = model_name.split("/")
         if len(parts) <= 1:
             break
-            
+
         model_name = "/".join(parts[1:])
         tried_models.append(model_name)
 
-    return None    
+    return None
 
 
 def litellm_get_model_max_input_tokens(model_name: str) -> int | None:
     """Get maximum input tokens for a model using litellm.
-    
+
     Args:
         model_name: The model identifier
-        
+
     Returns:
         Maximum input tokens or None if not found
     """
     try:
         info = litellm_get_model_info(model_name)
         return info.get("max_input_tokens", 8192)
-    except Exception as e:
+    except Exception:
         return 8192  # Default for many modern models
 
 
 def litellm_get_model_max_output_tokens(model_name: str) -> int | None:
     """Get maximum output tokens for a model using litellm.
-    
+
     Args:
         model_name: The model identifier
-        
+
     Returns:
         Maximum output tokens or None if not found
     """
     try:
         info = litellm_get_model_info(model_name)
         return info.get("max_output_tokens", 4096)
-    except Exception as e:
-        return 4096  # Conservative default
+    except Exception:
+        return 4096  # Conservative default

quantalogic/prompts.py

@@ -93,5 +93,6 @@ Task Format: <task>task_description</task>
 5. 🔍 Monitor and validate each action's impact
 6. 🛑 Fail fast and adapt when encountering blockers
 7. ✅ Verify completion criteria rigorously
-"""
+8. ✅ Return complete, full and usable results.
 
+"""