mirror of
https://github.com/danielmiessler/Fabric.git
synced 2026-02-13 23:45:08 -05:00
99 lines
2.3 KiB
Go
99 lines
2.3 KiB
Go
package fsdb
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/danielmiessler/fabric/common"
|
|
goopenai "github.com/sashabaranov/go-openai"
|
|
)
|
|
|
|
type SessionsEntity struct {
|
|
*StorageEntity
|
|
}
|
|
|
|
func (o *SessionsEntity) Get(name string) (session *Session, err error) {
|
|
session = &Session{Name: name}
|
|
|
|
if o.Exists(name) {
|
|
err = o.LoadAsJson(name, &session.Messages)
|
|
} else {
|
|
fmt.Printf("Creating new session: %s\n", name)
|
|
}
|
|
return
|
|
}
|
|
|
|
func (o *SessionsEntity) PrintSession(name string) (err error) {
|
|
if o.Exists(name) {
|
|
var session Session
|
|
if err = o.LoadAsJson(name, &session.Messages); err == nil {
|
|
fmt.Println(session.String())
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func (o *SessionsEntity) SaveSession(session *Session) (err error) {
|
|
return o.SaveAsJson(session.Name, session.Messages)
|
|
}
|
|
|
|
type Session struct {
|
|
Name string
|
|
Messages []*goopenai.ChatCompletionMessage
|
|
|
|
vendorMessages []*goopenai.ChatCompletionMessage
|
|
}
|
|
|
|
func (o *Session) IsEmpty() bool {
|
|
return len(o.Messages) == 0
|
|
}
|
|
|
|
func (o *Session) Append(messages ...*goopenai.ChatCompletionMessage) {
|
|
if o.vendorMessages != nil {
|
|
for _, message := range messages {
|
|
o.Messages = append(o.Messages, message)
|
|
o.appendVendorMessage(message)
|
|
}
|
|
} else {
|
|
o.Messages = append(o.Messages, messages...)
|
|
}
|
|
}
|
|
|
|
func (o *Session) GetVendorMessages() (ret []*goopenai.ChatCompletionMessage) {
|
|
if o.vendorMessages == nil {
|
|
o.vendorMessages = []*goopenai.ChatCompletionMessage{}
|
|
for _, message := range o.Messages {
|
|
o.appendVendorMessage(message)
|
|
}
|
|
}
|
|
ret = o.vendorMessages
|
|
return
|
|
}
|
|
|
|
func (o *Session) appendVendorMessage(message *goopenai.ChatCompletionMessage) {
|
|
if message.Role != common.ChatMessageRoleMeta {
|
|
o.vendorMessages = append(o.vendorMessages, message)
|
|
}
|
|
}
|
|
|
|
func (o *Session) GetLastMessage() (ret *goopenai.ChatCompletionMessage) {
|
|
if len(o.Messages) > 0 {
|
|
ret = o.Messages[len(o.Messages)-1]
|
|
}
|
|
return
|
|
}
|
|
|
|
func (o *Session) String() (ret string) {
|
|
for _, message := range o.Messages {
|
|
ret += fmt.Sprintf("\n--- \n[%v]\n%v", message.Role, message.Content)
|
|
if message.MultiContent != nil {
|
|
for _, part := range message.MultiContent {
|
|
if part.Type == goopenai.ChatMessagePartTypeImageURL {
|
|
ret += fmt.Sprintf("\n%v: %v", part.Type, *part.ImageURL)
|
|
} else if part.Type == goopenai.ChatMessagePartTypeText {
|
|
ret += fmt.Sprintf("\n%v: %v", part.Type, part.Text)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|