29 lines
941 B
Python
29 lines
941 B
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
智能体模型模块
|
|
"""
|
|
from pydantic_ai import ModelSettings
|
|
from pydantic_ai.models.openai import OpenAIChatModel, OpenAIResponsesModel
|
|
from pydantic_ai.providers.openai import OpenAIProvider
|
|
|
|
DEEPSEEK_V4_FLASH_CHAT_MODEL = OpenAIChatModel(
|
|
model_name="deepseek-v4-flash",
|
|
provider=OpenAIProvider(
|
|
base_url="https://tokenhub.tencentmaas.com/v1",
|
|
api_key="sk-D9Y1mCe8VlvNqLuSC4mAjqEwxJ2nW4C0h8a7EPn8kg9RLsHq",
|
|
),
|
|
)
|
|
|
|
DEEPSEEK_V4_FLASH_CHAT_MODEL_SETTINGS = ModelSettings(
|
|
temperature=0, extra_body={"thinking": {"type": "disabled"}} # 温度控制
|
|
) # 不启用思考模式
|
|
|
|
DEEPSEEK_V4_FLASH_RESPONSE_MODEL = OpenAIResponsesModel(
|
|
model_name="deepseek/deepseek-flash",
|
|
provider=OpenAIProvider(
|
|
base_url="https://tokenhub.tencentmaas.com/v1",
|
|
api_key="sk-D9Y1mCe8VlvNqLuSC4mAjqEwxJ2nW4C0h8a7EPn8kg9RLsHq",
|
|
),
|
|
)
|
|
|
|
"sk-4a79ec5c0af74680997ef56758604850" |