feat: add custom

model/api.go

@@ -0,0 +1,33 @@
+package model
+
+import (
+	"bytes"
+	"io"
+)
+
+type Inputer interface {
+	Body() *bytes.Buffer
+	Parse(io.Reader) error
+}
+
+type Outputer interface {
+	Output() string
+	OutputRaw() string
+}
+
+type MessageBuilder[T any] interface {
+	System(prompt string) T
+	User(prompt string) T
+	Assistant(prompt string) T
+}
+
+type Message struct {
+	Role    string `json:"role"`
+	Content string `json:"content"`
+}
+
+type Choice struct {
+	Index        int     `json:"index"`
+	Message      Message `json:"message"`
+	FinishReason string  `json:"finish_reason"`
+}