115 lines
2.4 KiB
V
115 lines
2.4 KiB
V
module openai
|
|
|
|
import freeflowuniverse.herolib.core.base
|
|
import freeflowuniverse.herolib.core.playbook
|
|
import freeflowuniverse.herolib.ui.console
|
|
|
|
__global (
|
|
openai_global map[string]&OpenAI
|
|
openai_default string
|
|
)
|
|
|
|
/////////FACTORY
|
|
|
|
@[params]
|
|
pub struct ArgsGet {
|
|
pub mut:
|
|
name string
|
|
}
|
|
|
|
fn args_get(args_ ArgsGet) ArgsGet {
|
|
mut args := args_
|
|
if args.name == '' {
|
|
args.name = 'default'
|
|
}
|
|
return args
|
|
}
|
|
|
|
pub fn get(args_ ArgsGet) !&OpenAI {
|
|
mut context := base.context()!
|
|
mut args := args_get(args_)
|
|
mut obj := OpenAI{
|
|
name: args.name
|
|
}
|
|
if args.name !in openai_global {
|
|
if !exists(args)! {
|
|
set(obj)!
|
|
} else {
|
|
heroscript := context.hero_config_get('openai', args.name)!
|
|
mut obj_ := heroscript_loads(heroscript)!
|
|
set_in_mem(obj_)!
|
|
}
|
|
}
|
|
return openai_global[args.name] or {
|
|
println(openai_global)
|
|
// bug if we get here because should be in globals
|
|
panic('could not get config for openai with name, is bug, was for: ${args.name}')
|
|
}
|
|
}
|
|
|
|
// register the config for the future
|
|
pub fn set(o OpenAI) ! {
|
|
set_in_mem(o)!
|
|
mut context := base.context()!
|
|
heroscript := heroscript_dumps(o)!
|
|
context.hero_config_set('openai', o.name, heroscript)!
|
|
}
|
|
|
|
// does the config exists?
|
|
pub fn exists(args_ ArgsGet) !bool {
|
|
mut context := base.context()!
|
|
mut args := args_get(args_)
|
|
return context.hero_config_exists('openai', args.name)
|
|
}
|
|
|
|
pub fn delete(args_ ArgsGet) ! {
|
|
mut args := args_get(args_)
|
|
mut context := base.context()!
|
|
context.hero_config_delete('openai', args.name)!
|
|
if args.name in openai_global {
|
|
// del openai_global[args.name]
|
|
}
|
|
}
|
|
|
|
// only sets in mem, does not set as config
|
|
fn set_in_mem(o OpenAI) ! {
|
|
mut o2 := obj_init(o)!
|
|
openai_global[o.name] = &o2
|
|
openai_default = o.name
|
|
}
|
|
|
|
@[params]
|
|
pub struct PlayArgs {
|
|
pub mut:
|
|
heroscript string // if filled in then plbook will be made out of it
|
|
plbook ?playbook.PlayBook
|
|
reset bool
|
|
}
|
|
|
|
pub fn play(args_ PlayArgs) ! {
|
|
mut args := args_
|
|
|
|
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 {
|
|
heroscript := install_action.heroscript()
|
|
mut obj2 := heroscript_loads(heroscript)!
|
|
set(obj2)!
|
|
}
|
|
}
|
|
}
|
|
|
|
// switch instance to be used for openai
|
|
pub fn switch(name string) {
|
|
openai_default = name
|
|
}
|
|
|
|
// helpers
|
|
|
|
@[params]
|
|
pub struct DefaultConfigArgs {
|
|
instance string = 'default'
|
|
}
|