from runtime import Args
from typings.deepseek_R1_siliconflow.deepseek_R1_siliconflow import Input, Output
import http.client
import json

def handler(args: Args[Input]) -> Output:
    # 从输入参数中获取用户输入的内容和 API 密钥
    user_input = args.input.user_input
    api_key = args.input.api_key  # 假设用户输入的 api_key 通过此字段传入

    if not api_key:
        return {"message": "API Key is missing. Please provide a valid API Key."}

    # API 请求的 URL 和路径
    conn = http.client.HTTPSConnection("api.siliconflow.cn")
    payload = json.dumps({
        "model": "deepseek-ai/DeepSeek-R1",  # 使用 DeepSeek-R1 模型
        "messages": [            {
                "role": "user",
                "content": user_input
            }
        ],
        "stream": False,
        "max_tokens": 512,
        "stop": ["null"],
        "temperature": 0.7,
        "top_p": 0.7,
        "top_k": 50,
        "frequency_penalty": 0.5,
        "n": 1,
        "response_format": {"type": "text"},
        
    })

    headers = {
        "Authorization": f"Bearer {api_key}",  # 使用用户提供的 API Key
        "Content-Type": "application/json"
    }

    # 发送 POST 请求
    conn.request("POST", "/v1/chat/completions", body=payload, headers=headers)
    response = conn.getresponse()

    # 检查请求是否成功
    if response.status == 200:
        # 解析响应数据
        response_data = json.loads(response.read().decode("utf-8"))
        
        # 提取 content 和 reasoning_content
        content = response_data["choices"][0]["message"].get("content", "")
        reasoning_content = response_data["choices"][0]["message"].get("reasoning_content", "")

        # 返回 API 的响应内容，包括 reasoning_content
        return {          
            "reasoning_content": reasoning_content,
            "message": content
        }
    else:
        # 如果请求失败，返回错误信息
        return {"message": f"API request failed with status code {response.status}"}

from runtime import Args
from typings.DeepSeek_R1.DeepSeek_R1 import Input, Output
import http.client
import json

def handler(args: Args[Input]) -> Output:
    # 从输入参数中获取用户输入的内容和 API 密钥
    user_input = args.input.user_input
    api_key = args.input.api_key

    if not api_key:
        return {"message": "API Key is missing. Please provide a valid API Key."}

    # API 请求的 URL 和路径
    conn = http.client.HTTPSConnection("api.ppinfra.com")
    payload = json.dumps({
        "model": "deepseek/deepseek-r1/community",
        "messages": [
            {
                "role": "system",
                "content": "你是DeepSeek，你会以诚实专业的态度帮助用户，用中文回答问题。"
            },
            {
                "role": "user",
                "content": user_input
            }
        ],
        "response_format": {"type": "text"},
        "max_tokens": 8192,
        "temperature": 1,
        "top_p": 1,
        "min_p": 0,
        "top_k": 50,
        "presence_penalty": 0,
        "frequency_penalty": 0,
        "repetition_penalty": 1
    })

    headers = {
        "Authorization": f"Bearer {api_key}",
        "Content-Type": "application/json"
    }

    # 发送 POST 请求
    conn.request("POST", "/v3/openai/chat/completions", body=payload, headers=headers)
    response = conn.getresponse()

    # 检查请求是否成功
    if response.status == 200:
        # 解析响应数据
        response_data = json.loads(response.read().decode("utf-8"))
        
        # 获取完整的响应内容
        full_content = response_data["choices"][0]["message"].get("content", "")
        
        # 提取思维链内容和最终回答
        reasoning_content = ""
        message = full_content
        
        # 如果内容中包含<think>标签
        if "<think>" in full_content and "</think>" in full_content:
            think_start = full_content.find("<think>") + len("<think>")
            think_end = full_content.find("</think>")
            
            # 提取思维链内容
            reasoning_content = full_content[think_start:think_end].strip()
            
            # 提取最终回答（移除思维链部分）
            message = full_content[think_end + len("</think>"):].strip()

        return {          
            "reasoning_content": reasoning_content,
            "message": message
        }
    else:
        # 如果请求失败，返回错误信息
        return {"message": f"API request failed with status code {response.status}"}

{
  "user_input": "9.11和9.9谁大",
  "api_key": "这里填写你的api"
}