utils

package
v0.2.3 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Sep 30, 2025 License: MIT Imports: 16 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func ConfirmAction

func ConfirmAction(message string) bool

ConfirmAction prompts the user for confirmation

func CountMockCollections

func CountMockCollections(ctx context.Context, cfg *config.VectorDBConfig) (int, error)

CountMockCollections counts mock collections

func CountMockDocuments

func CountMockDocuments(ctx context.Context, cfg *config.VectorDBConfig, collectionName string) (int, error)

CountMockDocuments counts mock documents

func CountWeaviateCollections

func CountWeaviateCollections(ctx context.Context, cfg *config.VectorDBConfig) (int, error)

CountWeaviateCollections counts Weaviate collections

func CountWeaviateDocuments

func CountWeaviateDocuments(ctx context.Context, cfg *config.VectorDBConfig, collectionName string) (int, error)

CountWeaviateDocuments counts Weaviate documents

func CreateMockClient

func CreateMockClient(cfg *config.VectorDBConfig) *mock.Client

CreateMockClient creates a mock client from configuration

func CreateMockCollection

func CreateMockCollection(ctx context.Context, cfg *config.VectorDBConfig, collectionName, embeddingModel string, customFields []weaviate.FieldDefinition) error

CreateMockCollection creates a mock collection

func CreateMockDocument

func CreateMockDocument(ctx context.Context, cfg *config.VectorDBConfig, collectionName, filePath string, chunkSize int, imageCollection string, skipSmallImages bool, minImageSize int)

CreateMockDocument creates a mock document

func CreateWeaviateClient

func CreateWeaviateClient(cfg *config.VectorDBConfig) (*weaviate.Client, error)

CreateWeaviateClient creates a Weaviate client from configuration

func CreateWeaviateCollection

func CreateWeaviateCollection(ctx context.Context, cfg *config.VectorDBConfig, collectionName, embeddingModel string, customFields []weaviate.FieldDefinition, schemaType string) error

CreateWeaviateCollection creates a Weaviate collection

func CreateWeaviateDocument

func CreateWeaviateDocument(ctx context.Context, cfg *config.VectorDBConfig, collectionName, filePath string, chunkSize int, imageCollection string, skipSmallImages bool, minImageSize int) error

CreateWeaviateDocument creates a Weaviate document

func DeleteAllMockCollections

func DeleteAllMockCollections(ctx context.Context, cfg *config.VectorDBConfig)

DeleteAllMockCollections deletes all documents from all mock collections

func DeleteAllMockDocuments

func DeleteAllMockDocuments(ctx context.Context, cfg *config.VectorDBConfig, collectionName string)

DeleteAllMockDocuments deletes all mock documents in a collection

func DeleteAllWeaviateCollections

func DeleteAllWeaviateCollections(ctx context.Context, cfg *config.VectorDBConfig)

DeleteAllWeaviateCollections deletes all documents from all Weaviate collections

func DeleteAllWeaviateDocuments

func DeleteAllWeaviateDocuments(ctx context.Context, cfg *config.VectorDBConfig, collectionName string)

DeleteAllWeaviateDocuments deletes all Weaviate documents in a collection

func DeleteMockCollections

func DeleteMockCollections(ctx context.Context, cfg *config.VectorDBConfig, collectionNames []string) error

DeleteMockCollections deletes mock collections

func DeleteMockCollectionsByPattern

func DeleteMockCollectionsByPattern(ctx context.Context, cfg *config.VectorDBConfig, pattern string) error

DeleteMockCollectionsByPattern deletes mock collections by pattern

func DeleteMockDocuments

func DeleteMockDocuments(ctx context.Context, cfg *config.VectorDBConfig, collectionName string, documentIDs []string, metadataFilters []string, virtual bool, pattern string, name string)

DeleteMockDocuments deletes mock documents

func DeleteWeaviateCollectionSchema

func DeleteWeaviateCollectionSchema(ctx context.Context, cfg *config.VectorDBConfig, collectionName string) error

DeleteWeaviateCollectionSchema deletes Weaviate collection schema

