embedworkflows

func ConfirmEmbeddingSearch ¶ added in v0.8.0

func ConfirmEmbeddingSearch(yes bool) types.NamedWorkflow[*tzap.Tzap, *tzap.Tzap]

func EmbedWorkflow ¶ added in v0.8.6

func EmbedWorkflow(searchResults types.SearchResults) types.NamedWorkflow[*tzap.Tzap, *tzap.Tzap]

func FetchOrCachedEmbeddingForFilesWorkflow ¶

func FetchOrCachedEmbeddingForFilesWorkflow(files []types.FileReader) types.NamedWorkflow[*tzap.Tzap, *tzap.Tzap]

func LoadAndFetchEmbeddings ¶ added in v0.8.0

func LoadAndFetchEmbeddings(files []types.FileReader, embedder *embed.Embedder, yes bool) types.NamedWorkflow[*tzap.Tzap, *tzap.Tzap]

func PrepareEmbedFilesWorkflow ¶

func PrepareEmbedFilesWorkflow(files []types.FileReader, embedder *embed.Embedder) types.NamedWorkflow[*tzap.Tzap, *tzap.Tzap]

func SaveAndLoadEmbeddingsToDB ¶

func SaveAndLoadEmbeddingsToDB() types.NamedWorkflow[*tzap.Tzap, *tzap.Tzap]

func SearchFilesWorkflow ¶

func SearchFilesWorkflow(query types.QueryRequest, excludeFiles []string, k int, n int) types.NamedWorkflow[*tzap.Tzap, *tzap.Tzap]

k is amount of embeddings to be included. When using inspiration files, embeddings are likely to be duplicated and as such are filtered out. n is used to increase how many embeddings are fetched but are trimmed to only contain top K after filtering.