实现 API

import json
import time

import requests

API_KEY = 'your_api_key_here'
BASE_URL = 'https://api.browser-use.com/api/v1'
HEADERS = {'Authorization': f'Bearer {API_KEY}'}


def create_task(instructions: str):
	"""创建新的浏览器自动化任务"""
	response = requests.post(f'{BASE_URL}/run-task', headers=HEADERS, json={'task': instructions})
	return response.json()['id']


def get_task_status(task_id: str):
	"""获取当前任务状态"""
	response = requests.get(f'{BASE_URL}/task/{task_id}/status', headers=HEADERS)
	return response.json()


def get_task_details(task_id: str):
	"""获取完整的任务详情，包括输出"""
	response = requests.get(f'{BASE_URL}/task/{task_id}', headers=HEADERS)
	return response.json()


def wait_for_completion(task_id: str, poll_interval: int = 2):
	"""轮询任务状态直到完成"""
	count = 0
	unique_steps = []
	while True:
		details = get_task_details(task_id)
		new_steps = details['steps']
		# 仅使用不在 unique_steps 中的新步骤
		if new_steps != unique_steps:
			for step in new_steps:
				if step not in unique_steps:
					print(json.dumps(step, indent=4))
			unique_steps = new_steps
		count += 1
		status = details['status']

		if status in ['finished', 'failed', 'stopped']:
			return details
		time.sleep(poll_interval)


def main():
	task_id = create_task('打开 https://www.google.com 并搜索 openai')
	print(f'创建的任务 ID: {task_id}')
	task_details = wait_for_completion(task_id)
	print(f"最终输出: {task_details['output']}")


if __name__ == '__main__':
	main()

def control_task():
	# 创建新任务
	task_id = create_task("访问 google.com 并搜索 Browser Use")

	# 等待 5 秒
	time.sleep(5)

	# 暂停任务
	requests.put(f"{BASE_URL}/pause-task?task_id={task_id}", headers=HEADERS)
	print("任务已暂停! 查看实时预览。")

	# 等待用户输入
	input("按回车键继续...")

	# 恢复任务
	requests.put(f"{BASE_URL}/resume-task?task_id={task_id}", headers=HEADERS)

	# 等待完成
	result = wait_for_completion(task_id)
	print(f"任务完成，输出: {result['output']}")