zhuchaowe/qwen_agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add DEFAULT_THINKING_ENABLE

Browse Source
This commit is contained in:
朱潮 2025-12-17 12:33:58 +08:00
parent 52b34846a4
commit c9a91f8f4d
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
utils/api_models.py
View File

@ -5,7 +5,7 @@ API data models and response schemas.
from typing import Dict, List, Optional, Any, AsyncGenerator
from pydantic import BaseModel, Field, field_validator, ConfigDict
from utils.settings import DEFAULT_THINKING_ENABLE
class Message(BaseModel):
role: str
@ -53,7 +53,7 @@ class ChatRequest(BaseModel):
robot_type: Optional[str] = "general_agent"
user_identifier: Optional[str] = ""
session_id: Optional[str] = None
enable_thinking: Optional[bool] = True
enable_thinking: Optional[bool] = DEFAULT_THINKING_ENABLE
class ChatRequestV2(BaseModel):
@ -64,7 +64,7 @@ class ChatRequestV2(BaseModel):
language: Optional[str] = "zh"
user_identifier: Optional[str] = ""
session_id: Optional[str] = None
enable_thinking: Optional[bool] = True
enable_thinking: Optional[bool] = DEFAULT_THINKING_ENABLE
class FileProcessRequest(BaseModel):

3
utils/settings.py
View File

@ -37,3 +37,6 @@ SENTENCE_TRANSFORMER_MODEL = os.getenv("SENTENCE_TRANSFORMER_MODEL", "TaylorAI/g
# Tool Output Length Control Settings
TOOL_OUTPUT_MAX_LENGTH = int(SUMMARIZATION_MAX_TOKENS/4)
TOOL_OUTPUT_TRUNCATION_STRATEGY = os.getenv("TOOL_OUTPUT_TRUNCATION_STRATEGY", "smart")
# THINKING ENABLE
DEFAULT_THINKING_ENABLE = os.getenv("DEFAULT_THINKING_ENABLE", "true") == "true"