99 lines
2.1 KiB
Go
99 lines
2.1 KiB
Go
package agent
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/mudler/local-agent-framework/llm"
|
|
)
|
|
|
|
type Character struct {
|
|
Name string `json:"name"`
|
|
Age int `json:"age"`
|
|
Occupation string `json:"job_occupation"`
|
|
NowDoing string `json:"doing_now"`
|
|
DoingNext string `json:"doing_next"`
|
|
DoneHistory []string `json:"done_history"`
|
|
Memories []string `json:"memories"`
|
|
Hobbies []string `json:"hobbies"`
|
|
MusicTaste []string `json:"music_taste"`
|
|
Sex string `json:"sex"`
|
|
}
|
|
|
|
func Load(path string) (*Character, error) {
|
|
data, err := os.ReadFile(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var c Character
|
|
err = json.Unmarshal(data, &c)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &c, nil
|
|
}
|
|
|
|
func (a *Agent) Save(path string) error {
|
|
data, err := json.Marshal(a.options.character)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return os.WriteFile(path, data, 0644)
|
|
}
|
|
|
|
func (a *Agent) generateIdentity(guidance string) error {
|
|
if guidance == "" {
|
|
guidance = "Generate a random character for roleplaying."
|
|
}
|
|
err := llm.GenerateJSONFromStruct(a.client, guidance, a.options.LLMAPI.Model, &a.options.character)
|
|
a.Character = a.options.character
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !a.validCharacter() {
|
|
return fmt.Errorf("invalid character")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (a *Agent) validCharacter() bool {
|
|
return a.Character.Name != "" &&
|
|
a.Character.Age != 0 &&
|
|
a.Character.Occupation != "" &&
|
|
a.Character.NowDoing != "" &&
|
|
a.Character.DoingNext != "" &&
|
|
len(a.Character.DoneHistory) != 0 &&
|
|
len(a.Character.Memories) != 0 &&
|
|
len(a.Character.Hobbies) != 0 &&
|
|
len(a.Character.MusicTaste) != 0
|
|
}
|
|
|
|
const fmtT = `=====================
|
|
Name: %s
|
|
Age: %d
|
|
Occupation: %s
|
|
Now doing: %s
|
|
Doing next: %s
|
|
Done history: %v
|
|
Memories: %v
|
|
Hobbies: %v
|
|
Music taste: %v
|
|
=====================`
|
|
|
|
func (a *Agent) String() string {
|
|
return fmt.Sprintf(
|
|
fmtT,
|
|
a.Character.Name,
|
|
a.Character.Age,
|
|
a.Character.Occupation,
|
|
a.Character.NowDoing,
|
|
a.Character.DoingNext,
|
|
a.Character.DoneHistory,
|
|
a.Character.Memories,
|
|
a.Character.Hobbies,
|
|
a.Character.MusicTaste,
|
|
)
|
|
}
|