types

type CompletionMessage struct {
	Content      string
	FunctionCall *FunctionCall `json:"function_call,omitempty"`
	FinishReason FinishReason  `json:"finish_reason,omitempty"`
}

type DBCollectionInterface[T any] interface {
	BatchSet(pairs []KeyValue[T]) (int, error)
	Get(key string) (T, bool)
	GetAll() []KeyValue[T]
	ScanGet(key string) (KeyValue[T], bool)
	Set(key string, value T) error
	StartInit()
}

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

type FileReader interface {
	// Open opens the file and returns a ReadCloser for accessing its contents.
	Open() (io.ReadCloser, error)
	// Name returns the name of the file.
	FilePath() string
	Stat() (fs.FileInfo, error)
}

type FileWalker interface {
	GetFiles() ([]FileReader, error)
}

type FinishReason string

type FunctionCall struct {
	Name      string `json:"name,omitempty"`
	Arguments string `json:"arguments,omitempty"`
}

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

	LoadFileDir(dir string) T
	LoadFiles(filepaths []string) T
	ChangeFilepath(filepath string) T

	LoadCompletion(filepath string) T
	LoadCompletionOrRequestCompletionMD5(filePath string) T
	LoadCompletionOrRequestCompletion(filepath string) T

	RequestChatCompletion() T
	StoreCompletion(filePath string) T
	RequestTextToSpeech(language, voice string) T

	//Tzap Primitives
	New() T
	AddTzap(tc T) T
	HijackTzap(tc T) T
	CloneTzap(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) T
	Accumulate(func(t T) T) T
	Exit() T

	ApplyWorkflow(nt NamedWorkflow[T, Z]) T
	ApplyErrorWorkflow(nt NamedWorkflow[T, Z], fn func(t Z) error) T
	ApplyWorkflowFN(nt func(t T) T) T
	ApplyWorkflowP(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

	CountTokens(content string) (int, error)
	OffsetTokens(content string, from int, to int) (string, error)
}

type KeyValue[T any] struct {
	Key   string `json:"key"`
	Value T      `json:"value"`
}

type MappedInterface map[string]interface{}

type Message struct {
	Role    string
	Content string
}

type Metadata struct {
	ID            string `json:"id"`
	Filename      string `json:"filename"`
	Start         int    `json:"start"`
	End           int    `json:"end"`
	LineStart     int    `json:"lineStart"`
	TruncatedEnd  int    `json:"truncatedEnd"`
	SplitPart     string `json:"splitPart"`
	RealSplitPart string `json:"realSplitPart"`
}

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

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

type QueryFilter struct {
	Filter map[string]string `json:"filter"`
	Values [1536]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 SearchResult struct {
	Vector     Vector    `json:"vector"`
	PCA        []float32 `json:"pca"`
	Similarity float32   `json:"score"`
}

type SearchResults struct {
	Results []SearchResult
}

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) ([][1536]float32, error)
	AddEmbeddingDocument(ctx context.Context, id string, embedding [1536]float32, metadata Metadata) error
	GetEmbeddingDocument(ctx context.Context, id string) (Vector, bool, error)
	DeleteEmbeddingDocument(ctx context.Context, id string) error
	DeleteEmbeddingDocuments(ctx context.Context, ids []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, functions string) (CompletionMessage, error)
	CountTokens(ctx context.Context, content string) (int, error)
	OffsetTokens(ctx context.Context, content string, from int, to int) (string, int, error)
	RawTokens(ctx context.Context, content string) ([]string, error)
}

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

type UnimplementedTGenerator struct {
	TGenerator
}

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