api

package

v1.0.20 Latest Latest Go to latest Published: Sep 17, 2025 License: MIT Imports: 6 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type API struct {
	// contains filtered or unexported fields
}

API provides methods to interact with the document storage.

func NewAPI(storage *storage.Storage, llm *llm.Embeddings) *API

NewAPI creates a new API instance.

func (a *API) DeleteDocument(url, context string) error

DeleteDocument deletes a document by URL.

func (a *API) GetContexts() ([]string, error)

GetContexts retrieves a list of unique contexts.

func (a *API) GetDocument(url string, context string) (*storage.Document, error)

GetDocument retrieves a document by URL.

func (a *API) ListDocuments(context string, limit int) ([]*storage.Document, error)

ListDocuments lists all documents, optionally filtered by context.

func (a *API) Llm() *llm.Embeddings

Llm returns the llm instance.

func (a *API) Search(query string, numResults int, context string) ([]SearchResult, error)

Search finds the most similar documents to a query, up to numResults, optionally filtered by context.

func (a *API) UpsertDirect(doc *storage.Document) error

UpsertDirect upserts a document directly.

func (a *API) UpsertDocument(baseURL, url, title, description, content, checksum, context, sourceType string, embeddings []float32) error

UpsertDocument stores a new document or updates an existing one.

type SearchResult struct {
	Doc   *storage.Document
	Score float64
}