106 lines
2.1 KiB
V
106 lines
2.1 KiB
V
module openai
|
|
|
|
import freeflowuniverse.herolib.core.base
|
|
import freeflowuniverse.herolib.core.playbook
|
|
|
|
__global (
|
|
openai_global map[string]&OpenAI
|
|
openai_default string
|
|
)
|
|
|
|
/////////FACTORY
|
|
|
|
@[params]
|
|
pub struct ArgsGet {
|
|
pub mut:
|
|
name string = 'default'
|
|
}
|
|
|
|
fn args_get(args_ ArgsGet) ArgsGet {
|
|
mut args := args_
|
|
if args.name == '' {
|
|
args.name = openai_default
|
|
}
|
|
if args.name == '' {
|
|
args.name = 'default'
|
|
}
|
|
return args
|
|
}
|
|
|
|
pub fn get(args_ ArgsGet) !&OpenAI {
|
|
mut args := args_get(args_)
|
|
if args.name !in openai_global {
|
|
if !config_exists() {
|
|
if default {
|
|
config_save()!
|
|
}
|
|
}
|
|
config_load()!
|
|
}
|
|
return openai_global[args.name] or {
|
|
println(openai_global)
|
|
panic('bug in get from factory: ')
|
|
}
|
|
}
|
|
|
|
fn config_exists(args_ ArgsGet) bool {
|
|
mut args := args_get(args_)
|
|
mut context := base.context() or { panic('bug') }
|
|
return context.hero_config_exists('openai', args.name)
|
|
}
|
|
|
|
fn config_load(args_ ArgsGet) ! {
|
|
mut args := args_get(args_)
|
|
mut context := base.context()!
|
|
mut heroscript := context.hero_config_get('openai', args.name)!
|
|
play(heroscript: heroscript)!
|
|
}
|
|
|
|
fn config_save(args_ ArgsGet) ! {
|
|
mut args := args_get(args_)
|
|
mut context := base.context()!
|
|
context.hero_config_set('openai', args.name, heroscript_default()!)!
|
|
}
|
|
|
|
fn set(o OpenAI) ! {
|
|
mut o2 := obj_init(o)!
|
|
openai_global['default'] = &o2
|
|
}
|
|
|
|
@[params]
|
|
pub struct PlayArgs {
|
|
pub mut:
|
|
name string = 'default'
|
|
heroscript string // if filled in then plbook will be made out of it
|
|
plbook ?playbook.PlayBook
|
|
reset bool
|
|
|
|
start bool
|
|
stop bool
|
|
restart bool
|
|
delete bool
|
|
configure bool // make sure there is at least one installed
|
|
}
|
|
|
|
pub fn play(args_ PlayArgs) ! {
|
|
mut args := args_
|
|
|
|
if args.heroscript == '' {
|
|
args.heroscript = heroscript_default()!
|
|
}
|
|
mut plbook := args.plbook or { playbook.new(text: args.heroscript)! }
|
|
|
|
mut install_actions := plbook.find(filter: 'openai.configure')!
|
|
if install_actions.len > 0 {
|
|
for install_action in install_actions {
|
|
mut p := install_action.params
|
|
cfg_play(p)!
|
|
}
|
|
}
|
|
}
|
|
|
|
// switch instance to be used for openai
|
|
pub fn switch(name string) {
|
|
openai_default = name
|
|
}
|