ai

README

低代码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)
	}
}

Documentation

Index

• func ExportForDev() (string, error)
• func Init()
• func InitFrom(filePath string)
• func Run(code string, refFile string, args ...any) (any, error)
• func RunFile(file string, args ...any) (any, error)
• func SetSSKey(key, iv []byte)
• type Exports
• type Runtime
  • func New() *Runtime
  • func (rt *Runtime) GetCallStack() []string
  • func (rt *Runtime) RunCode(code string) (any, error)
  • func (rt *Runtime) RunMain(args ...any) (any, error)
  • func (rt *Runtime) SetModuleLoader(fn func(filename string) string)
  • func (rt *Runtime) StartFromCode(code, refFile string) (any, error)
  • func (rt *Runtime) StartFromFile(file string) (any, error)
• type WatchRunner
  • func WatchRun(file string, extDirs, extTypes []string, args ...any) (*WatchRunner, error)
  • func (wr *WatchRunner) Stop()
  • func (wr *WatchRunner) WaitForKill()

Functions

func ExportForDev

func ExportForDev() (string, error)

func Init

func Init()

func InitFrom

func InitFrom(filePath string)

func Run

func Run(code string, refFile string, args ...any) (any, error)

func RunFile

func RunFile(file string, args ...any) (any, error)

func SetSSKey

func SetSSKey(key, iv []byte)

Types

type Exports

type Exports struct {
	LLMList []string
}

type Runtime

type Runtime struct {
	// contains filtered or unexported fields
}

func New

func New() *Runtime

func (*Runtime) GetCallStack

func (rt *Runtime) GetCallStack() []string

func (*Runtime) RunCode

func (rt *Runtime) RunCode(code string) (any, error)

func (*Runtime) RunMain

func (rt *Runtime) RunMain(args ...any) (any, error)

func (*Runtime) SetModuleLoader

func (rt *Runtime) SetModuleLoader(fn func(filename string) string)

func (*Runtime) StartFromCode

func (rt *Runtime) StartFromCode(code, refFile string) (any, error)

func (*Runtime) StartFromFile

func (rt *Runtime) StartFromFile(file string) (any, error)

type WatchRunner

type WatchRunner struct {
	// contains filtered or unexported fields
}

func WatchRun

func WatchRun(file string, extDirs, extTypes []string, args ...any) (*WatchRunner, error)

func (*WatchRunner) Stop

func (wr *WatchRunner) Stop()

func (*WatchRunner) WaitForKill

func (wr *WatchRunner) WaitForKill()