PyPI - quantalogic - Versions diffs - 0.2.0__py3-none-any.whl - Mend

quantalogic 0.2.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (68) hide show

quantalogic/__init__.py +20 -0
quantalogic/agent.py +638 -0
quantalogic/agent_config.py +138 -0
quantalogic/coding_agent.py +83 -0
quantalogic/event_emitter.py +223 -0
quantalogic/generative_model.py +226 -0
quantalogic/interactive_text_editor.py +190 -0
quantalogic/main.py +185 -0
quantalogic/memory.py +217 -0
quantalogic/model_names.py +19 -0
quantalogic/print_event.py +66 -0
quantalogic/prompts.py +99 -0
quantalogic/server/__init__.py +3 -0
quantalogic/server/agent_server.py +633 -0
quantalogic/server/models.py +60 -0
quantalogic/server/routes.py +117 -0
quantalogic/server/state.py +199 -0
quantalogic/server/static/js/event_visualizer.js +430 -0
quantalogic/server/static/js/quantalogic.js +571 -0
quantalogic/server/templates/index.html +134 -0
quantalogic/tool_manager.py +68 -0
quantalogic/tools/__init__.py +46 -0
quantalogic/tools/agent_tool.py +88 -0
quantalogic/tools/download_http_file_tool.py +64 -0
quantalogic/tools/edit_whole_content_tool.py +70 -0
quantalogic/tools/elixir_tool.py +240 -0
quantalogic/tools/execute_bash_command_tool.py +116 -0
quantalogic/tools/input_question_tool.py +57 -0
quantalogic/tools/language_handlers/__init__.py +21 -0
quantalogic/tools/language_handlers/c_handler.py +33 -0
quantalogic/tools/language_handlers/cpp_handler.py +33 -0
quantalogic/tools/language_handlers/go_handler.py +33 -0
quantalogic/tools/language_handlers/java_handler.py +37 -0
quantalogic/tools/language_handlers/javascript_handler.py +42 -0
quantalogic/tools/language_handlers/python_handler.py +29 -0
quantalogic/tools/language_handlers/rust_handler.py +33 -0
quantalogic/tools/language_handlers/scala_handler.py +33 -0
quantalogic/tools/language_handlers/typescript_handler.py +42 -0
quantalogic/tools/list_directory_tool.py +123 -0
quantalogic/tools/llm_tool.py +119 -0
quantalogic/tools/markitdown_tool.py +105 -0
quantalogic/tools/nodejs_tool.py +515 -0
quantalogic/tools/python_tool.py +469 -0
quantalogic/tools/read_file_block_tool.py +140 -0
quantalogic/tools/read_file_tool.py +79 -0
quantalogic/tools/replace_in_file_tool.py +300 -0
quantalogic/tools/ripgrep_tool.py +353 -0
quantalogic/tools/search_definition_names.py +419 -0
quantalogic/tools/task_complete_tool.py +35 -0
quantalogic/tools/tool.py +146 -0
quantalogic/tools/unified_diff_tool.py +387 -0
quantalogic/tools/write_file_tool.py +97 -0
quantalogic/utils/__init__.py +17 -0
quantalogic/utils/ask_user_validation.py +12 -0
quantalogic/utils/download_http_file.py +77 -0
quantalogic/utils/get_coding_environment.py +15 -0
quantalogic/utils/get_environment.py +26 -0
quantalogic/utils/get_quantalogic_rules_content.py +19 -0
quantalogic/utils/git_ls.py +121 -0
quantalogic/utils/read_file.py +54 -0
quantalogic/utils/read_http_text_content.py +101 -0
quantalogic/xml_parser.py +242 -0
quantalogic/xml_tool_parser.py +99 -0
quantalogic-0.2.0.dist-info/LICENSE +201 -0
quantalogic-0.2.0.dist-info/METADATA +1034 -0
quantalogic-0.2.0.dist-info/RECORD +68 -0
quantalogic-0.2.0.dist-info/WHEEL +4 -0
quantalogic-0.2.0.dist-info/entry_points.txt +3 -0

quantalogic/server/templates/index.html ADDED Viewed

