types

type Embeddings struct {
	Vectors []Vector `json:"vectors"`
}

type ITzap[T any, Z any] interface {
	AddUserMessage(contents ...string) T
	AddAssistantMessage(contents ...string) T
	AddSystemMessage(contents ...string) T
	SetHeader(header string) T

	LoadFileDir(dir string, ext string) T
	ChangeFilepath(filepath string) T
	FetchTask() T

	LoadTask(filepath string) T
	LoadTaskOrRequestNewTask(filepath string) T
	LoadTaskOrRequestNewTaskMD5(filepath string) T

	RequestChat() T
	RequestTextToSpeech(language, voice string) T

	//Tzap Primitives
	New() T
	AddTzap(tc T) T
	HijackTzap(tc T) T

	//ControlFlow
	WorkTzap(fn func(t T)) T
	IsolatedTzap(fn func(t T)) T
	MutationTzap(fn func(t T) T) T
	Map(func(T) T) T
	Accumulate(func(T) T) T
	Exit() T

	ApplyTemplate(nt NamedTemplate[T, Z]) T
	ApplyTemplateFN(nt func(T) T) T
	ApplyTemplateP(T) T

	Recursive(func(tzapThatCreatesNewChildren T) T) T
	CheckAndHandleGlobalOccurrences(references int, filename string, noOccurrence, handleOccurrence func(T) T) T
	CheckAndHandleRecurrences(references int, filename string, noReccurance, handleRecurrence func(T) T) T
	FileMustContainHandleGlobalOccurrences(references int, filename string, noOccurrence, handleOccurrence func(T) T) T
}

type KeyValue struct {
	Key   string `json:"key"`
	Value string `json:"value"`
}

type MappedInterface map[string]interface{}

type Message struct {
	Role    string
	Content string
}

type NamedTemplate[T any, Z any] struct {
	Name     string
	Template func(T) Z // Assuming the function takes no arguments and returns no values
}

type Query struct {
	Values []float32 `json:"values"`
}

type QueryFilter struct {
	Filter map[string]string `json:"filter"`
	Values []float32         `json:"values"`
}

type QueryJson struct {
	Queries []Query `json:"queries"`
}

type QueryRequest struct {
	TopK            int           `json:"topK"`
	IncludeMetadata bool          `json:"includeMetadata"`
	Namespace       string        `json:"namespace"`
	Queries         []QueryFilter `json:"queries"`
}

type SearchResults struct {
	Results []Vector
}

type TGenerator interface {
	TextToSpeech(ctx context.Context, content, language, voice string) (*[]byte, error)
	SpeechToText(ctx context.Context, audioContent *[]byte, language string) (string, error)
	FetchEmbedding(ctx context.Context, content ...string) ([][]float32, error)
	AddEmbeddingDocument(ctx context.Context, id string, embedding []float32, metadata map[string]string) error
	GetEmbeddingDocument(ctx context.Context, id string) (Vector, bool, error)
	DeleteEmbeddingDocument(ctx context.Context, id string) error
	SearchWithEmbedding(ctx context.Context, embedding QueryFilter, k int) (SearchResults, error)
	ListAllEmbeddingsIds(ctx context.Context) (SearchResults, error)
	GenerateChat(ctx context.Context, messages []Message, stream bool) (string, error)
	CountTokens(ctx context.Context, content string) (int, error)
	OffsetTokens(ctx context.Context, content string, from int, to int) (string, error)
}

type TzapConnector func() (TGenerator, config.Configuration)

type UnimplementedTGenerator struct {
	TGenerator
}

type Vector struct {
	ID        string            `json:"id"`
	TimeStamp int               `json:"timestamp"`
	Metadata  map[string]string `json:"metadata"`
	Values    []float32         `json:"values"`
}