deepinfra/model/api.go

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"`
}