@@ -0,0 +1,134 @@
+<!DOCTYPE html>
+<html lang="en" class="h-full bg-gray-100">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <meta http-equiv="Cache-Control" content="no-cache, no-store, must-revalidate">
+    <meta http-equiv="Pragma" content="no-cache">
+    <meta http-equiv="Expires" content="0">
+    <title>QuantaLogic AI Assistant</title>
+    <script src="https://cdn.tailwindcss.com"></script>
+    <script src="https://unpkg.com/marked/marked.min.js"></script>
+    <script src="https://cdn.jsdelivr.net/npm/prismjs@1.29.0/prism.min.js"></script>
+    <link href="https://cdn.jsdelivr.net/npm/prismjs@1.29.0/themes/prism-tomorrow.min.css" rel="stylesheet">
+    <script>
+        tailwind.config = {
+            theme: {
+                extend: {
+                    colors: {
+                        primary: {
+                            50: '#f0f9ff',
+                            100: '#e0f2fe',
+                            500: '#2563eb',
+                            600: '#1d4ed8',
+                            700: '#1e40af',
+                        },
+                    }
+                }
+            }
+        }
+    </script>
+    <script type="module" src="/static/js/quantalogic.js?v={{ range(1, 100000) | random }}"></script>
+</head>
+<body class="h-full flex flex-col">
+    <!-- Header -->
+    <header class="bg-white border-b border-gray-200 py-2">
+        <div class="max-w-7xl mx-auto px-4 sm:px-6 lg:px-8">
+            <div class="flex justify-between items-center">
+                <div class="flex items-center space-x-3">
+                    <h1 class="text-lg font-semibold text-gray-900">QuantaLogic</h1>
+                    <span class="text-sm text-gray-500">AI Assistant</span>
+                </div>
+                <div class="flex items-center space-x-3">
+                    <select id="modelSelect" class="block w-48 text-sm rounded-md border-gray-300 shadow-sm focus:border-primary-500 focus:ring-primary-500">
+                        <option value="openrouter/deepseek/deepseek-chat">DeepSeek Chat</option>
+                        <option value="ollama/qwen2.5-coder:14b">Qwen Coder</option>
+                        <option value="gpt-4o-mini">gpt-4o-mini</option>
+                    </select>
+                    <div id="connectionStatus" class="inline-flex items-center px-2 py-0.5 text-xs font-medium rounded-full"></div>
+                </div>
+            </div>
+        </div>
+    </header>
+    <!-- Main Content -->
+    <main class="flex-1 flex min-h-0">
+        <!-- Left Panel - Task Input -->
+        <div class="w-1/2 p-4 flex flex-col min-h-0">
+            <div class="bg-white rounded-lg shadow-sm p-4 flex flex-col flex-1">
+                <div class="flex items-center justify-between mb-4">
+                    <h2 class="text-lg font-semibold text-gray-900">Task Input</h2>
+                    <div class="flex items-center space-x-4">
+                        <label class="flex items-center space-x-2">
+                            <span class="text-sm text-gray-600">Max Iterations:</span>
+                            <input type="number" id="maxIterations"
+                                class="w-20 rounded-md border-gray-300 shadow-sm focus:border-primary-500 focus:ring-primary-500 sm:text-sm"
+                                value="30" min="1" max="100">
+                        </label>
+                    </div>
+                </div>
+                <textarea id="taskInput"
+                    class="flex-1 p-4 block w-full rounded-md border-gray-300 shadow-sm focus:border-primary-500 focus:ring-primary-500 sm:text-sm resize-none"
+                    placeholder="Describe your task here..."></textarea>
+                <div class="mt-4 flex justify-end">
+                    <button id="submitTask"
+                        class="inline-flex items-center px-4 py-2 border border-transparent text-sm font-medium rounded-md shadow-sm text-white bg-primary-600 hover:bg-primary-700 focus:outline-none focus:ring-2 focus:ring-offset-2 focus:ring-primary-500 disabled:opacity-50 disabled:cursor-not-allowed">
+                        <span>Submit Task</span>
+                        <div class="loading-indicator hidden ml-2">
+                            <svg class="animate-spin h-4 w-4 text-white" xmlns="http://www.w3.org/2000/svg" fill="none" viewBox="0 0 24 24">
+                                <circle class="opacity-25" cx="12" cy="12" r="10" stroke="currentColor" stroke-width="4"></circle>
+                                <path class="opacity-75" fill="currentColor" d="M4 12a8 8 0 018-8V0C5.373 0 0 5.373 0 12h4zm2 5.291A7.962 7.962 0 014 12H0c0 3.042 1.135 5.824 3 7.938l3-2.647z"></path>
+                            </svg>
+                        </div>
+                    </button>
+                </div>
+            </div>
+        </div>
+        <!-- Right Panel - Events and Results -->
+        <div class="w-1/2 p-4 flex flex-col min-h-0">
+            <div class="bg-white rounded-lg shadow-sm p-4 flex flex-col flex-1 overflow-hidden">
+                <div class="flex justify-between items-center mb-4">
+                    <h2 class="text-lg font-semibold text-gray-900">Events & Results</h2>
+                    <div class="flex items-center space-x-4">
+                        <label class="flex items-center space-x-2">
+                            <input type="checkbox" id="autoScroll" checked
+                                class="rounded text-primary-600 focus:ring-primary-500">
+                            <span class="text-sm text-gray-600">Auto-scroll</span>
+                        </label>
+                        <button id="clearEvents"
+                            class="inline-flex items-center px-3 py-1 text-sm text-gray-600 hover:text-gray-900 focus:outline-none">
+                            Clear
+                        </button>
+                    </div>
+                </div>
+                <div class="flex-1 flex flex-col min-h-0 overflow-hidden">
+                    <!-- Events Container -->
+                    <div id="eventsContainer" class="flex-1 overflow-y-auto space-y-2 min-h-0 mb-4">
+                        <!-- Events will be inserted here -->
+                    </div>
+                    <!-- Final Answer Section (Initially Hidden) -->
+                    <div id="finalAnswer" class="hidden border-t border-gray-200 pt-4 overflow-y-auto">
+                        <div class="bg-green-50 rounded-lg p-4">
+                            <div class="flex items-center justify-between mb-2">
+                                <h3 class="text-lg font-semibold text-green-800">Final Answer</h3>
+                                <button id="copyAnswer" class="inline-flex items-center px-3 py-1 text-sm font-medium rounded-md text-green-700 bg-green-100 hover:bg-green-200 focus:outline-none focus:ring-2 focus:ring-offset-2 focus:ring-green-500">
+                                    <svg class="h-4 w-4 mr-1" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+                                        <path stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="M8 16H6a2 2 0 01-2-2V6a2 2 0 012-2h8a2 2 0 012 2v2m-6 12h8a2 2 0 002-2v-8a2 2 0 00-2-2h-8a2 2 0 00-2 2v8a2 2 0 002 2z"/>
+                                    </svg>
+                                    <span>Copy</span>
+                                </button>
+                            </div>
+                            <div class="prose prose-sm max-w-none">
+                                <div class="answer-content"></div>
+                            </div>
+                        </div>
+                    </div>
+                </div>
+            </div>
+        </div>
+    </main>
+</body>
+</html>

