quantalogic 0.31.0__py3-none-any.whl → 0.31.1__py3-none-any.whl

quantalogic/__init__.py

@@ -11,12 +11,12 @@ warnings.filterwarnings(
 )
 
 # Import public API
-from .llm import generate_completion, generate_image, count_tokens  # noqa: E402
 from .agent import Agent  # noqa: E402
-from .event_emitter import EventEmitter  # noqa: E402
-from .memory import AgentMemory, VariableMemory  # noqa: E402
 from .console_print_events import console_print_events  # noqa: E402
 from .console_print_token import console_print_token  # noqa: E402
+from .event_emitter import EventEmitter  # noqa: E402
+from .llm import count_tokens, generate_completion, generate_image  # noqa: E402
+from .memory import AgentMemory, VariableMemory  # noqa: E402
 
 __all__ = [
     "Agent",

quantalogic/agent.py

@@ -412,6 +412,9 @@ class Agent(BaseModel):
 
     def _parse_tool_usage(self, content: str) -> dict:
         """Extract tool usage from the response content."""
+        if not content or not isinstance(content, str):
+            return {}
+        
         xml_parser = ToleranceXMLParser()
         tool_names = self.tools.tool_names()
         return xml_parser.extract_elements(text=content, element_names=tool_names)
@@ -502,6 +505,7 @@ class Agent(BaseModel):
 
         # Format the response message
         formatted_response = (
+            f"Your next step: you Must now plan the next tool call to complete the based on this new observation\n"
             f"\n--- Observations for iteration {iteration} / max {self.max_iterations} ---\n"
             f"\n--- Tool execution result in ${variable_name}$ ---\n"
             f"<{variable_name}>\n{response_display}\n</{variable_name}>\n\n"

quantalogic/get_model_info.py

@@ -6,6 +6,8 @@ model_info = {
     "openrouter/deepseek/deepseek-r1": {"max_output_tokens": 8 * 1024, "max_input_tokens": 1024 * 128},
     "openrouter/mistralai/mistral-large-2411": {"max_output_tokens": 128 * 1024, "max_input_tokens": 1024 * 128},
     "mistralai/mistral-large-2411": {"max_output_tokens": 128 * 1024, "max_input_tokens": 1024 * 128},
+    "deepseek/deepseek-chat": {"max_output_tokens": 8* 1024, "max_input_tokens": 1024*64},
+    "deepseek/deepseek-reasoner": {"max_output_tokens": 8* 1024, "max_input_tokens": 1024*64, "max_cot_tokens": 1024*32 },
 }
 
 

quantalogic/tools/execute_bash_command_tool.py

@@ -11,7 +11,7 @@ from loguru import logger
 
 # Platform-specific imports
 try:
-    if sys.platform != 'win32':
+    if sys.platform != "win32":
         import pty
 except ImportError as e:
     logger.warning(f"Could not import platform-specific module: {e}")
@@ -40,6 +40,7 @@ class ExecuteBashCommandTool(Tool):
             description="The working directory where the command will be executed. Defaults to the current directory.",
             required=False,
             example="/path/to/directory",
+            default=os.getcwd(),
         ),
         ToolArgument(
             name="timeout",
@@ -47,6 +48,7 @@ class ExecuteBashCommandTool(Tool):
             description="Maximum time in seconds to wait for the command to complete. Defaults to 60 seconds.",
             required=False,
             example="60",
+            default="60",
         ),
     ]
 
@@ -69,16 +71,16 @@ class ExecuteBashCommandTool(Tool):
                 cwd=cwd,
                 env=env_vars,
                 text=True,
-                encoding='utf-8'
+                encoding="utf-8",
             )
 
             try:
                 stdout, stderr = process.communicate(timeout=timeout_seconds)
                 return_code = process.returncode
-                
+
                 if return_code != 0 and stderr:
                     logger.warning(f"Command failed with error: {stderr}")
-                
+
                 formatted_result = (
                     "<command_output>"
                     f" <stdout>{stdout.strip()}</stdout>"
@@ -86,11 +88,11 @@ class ExecuteBashCommandTool(Tool):
                     f"</command_output>"
                 )
                 return formatted_result
-            
+
             except subprocess.TimeoutExpired:
                 process.kill()
                 return f"Command timed out after {timeout_seconds} seconds."
-                
+
         except Exception as e:
             return f"Unexpected error executing command: {str(e)}"
 
@@ -127,7 +129,7 @@ class ExecuteBashCommandTool(Tool):
                         if proc.poll() is not None:
                             break  # Process completed but select timed out
                         raise subprocess.TimeoutExpired(command, timeout_seconds)
-                    
+
                     for fd in rlist:
                         if fd == master:
                             data = os.read(master, 1024).decode()
@@ -140,7 +142,7 @@ class ExecuteBashCommandTool(Tool):
                         elif fd == sys.stdin:
                             user_input = os.read(sys.stdin.fileno(), 1024)
                             os.write(master, user_input)
