mirror of
https://github.com/browser-use/browser-use
synced 2026-05-06 17:52:15 +02:00
53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import logging
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
import anyio
|
|
|
|
from browser_use.llm.messages import BaseMessage
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
async def save_conversation(
|
|
input_messages: list[BaseMessage],
|
|
response: Any,
|
|
target: str | Path,
|
|
encoding: str | None = None,
|
|
) -> None:
|
|
"""Save conversation history to file asynchronously."""
|
|
target_path = Path(target)
|
|
# create folders if not exists
|
|
if target_path.parent:
|
|
await anyio.Path(target_path.parent).mkdir(parents=True, exist_ok=True)
|
|
|
|
await anyio.Path(target_path).write_text(
|
|
await _format_conversation(input_messages, response),
|
|
encoding=encoding or 'utf-8',
|
|
)
|
|
|
|
|
|
async def _format_conversation(messages: list[BaseMessage], response: Any) -> str:
|
|
"""Format the conversation including messages and response."""
|
|
lines = []
|
|
|
|
# Format messages
|
|
for message in messages:
|
|
lines.append(f' {message.role} ')
|
|
|
|
lines.append(message.text)
|
|
lines.append('') # Empty line after each message
|
|
|
|
# Format response
|
|
lines.append(' RESPONSE')
|
|
lines.append(json.dumps(json.loads(response.model_dump_json(exclude_unset=True)), indent=2))
|
|
|
|
return '\n'.join(lines)
|
|
|
|
|
|
# Note: _write_messages_to_file and _write_response_to_file have been merged into _format_conversation
|
|
# This is more efficient for async operations and reduces file I/O
|