quantalogic/tool_manager.py ADDED Viewed

@@ -0,0 +1,68 @@
+"""Tool dictionary for the agent."""
+from loguru import logger
+from pydantic import BaseModel
+from quantalogic.tools.tool import Tool
+class ToolManager(BaseModel):
+    """Tool dictionary for the agent."""
+    tools: dict[str, Tool] = {}
+    def tool_names(self) -> list[str]:
+        """Get the names of all tools in the tool dictionary."""
+        logger.debug("Getting tool names")
+        return list(self.tools.keys())
+    def add(self, tool: Tool):
+        """Add a tool to the tool dictionary."""
+        logger.info(f"Adding tool: {tool.name} to tool dictionary")
+        self.tools[tool.name] = tool
+    def add_list(self, tools: list[Tool]):
+        """Add a list of tools to the tool dictionary."""
+        logger.info(f"Adding {len(tools)} tools to tool dictionary")
+        for tool in tools:
+            self.add(tool)
+    def remove(self, tool_name: str) -> bool:
+        """Remove a tool from the tool dictionary."""
+        logger.info(f"Removing tool: {tool_name} from tool dictionary")
+        del self.tools[tool_name]
+        return True
+    def get(self, tool_name: str) -> Tool:
+        """Get a tool from the tool dictionary."""
+        logger.debug(f"Getting tool: {tool_name} from tool dictionary")
+        return self.tools[tool_name]
+    def list(self):
+        """List all tools in the tool dictionary."""
+        logger.debug("Listing all tools")
+        return list(self.tools.keys())
+    def execute(self, tool_name: str, **kwargs) -> str:
+        """Execute a tool from the tool dictionary."""
+        logger.info(f"Executing tool: {tool_name} with arguments: {kwargs}")
+        try:
+            result = self.tools[tool_name].execute(**kwargs)
+            logger.debug(f"Tool {tool_name} execution completed successfully")
+            return result
+        except Exception as e:
+            logger.error(f"Error executing tool {tool_name}: {str(e)}")
+            raise
+    def to_markdown(self):
+        """Create a comprehensive Markdown representation of the tool dictionary."""
+        logger.debug("Creating Markdown representation of tool dictionary")
+        markdown = ""
+        index: int = 1
+        for tool_name, tool in self.tools.items():
+            # use the tool's to_markdown method
+            markdown += f"### {index}. {tool_name}\n"
+            markdown += tool.to_markdown()
+            markdown += "\n"
+            index += 1
+        return markdown

