Skip to content

archaeo_super_prompt.modeling.similarity

source module archaeo_super_prompt.modeling.similarity

Functions

source get_ollama_embedding_model(model_id='nomic-embed-text', batch_size=100)

Return a dspy client for an Embedding model from the ollama server.

source get_vllm_embedding_model(model_id='nomic-embed-text', batch_size=100)

Return a dspy client for an Embedding model from the vllm server.

source embed(text: str, embedder: dspy.Embedder)