20 lines
615 B
Python
20 lines
615 B
Python
# -*- 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"}} # 温度控制
|
|
) # 不启用思考模式
|