func DeleteWeaviateCollections

func DeleteWeaviateCollections(ctx context.Context, cfg *config.VectorDBConfig, collectionNames []string) error

DeleteWeaviateCollections deletes Weaviate collections

func DeleteWeaviateCollectionsByPattern

func DeleteWeaviateCollectionsByPattern(ctx context.Context, cfg *config.VectorDBConfig, pattern string) error

DeleteWeaviateCollectionsByPattern deletes Weaviate collections by pattern

func DeleteWeaviateDocuments

func DeleteWeaviateDocuments(ctx context.Context, cfg *config.VectorDBConfig, collectionName string, documentIDs []string, metadataFilters []string, virtual bool, pattern string, name string)

DeleteWeaviateDocuments deletes Weaviate documents

func DisplayRegularDocuments

func DisplayRegularDocuments(documents []weaviate.Document, collectionName string, showLong bool, shortLines int)

DisplayRegularDocuments displays regular documents with styling

func DisplayVirtualDocuments

func DisplayVirtualDocuments(documents []weaviate.Document, collectionName string, showLong bool, shortLines int, summary bool)

DisplayVirtualDocuments displays virtual documents with aggregation and styling

func GetStandaloneDocumentKey

func GetStandaloneDocumentKey(doc weaviate.Document) string

GetStandaloneDocumentKey extracts a unique key for standalone documents

func GetStyledEmoji

func GetStyledEmoji(emoji string) string

GetStyledEmoji returns a styled emoji string

func GetStyledKeyDimmed

func GetStyledKeyDimmed(key string) string

GetStyledKeyDimmed returns a dimmed styled key string

func GetStyledKeyProminent

func GetStyledKeyProminent(key string) string

GetStyledKeyProminent returns a styled key string

func GetStyledNumber

func GetStyledNumber(num interface{}) string

GetStyledNumber returns a styled number string

func GetStyledValueDimmed

func GetStyledValueDimmed(value string) string

GetStyledValueDimmed returns a dimmed styled value string

func GetValueType

func GetValueType(value interface{}) string

GetValueType returns a string representation of the Go type

func IsImageContent

func IsImageContent(content string) bool

IsImageContent checks if content looks like base64 image data

func IsImageDocument

func IsImageDocument(metadata map[string]interface{}) bool

IsImageDocument checks if metadata indicates this is an image document

func IsImageVirtualDocument

func IsImageVirtualDocument(vdoc VirtualDocument) bool

IsImageVirtualDocument checks if a virtual document represents an image collection

func ListMockCollections

func ListMockCollections(ctx context.Context, cfg *config.VectorDBConfig, limit int, virtual bool)

ListMockCollections lists mock collections

func ListMockDocuments

func ListMockDocuments(ctx context.Context, cfg *config.VectorDBConfig, collectionName string, limit int, showLong bool, shortLines int, virtual bool, summary bool)

ListMockDocuments lists mock documents

func ListWeaviateCollections

func ListWeaviateCollections(ctx context.Context, cfg *config.VectorDBConfig, limit int, virtual bool)

ListWeaviateCollections lists Weaviate collections

func ListWeaviateDocuments

func ListWeaviateDocuments(ctx context.Context, cfg *config.VectorDBConfig, collectionName string, limit int, showLong bool, shortLines int, virtual bool, summary bool)

ListWeaviateDocuments lists Weaviate documents

func LoadConfigWithOverrides

func LoadConfigWithOverrides() (*config.Config, error)

LoadConfigWithOverrides loads configuration with command-line overrides

func ParseFieldDefinitions

func ParseFieldDefinitions(fieldsStr string) ([]weaviate.FieldDefinition, error)

ParseFieldDefinitions parses field definitions from a string

func PrintError

func PrintError(message string)

PrintError prints an error message in red

func PrintHeader

func PrintHeader(message string)

PrintHeader prints a header message in bold cyan

func PrintInfo

func PrintInfo(message string)

PrintInfo prints an info message in blue

func PrintStyledEmoji