-                    
+
                     if break_loop or proc.poll() is not None:
                         while True:
                             data = os.read(master, 1024).decode()
@@ -160,7 +162,7 @@ class ExecuteBashCommandTool(Tool):
                 os.close(master)
                 proc.wait()
 
-            stdout_content = ''.join(stdout_buffer)
+            stdout_content = "".join(stdout_buffer)
             return_code = proc.returncode
             formatted_result = (
                 "<command_output>"
@@ -186,7 +188,7 @@ class ExecuteBashCommandTool(Tool):
         if env:
             env_vars.update(env)
 
-        if sys.platform == 'win32':
+        if sys.platform == "win32":
             return self._execute_windows(command, cwd, timeout_seconds, env_vars)
         else:
             if not pty:
@@ -197,4 +199,4 @@ class ExecuteBashCommandTool(Tool):
 
 if __name__ == "__main__":
     tool = ExecuteBashCommandTool()
-    print(tool.to_markdown())
+    print(tool.to_markdown())

quantalogic/xml_parser.py

@@ -298,5 +298,6 @@ class ToleranceXMLParser:
 
         except Exception as e:
             error_msg = f"Error extracting XML elements: {str(e)}"
+            error_msg = error_msg + f"\n{text}\n"
             logger.error(error_msg)
             raise ValueError(error_msg)

{quantalogic-0.31.0.dist-info → quantalogic-0.31.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quantalogic
-Version: 0.31.0
+Version: 0.31.1
 Summary: QuantaLogic ReAct Agents
 Author: Raphaël MANSUY
 Author-email: raphael.mansuy@gmail.com
@@ -116,7 +116,7 @@ We created [QuantaLogic](https://www.quantalogic.app) because we saw a significa
 
 ## Usage
 
-**Usage:** `quantalogic [OPTIONS] COMMAND [ARGS]...`  
+**Usage:** `quantalogic [OPTIONS] COMMAND i[ARGS]...`  
 **Environment Variables:** Set `OPENAI_API_KEY`, `ANTHROPIC_API_KEY`, and `DEEPSEEK_API_KEY` for API integration.
 
 **Options:**
@@ -131,6 +131,7 @@ We created [QuantaLogic](https://www.quantalogic.app) because we saw a significa
 - `--help`: Show help message
 
 **Commands:**
+
 - `task`: Execute a task with the QuantaLogic AI Assistant
   - `--file PATH`: Path to task file
   - `--model-name TEXT`: Specify model
@@ -143,6 +144,25 @@ We created [QuantaLogic](https://www.quantalogic.app) because we saw a significa
   - `--compact-every-n-iteration`: Memory optimization
   - `--no-stream`: Disable streaming
 
+
+- `list-models`: List available models with optional filtering.
+  - `--search TEXT`: Filter models by name or description.
+  - `--help`: Show help message.
+
+  Example:
+  ```bash
+  quantalogic list-models --search qwen
+  ```
+
+  Output:
+  ```
+  Model Name           Description
+  -------------------  -------------------------------------------------------
+  dashscope/qwen-max   Alibaba's Qwen-Max model optimized for maximum performance
+  dashscope/qwen-plus  Alibaba's Qwen-Plus model offering balanced performance
+  ```
+
+
 ## Release Notes
 
 See our [Release Notes](RELEASE_NOTES.MD) for detailed version history and changes.

{quantalogic-0.31.0.dist-info → quantalogic-0.31.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
-quantalogic/__init__.py,sha256=s8G7kW8JI_WP201l950vIHxtLHk0V7mERsHuIR--P9w,925
-quantalogic/agent.py,sha256=2qgx5mJGx12otW2KzPoM0ZSugVxSTP3sXnvilXcvnrQ,33020
+quantalogic/__init__.py,sha256=Su8CnOEdqKu4zTytjiP9P5olg-oIDuUA3fMWM1WUdRY,925
+quantalogic/agent.py,sha256=5TscD78hIQBaF9pfxIRysC3HjnTJ6e0nyKDwrZMXHak,33223
 quantalogic/agent_config.py,sha256=SIRVSF0kkrYfvtyHiMCJhnm_nYqJCD2p1pN-reMIy24,7868
 quantalogic/agent_factory.py,sha256=HWKwN_DN57EPmME-hoCD2uJE0DqsPCzGU_V7nq54XzI,5284
 quantalogic/coding_agent.py,sha256=Z7ik6LUvLKDnaW9Ax1iZGC7p1WMnlYEUIlE5lkBP414,4975
@@ -9,7 +9,7 @@ quantalogic/console_print_token.py,sha256=qSU-3kmoZk4T5-1ybrEBi8tIXDPcz7eyWKhGh3
 quantalogic/docs_cli.py,sha256=3giVbUpespB9ZdTSJ955A3BhcOaBl5Lwsn1AVy9XAeY,1663
 quantalogic/event_emitter.py,sha256=jqot2g4JRXc88K6PW837Oqxbf7shZfO-xdPaUWmzupk,7901
 quantalogic/generative_model.py,sha256=ut_BFy4BqDxNqUXVbM8e_C_CzwNuJkvGWRsbpbKaees,13423
-quantalogic/get_model_info.py,sha256=nAvb_x4MnZei9CvTxUwqFUtFKkTp4-qiKB7ZWZxLxp8,1762
+quantalogic/get_model_info.py,sha256=f64GpkpzeWXKRMBGG9edfAVP1-S-TclmxSaQC91vglw,1976
 quantalogic/interactive_text_editor.py,sha256=1vW4poJl7SItRGEeGQgtCFcmRDXmfCM8PE-uBtDBJuE,16658
 quantalogic/llm.py,sha256=yhuC1b5TCXojDXZEJK_PMcWUUxzrpI-gwzlIszAdJMM,4677
 quantalogic/main.py,sha256=__-4pX2pgoSFvt-aLdp6Qlrq55_SrwP_l8u2uTaQbjg,9262
@@ -35,7 +35,7 @@ quantalogic/tools/download_http_file_tool.py,sha256=wTfanbXjIRi5-qrbluuLvNmDNhvm
 quantalogic/tools/duckduckgo_search_tool.py,sha256=xVaEb_SUK5NL3lwMQXj1rGQYYvNT-td-qaB9QCes27Q,7014
 quantalogic/tools/edit_whole_content_tool.py,sha256=nXmpAvojvqvAcqNMy1kUKZ1ocboky_ZcnCR4SNCSPgw,2360
 quantalogic/tools/elixir_tool.py,sha256=fzPPtAW-Koy9KB0r5k2zV1f1U0WphL-LXPPOBkeNkug,7652
-quantalogic/tools/execute_bash_command_tool.py,sha256=kl3RSOZCOc-U52dwd0h6BxXvjMlAX7D0Bo2-HkCOcxo,6908
+quantalogic/tools/execute_bash_command_tool.py,sha256=YK_cMuODJDOYheZKGmlpZTxJdVbimFLCUlND2_zmyMg,6869
 quantalogic/tools/generate_database_report_tool.py,sha256=QbZjtmegGEOEZAIa-CSeBo5O9dYBZTk_PWrumyFUg1Q,1890
 quantalogic/tools/grep_app_tool.py,sha256=BDxygwx7WCbqbiP2jmSRnIsoIUVYG5A4SKzId524ys4,19957
 quantalogic/tools/input_question_tool.py,sha256=UoTlNhdmdr-eyiVtVCG2qJe_R4bU_ag-DzstSdmYkvM,1848
@@ -86,10 +86,10 @@ quantalogic/utils/read_http_text_content.py,sha256=n3IayT5KcqctIVVF2gOQQAMf3Ow6e
 quantalogic/version.py,sha256=ea_cRutaQk5_lwlLbUUvPFuOT7Of7-gAsDl7wdveS-g,107
 quantalogic/version_check.py,sha256=cttR1lR3OienGLl7NrK1Te1fhDkqSjCci7HC1vFUTSY,1627
 quantalogic/welcome_message.py,sha256=IXMhem8h7srzNUwvw8G_lmEkHU8PFfote021E_BXmVk,3039
-quantalogic/xml_parser.py,sha256=uMLQNHTRCg116FwcjRoquZmSwVtE4LEH-6V2E3RD-dA,11466
+quantalogic/xml_parser.py,sha256=8yDxvKzAEnefNwUAR-wjerMDOj5T5cxak4WPIA83SBw,11516
 quantalogic/xml_tool_parser.py,sha256=Vz4LEgDbelJynD1siLOVkJ3gLlfHsUk65_gCwbYJyGc,3784
-quantalogic-0.31.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-quantalogic-0.31.0.dist-info/METADATA,sha256=SQnNSbR3pqL1-i7F2dUruuLT650tM-qp35rktOO0FC8,22262
-quantalogic-0.31.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-quantalogic-0.31.0.dist-info/entry_points.txt,sha256=h74O_Q3qBRCrDR99qvwB4BpBGzASPUIjCfxHq6Qnups,183
-quantalogic-0.31.0.dist-info/RECORD,,
+quantalogic-0.31.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+quantalogic-0.31.1.dist-info/METADATA,sha256=pLu3KI4tGKodQWubeVQm1G2P7tC9yZp22ZpOl35N074,22789
+quantalogic-0.31.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+quantalogic-0.31.1.dist-info/entry_points.txt,sha256=h74O_Q3qBRCrDR99qvwB4BpBGzASPUIjCfxHq6Qnups,183
+quantalogic-0.31.1.dist-info/RECORD,,