quantalogic/tools/__init__.py ADDED Viewed

@@ -0,0 +1,46 @@
+"""Tools for the QuantaLogic agent."""
+from .agent_tool import AgentTool
+from .download_http_file_tool import DownloadHttpFileTool
+from .edit_whole_content_tool import EditWholeContentTool
+from .elixir_tool import ElixirTool
+from .execute_bash_command_tool import ExecuteBashCommandTool
+from .input_question_tool import InputQuestionTool
+from .list_directory_tool import ListDirectoryTool
+from .llm_tool import LLMTool
+from .markitdown_tool import MarkitdownTool
+from .nodejs_tool import NodeJsTool
+from .python_tool import PythonTool
+from .read_file_block_tool import ReadFileBlockTool
+from .read_file_tool import ReadFileTool
+from .replace_in_file_tool import ReplaceInFileTool
+from .ripgrep_tool import RipgrepTool
+from .search_definition_names import SearchDefinitionNames
+from .task_complete_tool import TaskCompleteTool
+from .tool import Tool, ToolArgument
+from .unified_diff_tool import UnifiedDiffTool
+from .write_file_tool import WriteFileTool
+__all__ = [
+    "Tool",
+    "ToolArgument",
+    "TaskCompleteTool",
+    "ReadFileTool",
+    "WriteFileTool",
+    "InputQuestionTool",
+    "ListDirectoryTool",
+    "LLMTool",
+    "ExecuteBashCommandTool",
+    "PythonTool",
+    "ElixirTool",
+    "NodeJsTool",
+    "UnifiedDiffTool",
+    "ReplaceInFileTool",
+    "AgentTool",
+    "ReadFileBlockTool",
+    "RipgrepTool",
+    "SearchDefinitionNames",
+    "MarkitdownTool",
+    "DownloadHttpFileTool",
+    "EditWholeContentTool",
+]

quantalogic/tools/agent_tool.py ADDED Viewed

@@ -0,0 +1,88 @@
+"""Agent Tool for delegating tasks to another agent."""
+import logging
+from typing import TYPE_CHECKING, Any
+from pydantic import Field, model_validator
+from quantalogic.tools.tool import Tool, ToolArgument
+# Use conditional import to resolve circular dependency
+if TYPE_CHECKING:
+    from quantalogic.agent import Agent
+class AgentTool(Tool):
+    """Tool to execute tasks using another agent."""
+    name: str = Field(default="agent_tool")
+    description: str = Field(
+        default=(
+            "Executes tasks using a specified agent. "
+            "This tool enables an agent to delegate tasks to another agent."
+            "A delegate agent doesn't have access to the memory and the conversation of the main agent."
+            "Context must be provided by the main agent."
+            "You must use variable interpolation syntax to use the context of the main agent."
+        )
+    )
+    agent_role: str = Field(..., description="The role of the agent (e.g., expert, assistant)")
+    agent: Any = Field(..., description="The agent to delegate tasks to")
+    # Type hint for static type checkers
+    if TYPE_CHECKING:
+        agent: "Agent"
+    # Allow extra fields
+    model_config = {"extra": "allow"}
+    # Tool Arguments
+    arguments: list[ToolArgument] = Field(
+        default_factory=lambda: [
+            ToolArgument(
+                name="task",
+                arg_type="string",
+                description="The task to delegate to the specified agent.",
+                required=True,
+                example="Summarize the latest news.",
+            ),
+        ]
+    )
+    @model_validator(mode="before")
+    def validate_agent(cls, values: dict[str, Any]) -> dict[str, Any]:
+        """Validate the provided agent and its role."""
+        agent = values.get("agent")
+        # Lazy import to avoid circular dependency
+        from quantalogic.agent import Agent
+        if not isinstance(agent, Agent):
+            raise ValueError("The agent must be an instance of the Agent class.")
+        return values
+    def execute(self, task: str) -> str:
+        """Execute the tool to delegate a task to the specified agent.
+        Args:
+            task (str): The task to delegate to the agent.
+        Returns:
+            str: The result of the delegated task.
+        """
+        try:
+            logging.info(f"Delegating task to agent with role '{self.agent_role}': {task}")
+            # Lazy import to avoid circular dependency
+            from quantalogic.agent import Agent
+            # Ensure the agent is of the correct type
+            if not isinstance(self.agent, Agent):
+                raise ValueError("The agent must be an instance of the Agent class.")
+            # Delegate the task to the agent
+            result = self.agent.solve_task(task)
+            logging.info(f"Task delegation completed for agent with role '{self.agent_role}'")
+            return result
+        except Exception as e:
+            logging.error(f"Error delegating task to agent: {str(e)}")
+            raise