func PrintStyledEmoji(emoji string)

PrintStyledEmoji prints a styled emoji

func PrintStyledFilename

func PrintStyledFilename(filename string)

PrintStyledFilename prints a styled filename

func PrintStyledID

func PrintStyledID(id string)

PrintStyledID prints a styled ID

func PrintStyledKey

func PrintStyledKey(key string)

PrintStyledKey prints a styled key

func PrintStyledKeyNumberProminentWithEmoji

func PrintStyledKeyNumberProminentWithEmoji(key string, num int, emoji string)

PrintStyledKeyNumberProminentWithEmoji prints a prominent key-number pair with emoji

func PrintStyledKeyProminent

func PrintStyledKeyProminent(key string)

PrintStyledKeyProminent prints a prominent styled key

func PrintStyledKeyValueDimmed

func PrintStyledKeyValueDimmed(key, value string)

PrintStyledKeyValueDimmed prints a dimmed key-value pair

func PrintStyledKeyValueProminentWithEmoji

func PrintStyledKeyValueProminentWithEmoji(key, value, emoji string)

PrintStyledKeyValueProminentWithEmoji prints a prominent key-value pair with emoji

func PrintStyledNumber

func PrintStyledNumber(num int)

PrintStyledNumber prints a styled number

func PrintStyledValue

func PrintStyledValue(value string)

PrintStyledValue prints a styled value

func PrintStyledValueDimmed

func PrintStyledValueDimmed(value string)

PrintStyledValueDimmed prints a dimmed styled value

func PrintSuccess

func PrintSuccess(message string)

PrintSuccess prints a success message in green

func PrintWarning

func PrintWarning(message string)

PrintWarning prints a warning message in yellow

func ShowDocumentMetadata

func ShowDocumentMetadata(doc weaviate.Document, collectionName string)

ShowDocumentMetadata shows expanded metadata for a document

func ShowDocumentSchema

func ShowDocumentSchema(doc weaviate.Document, collectionName string)

ShowDocumentSchema shows the schema of a Weaviate document

func ShowMockCollection

func ShowMockCollection(ctx context.Context, cfg *config.VectorDBConfig, collectionName string, shortLines int, noTruncate bool, verbose bool, showSchema bool, showMetadata bool)

ShowMockCollection shows mock collection details

func ShowMockDocument

func ShowMockDocument(ctx context.Context, cfg *config.VectorDBConfig, collectionName string, documentIDs []string, showLong bool, shortLines int, metadataFilters []string, name string, showSchema bool, expandMetadata bool)

ShowMockDocument shows mock document details

func ShowWeaviateCollection

func ShowWeaviateCollection(ctx context.Context, cfg *config.VectorDBConfig, collectionName string, shortLines int, noTruncate bool, verbose bool, showSchema bool, showMetadata bool)

ShowWeaviateCollection shows Weaviate collection details

func ShowWeaviateDocument

func ShowWeaviateDocument(ctx context.Context, cfg *config.VectorDBConfig, collectionName string, documentIDs []string, showLong bool, shortLines int, metadataFilters []string, name string, showSchema bool, expandMetadata bool)

ShowWeaviateDocument shows Weaviate document details

func SmartTruncate

func SmartTruncate(value, key string, shortLines int) string

SmartTruncate intelligently truncates content based on key type

func TruncateMetadataValue

func TruncateMetadataValue(value interface{}, maxLength int) string

TruncateMetadataValue truncates a metadata value to a maximum length

func TruncateStringByLines

func TruncateStringByLines(text string, maxLines int) string

TruncateStringByLines truncates text to a maximum number of lines

Types

type VirtualDocument

type VirtualDocument struct {
	OriginalFilename string
	TotalChunks      int
	Chunks           []weaviate.Document
	Metadata         map[string]interface{}
}

VirtualDocument represents a virtual document structure

func AggregateDocumentsByOriginal

func AggregateDocumentsByOriginal(documents []weaviate.Document) []VirtualDocument

AggregateDocumentsByOriginal aggregates documents by their original filename/source

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL