")
+ print("Example: python openmanus_client_example.py ../mcp_server.py")
+ sys.exit(1)
+
+ client = OpenManusClient()
+ try:
+ await client.connect_to_server(server_script_path=sys.argv[1])
+
+ # Run examples first
+ await client.run_examples()
+
+ # Then start interactive chat loop
+ await client.chat_loop()
+
+ except Exception as e:
+ print(f"Error: {str(e)}")
+ finally:
+ await client.cleanup()
+
+
+if __name__ == "__main__":
+ asyncio.run(main())
diff --git a/openmanus_server/openmanus_server.py b/openmanus_server/openmanus_server.py
new file mode 100644
index 0000000..d73add4
--- /dev/null
+++ b/openmanus_server/openmanus_server.py
@@ -0,0 +1,196 @@
+import argparse
+import asyncio
+import json
+import logging
+import os
+import sys
+from typing import Optional
+
+from mcp.server.fastmcp import FastMCP
+
+
+# Add current directory to Python path
+current_dir = os.path.dirname(os.path.abspath(__file__))
+parent_dir = os.path.dirname(current_dir)
+sys.path.insert(0, parent_dir)
+sys.path.insert(0, current_dir)
+
+# Configure logging
+logging.basicConfig(
+ level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+)
+logger = logging.getLogger("mcp-server")
+
+# Import OpenManus tools
+from app.tool.browser_use_tool import BrowserUseTool
+from app.tool.file_saver import FileSaver
+from app.tool.google_search import GoogleSearch
+from app.tool.python_execute import PythonExecute
+from app.tool.terminate import Terminate
+
+
+# Initialize FastMCP server
+openmanus = FastMCP("openmanus")
+
+# Initialize tool instances
+browser_tool = BrowserUseTool()
+google_search_tool = GoogleSearch()
+python_execute_tool = PythonExecute()
+file_saver_tool = FileSaver()
+terminate_tool = Terminate()
+
+
+# Browser tool
+@openmanus.tool()
+async def browser_use(
+ action: str,
+ url: Optional[str] = None,
+ index: Optional[int] = None,
+ text: Optional[str] = None,
+ script: Optional[str] = None,
+ scroll_amount: Optional[int] = None,
+ tab_id: Optional[int] = None,
+) -> str:
+ """Execute various browser operations.
+
+ Args:
+ action: The browser operation to execute, possible values include:
+ - navigate: Navigate to specified URL
+ - click: Click on an element on the page
+ - input_text: Input text into a text field
+ - screenshot: Take a screenshot of the current page
+ - get_html: Get HTML of the current page
+ - get_text: Get text content of the current page
+ - execute_js: Execute JavaScript code
+ - scroll: Scroll the page
+ - switch_tab: Switch to specified tab
+ - new_tab: Open new tab
+ - close_tab: Close current tab
+ - refresh: Refresh current page
+ url: URL for 'navigate' or 'new_tab' operations
+ index: Element index for 'click' or 'input_text' operations
+ text: Text for 'input_text' operation
+ script: JavaScript code for 'execute_js' operation
+ scroll_amount: Scroll pixels for 'scroll' operation (positive for down, negative for up)
+ tab_id: Tab ID for 'switch_tab' operation
+ """
+ logger.info(f"Executing browser operation: {action}")
+ result = await browser_tool.execute(
+ action=action,
+ url=url,
+ index=index,
+ text=text,
+ script=script,
+ scroll_amount=scroll_amount,
+ tab_id=tab_id,
+ )
+ return json.dumps(result.model_dump())
+
+
+@openmanus.tool()
+async def get_browser_state() -> str:
+ """Get current browser state, including URL, title, tabs and interactive elements."""
+ logger.info("Getting browser state")
+ result = await browser_tool.get_current_state()
+ return json.dumps(result.model_dump())
+
+
+# Google search tool
+@openmanus.tool()
+async def google_search(query: str, num_results: int = 10) -> str:
+ """Execute Google search and return list of relevant links.
+
+ Args:
+ query: Search query
+ num_results: Number of results to return (default is 10)
+ """
+ logger.info(f"Executing Google search: {query}")
+ results = await google_search_tool.execute(query=query, num_results=num_results)
+ return json.dumps(results)
+
+
+# Python execution tool
+@openmanus.tool()
+async def python_execute(code: str, timeout: int = 5) -> str:
+ """Execute Python code and return results.
+
+ Args:
+ code: Python code to execute
+ timeout: Execution timeout in seconds
+ """
+ logger.info("Executing Python code")
+ result = await python_execute_tool.execute(code=code, timeout=timeout)
+ return json.dumps(result)
+
+
+# File saver tool
+@openmanus.tool()
+async def file_saver(content: str, file_path: str, mode: str = "w") -> str:
+ """Save content to local file.
+
+ Args:
+ content: Content to save
+ file_path: File path
+ mode: File open mode (default is 'w')
+ """
+ logger.info(f"Saving file: {file_path}")
+ result = await file_saver_tool.execute(
+ content=content, file_path=file_path, mode=mode
+ )
+ return result
+
+
+# Terminate tool
+@openmanus.tool()
+async def terminate(status: str) -> str:
+ """Terminate program execution.
+
+ Args:
+ status: Termination status, can be 'success' or 'failure'
+ """
+ logger.info(f"Terminating execution: {status}")
+ result = await terminate_tool.execute(status=status)
+ return result
+
+
+# Clean up resources
+async def cleanup():
+ """Clean up all tool resources"""
+ logger.info("Cleaning up resources")
+ await browser_tool.cleanup()
+
+
+# Register cleanup function
+import atexit
+
+
+atexit.register(lambda: asyncio.run(cleanup()))
+
+
+def parse_args():
+ """Parse command line arguments"""
+ parser = argparse.ArgumentParser(description="OpenManus MCP Server")
+ parser.add_argument(
+ "--transport",
+ choices=["stdio", "http"],
+ default="stdio",
+ help="Communication method: stdio or http (default: stdio)",
+ )
+ parser.add_argument(
+ "--host", default="127.0.0.1", help="HTTP server host (default: 127.0.0.1)"
+ )
+ parser.add_argument(
+ "--port", type=int, default=8000, help="HTTP server port (default: 8000)"
+ )
+ return parser.parse_args()
+
+
+if __name__ == "__main__":
+ args = parse_args()
+
+ if args.transport == "stdio":
+ logger.info("Starting OpenManus server (stdio mode)")
+ openmanus.run(transport="stdio")
+ else:
+ logger.info(f"Starting OpenManus server (HTTP mode) at {args.host}:{args.port}")
+ openmanus.run(transport="http", host=args.host, port=args.port)
diff --git a/pytest.ini b/pytest.ini
deleted file mode 100644
index 4bbeada..0000000
--- a/pytest.ini
+++ /dev/null
@@ -1,14 +0,0 @@
-[pytest]
-testpaths = tests
-python_files = test_*.py
-python_classes = Test*
-python_functions = test_*
-
-# Log settings
-log_cli = true
-log_cli_level = INFO
-log_cli_format = %(asctime)s [%(levelname)8s] %(message)s (%(filename)s:%(lineno)s)
-log_cli_date_format = %Y-%m-%d %H:%M:%S
-
-# Make sure asyncio works properly
-asyncio_mode = auto
diff --git a/run.bat b/run.bat
deleted file mode 100644
index 4c48f2f..0000000
--- a/run.bat
+++ /dev/null
@@ -1,3 +0,0 @@
-@echo off
-venv\Scripts\python.exe app.py
-pause
diff --git a/static/main.js b/static/main.js
deleted file mode 100644
index c219d20..0000000
--- a/static/main.js
+++ /dev/null
@@ -1,307 +0,0 @@
-let currentEventSource = null;
-
-function createTask() {
- const promptInput = document.getElementById('prompt-input');
- const prompt = promptInput.value.trim();
-
- if (!prompt) {
- alert("Please enter a valid prompt");
- promptInput.focus();
- return;
- }
-
- if (currentEventSource) {
- currentEventSource.close();
- currentEventSource = null;
- }
-
- const container = document.getElementById('task-container');
- container.innerHTML = 'Initializing task...
';
- document.getElementById('input-container').classList.add('bottom');
-
- fetch('/tasks', {
- method: 'POST',
- headers: {
- 'Content-Type': 'application/json'
- },
- body: JSON.stringify({ prompt })
- })
- .then(response => {
- if (!response.ok) {
- return response.json().then(err => { throw new Error(err.detail || 'Request failed') });
- }
- return response.json();
- })
- .then(data => {
- if (!data.task_id) {
- throw new Error('Invalid task ID');
- }
- setupSSE(data.task_id);
- loadHistory();
- })
- .catch(error => {
- container.innerHTML = `Error: ${error.message}
`;
- console.error('Failed to create task:', error);
- });
-}
-
-function setupSSE(taskId) {
- let retryCount = 0;
- const maxRetries = 3;
- const retryDelay = 2000;
-
- const container = document.getElementById('task-container');
-
- function connect() {
- const eventSource = new EventSource(`/tasks/${taskId}/events`);
- currentEventSource = eventSource;
-
- let heartbeatTimer = setInterval(() => {
- container.innerHTML += 'ยท
';
- }, 5000);
-
- const pollInterval = setInterval(() => {
- fetch(`/tasks/${taskId}`)
- .then(response => response.json())
- .then(task => {
- updateTaskStatus(task);
- })
- .catch(error => {
- console.error('Polling failed:', error);
- });
- }, 10000);
-
- const handleEvent = (event, type) => {
- clearInterval(heartbeatTimer);
- try {
- const data = JSON.parse(event.data);
- container.querySelector('.loading')?.remove();
- container.classList.add('active');
-
- const stepContainer = ensureStepContainer(container);
- const { formattedContent, timestamp } = formatStepContent(data, type);
- const step = createStepElement(type, formattedContent, timestamp);
-
- stepContainer.appendChild(step);
- autoScroll(stepContainer);
-
- fetch(`/tasks/${taskId}`)
- .then(response => response.json())
- .then(task => {
- updateTaskStatus(task);
- })
- .catch(error => {
- console.error('Status update failed:', error);
- });
- } catch (e) {
- console.error(`Error handling ${type} event:`, e);
- }
- };
-
- const eventTypes = ['think', 'tool', 'act', 'log', 'run', 'message'];
- eventTypes.forEach(type => {
- eventSource.addEventListener(type, (event) => handleEvent(event, type));
- });
-
- eventSource.addEventListener('complete', (event) => {
- clearInterval(heartbeatTimer);
- clearInterval(pollInterval);
- container.innerHTML += `
-
-
โ
Task completed
-
${lastResultContent}
-
- โ Error: ${data.message}
-
- `;
- eventSource.close();
- currentEventSource = null;
- } catch (e) {
- console.error('Error handling failed:', e);
- }
- });
-
- eventSource.onerror = (err) => {
- if (eventSource.readyState === EventSource.CLOSED) return;
-
- console.error('SSE connection error:', err);
- clearInterval(heartbeatTimer);
- clearInterval(pollInterval);
- eventSource.close();
-
- if (retryCount < maxRetries) {
- retryCount++;
- container.innerHTML += `
-
- โ Connection lost, retrying in ${retryDelay/1000} seconds (${retryCount}/${maxRetries})...
-
- `;
- setTimeout(connect, retryDelay);
- } else {
- container.innerHTML += `
-
- โ Connection lost, please try refreshing the page
-
- `;
- }
- };
- }
-
- connect();
-}
-
-function loadHistory() {
- fetch('/tasks')
- .then(response => {
- if (!response.ok) {
- return response.text().then(text => {
- throw new Error(`่ฏทๆฑๅคฑ่ดฅ: ${response.status} - ${text.substring(0, 100)}`);
- });
- }
- return response.json();
- })
- .then(tasks => {
- const listContainer = document.getElementById('task-list');
- listContainer.innerHTML = tasks.map(task => `
-
-
${task.prompt}
-
- ${new Date(task.created_at).toLocaleString()} -
-
- ${task.status || 'ๆช็ฅ็ถๆ'}
-
-
-
ๅ ่ฝฝๅคฑ่ดฅ: ${error.message}
`;
- });
-}
-
-
-function ensureStepContainer(container) {
- let stepContainer = container.querySelector('.step-container');
- if (!stepContainer) {
- container.innerHTML = '';
- stepContainer = container.querySelector('.step-container');
- }
- return stepContainer;
-}
-
-function formatStepContent(data, eventType) {
- return {
- formattedContent: data.result,
- timestamp: new Date().toLocaleTimeString()
- };
-}
-
-function createStepElement(type, content, timestamp) {
- const step = document.createElement('div');
- step.className = `step-item ${type}`;
- step.innerHTML = `
-
-
${getEventIcon(type)} [${timestamp}] ${getEventLabel(type)}:
-
${content}
-
-
-
-
-
-
-
Welcome to OpenManus Local Version
-
Please enter a task prompt to start a new task
-
-
-
-
-
-
-
-
-
-