# -*- coding: utf-8 -*-
"""
智能体相关模块
"""
from pydantic_ai import ModelSettings
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.openai import OpenAIProvider

DEEPSEEK_V4_FLASH_MODEL = OpenAIChatModel(
    model_name="deepseek-v4-flash",
    provider=OpenAIProvider(
        base_url="https://tokenhub.tencentmaas.com/v1",
        api_key="sk-D9Y1mCe8VlvNqLuSC4mAjqEwxJ2nW4C0h8a7EPn8kg9RLsHq",
    ),
)

DEEPSEEK_V4_FLASH_MODEL_DISABLED_THINKING = ModelSettings(
    temperature=0, extra_body={"thinking": {"type": "disabled"}}  # 温度控制
)  # 不启用思考模式