# cloud.ai

const ai = cloud.ai()
const model = ai.createModel('cloudbase')

function createModel(model: string, options?: { defaultModelSubUrl?: string }): ReactModel

const ai = cloud.ai()
const model = ai.createModel('cloudbase')
const res = await model.generateText({
  model: 'hy3-preview',
  messages: [{ role: 'user', content: '你好' }],
})

console.log(res.text)
console.log(res.usage)

function generateText(input: ReactModelInput, options?: { timeout?: number }): Promise<{
  text: string
  messages: Array<ChatModelMessage>
  usage: Usage
  rawResponses: Array<unknown>
  error?: any
}>

interface BaseChatModelInput {
  model: string
  messages: Array<ChatModelMessage>
  temperature?: number
  top_p?: number
  tools?: Array<ModelTool>
  tool_choice?: 'none' | 'auto' | 'custom'
}

const ai = cloud.ai()
const model = ai.createModel('cloudbase')
const res = await model.streamText({
  model: 'hy3-preview',
  messages: [{ role: 'user', content: 'hi' }],
})

for await (const str of res.textStream) {
  console.log(str)
}

for await (const data of res.dataStream) {
  console.log(data)
}

const messages = await res.messages
const usage = await res.usage

function streamText(input: ReactModelInput, options?: { timeout?: number }): Promise<{
  dataStream: AsyncIterable<BaseDoStreamOutputChunk & { rawResponse?: any }>
  textStream: AsyncIterable<string>
  messages: Promise<Array<ChatModelMessage>>
  usage: Promise<Usage>
  error?: any
}>

const ai = cloud.ai()

const getWeatherTool = {
  name: 'get_weather',
  description: "返回某个城市某天的温度。调用示例：get_weather({city: '北京', date: '03-19'})",
  fn: ({ city, date }) => `${city}在${date}的温度是：${20 + Math.random() * 10}`,
  parameters: {
    type: 'object',
    properties: {
      city: { type: 'string', description: '要查询的城市' },
      date: { type: 'string', description: '要查询的日期' },
    },
    required: ['city', 'date'],
  },
}

ai.registerFunctionTool(getWeatherTool)

const model = ai.createModel('cloudbase')
const result = await model.generateText({
  model: 'hy3-preview',
  tools: [getWeatherTool],
  messages: [{ role: 'user', content: '请告诉我北京今天的天气' }],
})

console.log(result.text)

function registerFunctionTool(functionTool: FunctionTool): void

type FunctionTool = {
  name: string
  description: string
  fn: CallableFunction
  parameters: object
}

type ModelTool = {
  type: string
  function: {
    name: string
    description: string
    parameters: object
  }
}

const ai = cloud.ai()
const imageModel = ai.createImageModel('hunyuan-image')

function createImageModel<T extends 'hunyuan-image' | (string & {})>(provider: T): DefaultImageModel<T>

const cloud = require('wx-server-sdk')

cloud.init({
  env: cloud.DYNAMIC_CURRENT_ENV,
  timeout: 150000,
})

exports.main = async (event) => {
  const ai = cloud.ai()
  const imageModel = ai.createImageModel('hunyuan-image')
  const res = await imageModel.generateImage({
    model: 'HY-Image-3.0-Plus-4090-Tob-v1.0',
    prompt: '一只胖胖的橘猫坐在窗台上打盹，水彩风格，温暖色调',
    size: '1024x1024',
    seed: 42,
    footnote: 'Generated by Cloudbase',
    revise: { value: true },
    enable_thinking: { value: false },
  })
  console.log(res.data[0].url)
  console.log(res.data[0].revised_prompt)
  return { url: res.data[0].url }
}

const res = await imageModel.generateImage({
  model: 'HY-Image-v3.0-I2I-ToB-v1.0.1',
  prompt: '把猫咪改成水彩风格',
  image_urls: ['https://example.com/cat.jpg'],
  revise: { value: true },
})

console.log(res.data[0].url)
console.log(res.data[0].revised_prompt)

function generateImage(input: HunyuanARGenerateImageInput): Promise<HunyuanARGenerateImageOutput>

const https = require('https')

function downloadImage(url) {
  return new Promise((resolve, reject) => {
    https.get(url, (response) => {
      const chunks = []
      response.on('data', (chunk) => chunks.push(chunk))
      response.on('end', () => resolve(Buffer.concat(chunks)))
      response.on('error', reject)
    })
  })
}

exports.main = async (event) => {
  const imageModel = cloud.ai().createImageModel('hunyuan-image')
  const res = await imageModel.generateImage({
    model: 'HY-Image-3.0-Plus-4090-Tob-v1.0',
    prompt: event.prompt,
  })
  const imageBuffer = await downloadImage(res.data[0].url)
  const uploadRes = await cloud.uploadFile({
    cloudPath: `ai-images/${Date.now()}.jpg`,
    fileContent: imageBuffer,
  })
  return { fileID: uploadRes.fileID }
}

const ai = cloud.ai()
const res = await ai.bot.sendMessage({
  botId: 'botId-xxx',
  msg: '你好',
  history: [{ role: 'user', content: '你是李白。' }],
})

for await (const str of res.textStream) {
  console.log(str)
}

for await (const data of res.dataStream) {
  console.log(data)
}

function sendMessage(props: IBotSendMessage, options?: { timeout?: number }): Promise<StreamResult>

interface IBotSendMessage {
  botId: string
  msg: string
  history: Array<{
    role: string
    content: string
  }>
}

// dataStream 元素
type BotEventStreamData = {
  content: string
}

// eventSourceStream 元素
interface ParsedEvent {
  type: 'event'
  event?: string
  id?: string
  data: string
}

type UserMessage = { role: 'user'; content: string }
type SystemMessage = { role: 'system'; content: string }
type PlainAssistantMessage = { role: 'assistant'; content: string }
type ToolCallAssistantMessage = {
  role: 'assistant'
  tool_calls: Array<ToolCall>
  content?: string
}
type AssistantMessage = PlainAssistantMessage | ToolCallAssistantMessage
type ToolMessage = { role: 'tool'; tool_call_id: string; content: string }
type ChatModelMessage = UserMessage | SystemMessage | AssistantMessage | ToolMessage

type ToolCall = {
  id: string
  type: string
  function: {
    name: string
    arguments: string
  }
}

type Usage = {
  completion_tokens: number
  prompt_tokens: number
  total_tokens: number
}

interface IOnStepFinish {
  messages: Array<ChatModelMessage>
  text?: string
  toolCall?: ToolCall
  toolResult?: unknown
  finishReason?: string
  stepUsage?: Usage
  totalUsage?: Usage
}

const cloud = require('wx-server-sdk')

cloud.init({
  env: cloud.DYNAMIC_CURRENT_ENV,
  timeout: 60000,
})

exports.main = async (event) => {
  const ai = cloud.ai()
  const { type, messages, prompt } = event

  if (type === 'image') {
    const imageModel = ai.createImageModel('hunyuan-image')
    const res = await imageModel.generateImage({
      model: 'HY-Image-3.0-Plus-4090-Tob-v1.0',
      prompt: prompt || '一只可爱的猫咪在草地上玩耍',
    })
    return { url: res.data[0].url }
  }

  const model = ai.createModel('cloudbase')
  const result = await model.generateText({
    model: 'hy3-preview',
    messages: messages || [{ role: 'user', content: '你好' }],
  })
  return { text: result.text, usage: result.usage }
}