quantalogic/tools/download_http_file_tool.py ADDED Viewed

@@ -0,0 +1,64 @@
+"""Tool for downloading a file from an HTTP URL and saving it to a local file."""
+from urllib.parse import urlparse
+from quantalogic.tools.tool import Tool, ToolArgument
+from quantalogic.utils.download_http_file import download_http_file
+class DownloadHttpFileTool(Tool):
+    """Tool for downloading a file from an HTTP URL and saving it to a local file."""
+    name: str = "download_http_file_tool"
+    description: str = "Downloads a file from an HTTP URL and saves it to a local file."
+    arguments: list = [
+        ToolArgument(
+            name="url",
+            arg_type="string",
+            description="The URL of the file to download.",
+            required=True,
+            example="https://example.com/data.txt",
+        ),
+        ToolArgument(
+            name="output_path",
+            arg_type="string",
+            description="The local path where the downloaded file will be saved.",
+            required=True,
+            example="/path/to/save/data.txt",
+        ),
+    ]
+    def _is_url(self, path: str) -> bool:
+        """Check if the given path is a valid URL."""
+        try:
+            result = urlparse(path)
+            return all([result.scheme, result.netloc])
+        except ValueError:
+            return False
+    def execute(self, url: str, output_path: str) -> str:
+        """Downloads a file from an HTTP URL and saves it to a local file.
+        Args:
+            url (str): The URL of the file to download.
+            output_path (str): The local path where the downloaded file will be saved.
+        Returns:
+            str: A message indicating the result of the download operation.
+        """
+        if not self._is_url(url):
+            return f"Error: {url} is not a valid URL."
+        try:
+            result = download_http_file(url, output_path)
+            if result:
+                return f"File downloaded successfully and saved to {output_path}."
+            else:
+                return f"Error downloading file from {url}: Unable to download."
+        except Exception as e:
+            return f"Error downloading file from {url}: {str(e)}"
+if __name__ == "__main__":
+    tool = DownloadHttpFileTool()
+    print(tool.to_markdown())

quantalogic/tools/edit_whole_content_tool.py ADDED Viewed

@@ -0,0 +1,70 @@
+"""Tool for editing the whole content of a file."""
+import os
+from quantalogic.tools.tool import Tool, ToolArgument
+class EditWholeContentTool(Tool):
+    """Tool for replace the whole content of a file."""
+    name: str = "edit_whole_content_tool"
+    description: str = "Edits the whole content of an existing file."
+    need_validation: bool = True
+    arguments: list = [
+        ToolArgument(
+            name="file_path",
+            arg_type="string",
+            description="The path to the file to edit. Using an absolute path is recommended.",
+            required=True,
+            example="/path/to/file.txt",
+        ),
+        ToolArgument(
+            name="content",
+            arg_type="string",
+            description="""
+            The content to write to the file. Use CDATA to escape special characters.
+            Don't add newlines at the beginning or end of the content.
+            """,
+            required=True,
+            example="Hello, world!",
+        ),
+    ]
+    def execute(self, file_path: str, content: str) -> str:
+        """Writes a file with the given content.
+        Args:
+            file_path (str): The path to the file to write.
+            content (str): The content to write to the file.
+        Returns:
+            str: The content of the file.
+        Raises:
+            FileExistsError: If the file already exists and append_mode is False and overwrite is False.
+        """
+        ## Handle tilde expansion
+        if file_path.startswith("~"):
+            # Expand the tilde to the user's home directory
+            file_path = os.path.expanduser(file_path)
+        # Convert relative paths to absolute paths using current working directory
+        if not os.path.isabs(file_path):
+            file_path = os.path.abspath(os.path.join(os.getcwd(), file_path))
+        # Ensure parent directory exists
+        os.makedirs(os.path.dirname(file_path), exist_ok=True)
+        if not os.path.exists(file_path):
+            raise FileExistsError(f"File {file_path} does not exist.")
+        with open(file_path, "w", encoding="utf-8") as f:
+            f.write(content)
+        file_size = os.path.getsize(file_path)
+        return f"File {file_path} written successfully. Size: {file_size} bytes."
+if __name__ == "__main__":
+    tool = EditWholeContentTool()
+    print(tool.to_markdown())