functions.py

import json

FUNCTIONS = [
    {
        "name": "search_duckduckgo",
        "description": "使用DuckDuckGo搜索引擎查询信息。可以搜索最新新闻、文章、博客等内容。",
        "parameters": {
            "type": "object",
            "properties": {
                "keywords": {
                    "type": "array",
                    "items": {"type": "string"},
                    "description": "搜索的关键词列表。例如：['Python', '机器学习', '最新进展']。"
                }
            },
            "required": ["keywords"]
        }
    },
    {
        "name": "search_papers",
        "description": "使用Crossref API搜索学术论文。",
        "parameters": {
            "type": "object",
            "properties": {
                "query": {
                    "type": "string",
                    "description": "搜索查询字符串。例如：'climate change'。"
                }
            },
            "required": ["query"]
        }
    },
    {
        "name": "send_email",
        "description": "发送电子邮件。",
        "parameters": {
            "type": "object",
            "properties": {
                "to": {
                    "type": "string",
                    "description": "收件人邮箱地址"
                },
                "subject": {
                    "type": "string",
                    "description": "邮件主题"
                },
                "content": {
                    "type": "string",
                    "description": "邮件内容"
                }
            },
            "required": ["to", "subject", "content"]
        }
    }
]

FUNCTIONS_GROUP_1 = [FUNCTIONS[0], FUNCTIONS[1]]  # search_duckduckgo, search_papers
FUNCTIONS_GROUP_2 = [FUNCTIONS[2]]  # send_email

def get_function_descriptions(functions):
    return [{"name": f["name"], "description": f["description"]} for f in functions]