How to use jjstuart/rego-retrieval with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("jjstuart/rego-retrieval") sentences = [ "policy for how can i verify if a tekton task version is still supported by checking for the build.appstudio.redhat.com/expires-on annotation?", "Helper: lib.to_array\nSignature: to_array(s)\nDescription: ", "Helper: lib.pipelinerun_attestations\nSignature: pipelinerun_attestations\nDescription: ", "Helper: lib.k8s.name\nSignature: name(resource)\nDescription: " ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
