# -*- coding: utf-8 -*-
"""
生成产品需求文档智能体
"""
from pydantic_ai import Agent
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.openai import OpenAIProvider

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

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

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

# 实例化智能体
chat_agent: Agent = Agent(
    name="chat_agent",
    model=OpenAIChatModel(
        model_name="deepseek-v4-flash",
        provider=OpenAIProvider(
            base_url="https://tokenhub.tencentmaas.com/v1",
            api_key="sk-D9Y1mCe8VlvNqLuSC4mAjqEwxJ2nW4C0h8a7EPn8kg9RLsHq",
        ),
    ),
    instructions=instructions,
    capabilities=None,
    output_type=str,
    retries=1,
)