Allow WordLlama.rank to not sort the results

In some cases, it is useful to get a similarity measure for a list of documents, but not having it sorted (for example, to correlate with some characteristic of the documents in the original list).

This change adds a parameter to rank(), so that the caller can decide the default, sorted, result (as it is now), or an unsorted result.

wordllama/inference.py

@@ -239,15 +239,18 @@ def similarity(self, text1: str, text2: str) -> float:
         embedding2 = self.embed(text2)
         return self.vector_similarity(embedding1[0], embedding2[0]).item()
 
-    def rank(self, query: str, docs: List[str]) -> List[Tuple[str, float]]:
+    def rank(self, query: str, docs: List[str], sort: bool = True) -> List[Tuple[str, float]]:
         """Rank documents based on their similarity to a query.
 
+        Result may be sorted by similarity score in descending order, or not (see `sort` parameter)
+
         Args:
             query (str): The query text.
             docs (List[str]): The list of document texts to rank.
+            sort (bool): Sort documents by similarity, or not (respect the order in `docs`)
 
         Returns:
-            List[Tuple[str, float]]: A list of tuples `(doc, score)`, sorted by similarity score in descending order.
+            List[Tuple[str, float]]: A list of tuples `(doc, score)`.
         """
         assert isinstance(query, str), "Query must be a string"
         assert (
@@ -259,7 +262,8 @@ def rank(self, query: str, docs: List[str]) -> List[Tuple[str, float]]:
 
         scores = np.atleast_1d(scores.squeeze())
         similarities = list(zip(docs, scores.tolist()))
-        similarities.sort(key=lambda x: x[1], reverse=True)
+        if sort:
+            similarities.sort(key=lambda x: x[1], reverse=True)
         return similarities
 
     def deduplicate(