# -*- coding: utf-8 -*-
"""
任务模块
"""
from typing import Any, AsyncGenerator, Callable, Dict, Optional
from typing import AsyncGenerator, List

from pydantic import BaseModel
from pydantic_ai import Agent, ModelMessage
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.openai import OpenAIProvider

from application.domain_models import Dialog, TaskType
from models import DEEPSEEK_V4_FLASH_MODEL

instruction = """
# 角色
专业友好AI助手，结构化解答各类问题。

# 输出硬性规则
1. 全文强制标准Markdown，禁止纯文本；不要额外说明排版格式，直接输出内容；
2. 层级使用 `#/##/###`，列表用 `-` 无序列表或数字有序列表；
3. 代码块用 ```语言名``` 包裹；
4. 重点内容标注 **粗体**/*斜体*；
5. 思考、工具日志仅输出文本，适配前端折叠面板，禁止输出HTML标签；
6. 内容分点拆分，排版整洁适配前端Markdown渲染。

# 行文要求
语言通俗，逻辑完整简洁，无多余废话。
"""


async def run_stream_events(
    task_type: TaskType,
    user_prompt: str,
    message_history: List[ModelMessage],
) -> AsyncGenerator:
    """
    以流式事件模式运行
    """
    match task_type:
        case TaskType.CHAT:
            agent = Agent(
                model=DEEPSEEK_V4_FLASH_MODEL,
                instructions=instruction,
            )
            async with agent.run_stream_events(
                user_prompt=user_prompt,
                message_history=message_history,
            ) as events:
                async for event in events:
                    yield event

        case "flight":
            yield "未知任务类型"