低代码AI能力工具包

命令行工具

Install

go install apigo.cc/ai/ai/ai@latest

Usage

ai -h | --help		show usage
ai -e | --export	export ai.ts file to ./lib for develop
ai test | test.js	run test.js, if not specified, run ./ai.js

Sample

test.js

import {glm} from './lib/ai'
import console from './lib/console'

function main(...args) {
    if(!args[0]) throw new Error('no ask')
    let r = glm.fastAsk(args[0], r => {
        console.print(r)
    })
    console.println()
    return r
}

run sample

ai test "你好"

Module

mod/glm.js

import {glm} from './lib/ai'
import console from './lib/console'

function chat(...args) {
    if(!args[0]) throw new Error('no prompt')
    return glm.fastAsk(args[0])
}

function draw(...args) {
    if(!args[0]) throw new Error('no ask')
    return glm.makeImage(args[0], {size:'1024x1024'})
}

module.exports = {chat, draw}

test.js

import glm from './mod/glm'

function main(...args) {
    console.println(glm.chat(args[0]))
}

run sample

ai test "你好"

Configure

llm.yml

openai:
  apiKey: ...
zhipu:
  apiKey: ...

or use env.yml

llm.yml

llm:
  openai:
    apiKey: ...
  zhipu:
    apiKey: ...

encrypt apiKey

install sskey

go install github.com/ssgo/tool/sskey@latest
sskey -e 'your apiKey'

copy url base64 format encrypted apiKey into llm.yml or env.yml

config with special endpoint

llm:
  openai:
    apiKey: ...
    endpoint: https://api.openai.com/v1

config multi api

llm:
  glm:
    apiKey: ...
    llm: zhipu
  glm2:
    apiKey: ...
    endpoint: https://......
    llm: zhipu

调用 JavaScript API

Install

go get apigo.cc/ai/ai

Usage

package main

import (
	"apigo.cc/ai/ai/js"
	"fmt"
)

func main() {
    result, err := js.Run(`return ai.glm.fastAsk(args[0])`, "", "你好")
    // js.RunFile
    // js.StartFromFile
    // js.StartFromCode
    if err != nil {
        fmt.Println(err.Error())
    } else if result != nil {
        fmt.Println(result)
    }
}

调用 Go API

Install

go get apigo.cc/ai/ai

Usage

package main

import (
	"apigo.cc/ai/ai"
	"apigo.cc/ai/ai/llm"
	"fmt"
)

func main() {
	ai.Init()
	glm := llm.Get("zhipu")

	r, usage, err := glm.FastAsk(llm.Messages().User().Text("你是什么模型").Make(), func(text string) {
		fmt.Print(text)
	})

	if err != nil {
		fmt.Println(err)
	} else {
		fmt.Println()
		fmt.Println("result:", r)
		fmt.Println("usage:", usage)
	}
}