Feat: read from chat files

This commit is contained in:
Grail Finder
2024-11-16 06:41:52 +03:00
parent 3cbad31a16
commit 84b06c0da5
5 changed files with 147 additions and 29 deletions

View File

@@ -1,5 +1,10 @@
package models
import (
"fmt"
"strings"
)
// type FuncCall struct {
// XMLName xml.Name `xml:"tool_call"`
// Name string `xml:"name"`
@@ -56,6 +61,22 @@ type MessagesStory struct {
Content string `json:"content"`
}
func (m MessagesStory) ToText() string {
icon := ""
switch m.Role {
case "assistant":
icon = "<🤖>: "
case "user":
icon = "<user>: "
case "system":
icon = "<system>: "
case "tool":
icon = "<tool>: "
}
textMsg := fmt.Sprintf("%s%s\n", icon, m.Content)
return strings.ReplaceAll(textMsg, "\n\n", "\n")
}
type ChatBody struct {
Model string `json:"model"`
Stream bool `json:"stream"`