chat-ai-api/app/utils/chat_utils.py

from typing import List, Optional, Dict
from ..schemas.chat import ChatMessage
from ..core.ark_client import config
from ..db.mongo import get_last_response_id


def convert_messages_for_api(messages: List[ChatMessage]) -> List[Dict[str, str]]:
    return [{"role": msg.role, "content": msg.content} for msg in messages]


def get_latest_user_message(messages: List[ChatMessage]) -> Optional[ChatMessage]:
    for message in reversed(messages):
        if message.role == "user":
            return message
    return None


def get_previous_response_id(user_id: str, session_id: str) -> Optional[str]:
    return get_last_response_id(user_id, session_id)

# 联网搜索工具
def get_web_search_tools() -> list:
    return  [{
        "type": "web_search",
        "max_keyword": 20,
        "limit": 20,
        "sources": ["douyin", "moji", "toutiao"],# 附加搜索来源（抖音百科、墨迹天气、头条图文等平台）
        "user_location": {  # 用户地理位置（用于优化搜索结果）
            "type": "approximate",  # 大致位置
            "country": "中国",
            "region": "浙江",
            "city": "杭州"
        }
    }]
# 私域知识库
def get_knowledge_search_tools() -> list:
    return [{
            "type": "knowledge_search",
            "knowledge_resource_id": "kb-916f7f4baa369d97",  # 替换为实际知识库ID
            "limit": 10,  # 最多返回10条搜索结果
        }]

# 豆包助手工具
def get_doubao_tools() -> list:
    return [{
        "type": "doubao_app",
        "feature": {
            # 联网搜索功能
            "ai_search": {
                "type": "disabled",
                "role_description": config.ROLE_DESCRIPTION
            },
            # 边想边搜功能
            "reasoning_search":{
                "type": "enabled",
                "role_description": config.ROLE_DESCRIPTION
            }
        },
        "user_location": {
            "type": "approximate",
            "country": "中国",
            "region": "浙江",
            "city": "杭州"
        }
    }]