core/engine/src/agents/writer.py - main - alexefan136/flowstack

core/engine/src/agents/writer.py
67 строк3 KB
Alexander Efanov
Обновление репозитория
15 июл 2026, 12:19
15 июл 2026, 12:1976704c6
# src/agents/writer.py
"""Агент-писатель для создания контента и текстов.

Этот модуль предоставляет WriterAgent — специализированного агента
для написания статей, эссе, технической документации и другого
текстового контента с фокусом на ясность, вовлечённость и стиль.

Примеры использования:

    # Писатель с дефолтными параметрами
    agent = WriterAgent()
    result = await agent.run({"topic": "Искусственный интеллект в 2025"})

    # Писатель с кастомным именем
    agent = WriterAgent(name="Technical Writer")

    # Писатель с кастомным system prompt
    agent = WriterAgent(
        name="Creative Writer",
        system_prompt="Ты — креативный писатель, специализирующийся на..."
    )
"""
from __future__ import annotations

from .base import LLMAgent


class WriterAgent(LLMAgent):
    """Агент для создания текстового контента.
    
    Специализируется на:
    - Написании статей, эссе и блогов
    - Создании технической документации
    - Редактировании и улучшении текстов
    - Адаптации стиля под целевую аудиторию
    
    Отличается от других агентов фокусом на качество письма,
    структуру текста и вовлечённость читателя.
    """

    DEFAULT_SYSTEM_PROMPT = """You are a professional writer. Your task is to:
1. Take the research provided
2. Write clear, engaging content
3. Use proper grammar and style
4. Format with markdown for readability

Make the content accessible to a general audience."""

    def __init__(
        self,
        name: str = "Writer",
        system_prompt: str | None = None,
    ):
        """Инициализация агента-писателя.
        
        Args:
            name: Имя агента (по умолчанию "Writer")
            system_prompt: Кастомный system prompt.
                Если None, используется DEFAULT_SYSTEM_PROMPT.
        """
        super().__init__(
            name=name,
            system_prompt=system_prompt or self.DEFAULT_SYSTEM_PROMPT,
        )


__all__ = ["WriterAgent"]