Inference provider code snippet needs update
#127
by 0xarnav - opened
The sample code given for the inference API does not work as the method definition has changed.
result = client.sentence_similarity(
{
"source_sentence": "That is a happy person",
"sentences": [
"That is a happy dog",
"That is a very happy person",
"Today is a sunny day"
]
},
model="sentence-transformers/all-MiniLM-L6-v2",
)
The sentence_similarity method now accepts sentence and other_sentences as parameters.
(method) def sentence_similarity(
sentence: str,
other_sentences: List[str],
*,
model: str | None = None
) -> List[float]
I suppose this can't be edited by the org, but a note would be helpful.
Thank you for reporting! I've sent this up the chain, so that hopefully the sample code itself can be fixed!
- Tom Aarsen