davidkhala.ai 0.0.8__tar.gz → 0.1.0__tar.gz

{davidkhala_ai-0.0.8 → davidkhala_ai-0.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: davidkhala.ai
-Version: 0.0.8
+Version: 0.1.0
 Summary: misc AI modules
 Requires-Python: >=3.13
 Provides-Extra: ali

{davidkhala_ai-0.0.8 → davidkhala_ai-0.1.0}/davidkhala/ai/api/siliconflow.py

@@ -47,3 +47,16 @@ class SiliconFlow(API):
         }
         response = self.request(f"{self.base_url}/embeddings", "POST", json=json)
         return [_['embedding'] for _ in response['data']]
+
+    def which(self, query: str, documents: list[str], **kwargs)->tuple[str,int]:
+        json = {
+            'model': self.model,
+            'query': query,
+            'documents': documents,
+            **kwargs
+        }
+        response = self.request(f"{self.base_url}/rerank", "POST", json=json)
+        most_relevant_index = max(response['results'], key=lambda x: x['relevance_score'])['index']
+
+        return documents[most_relevant_index], most_relevant_index
+

{davidkhala_ai-0.0.8 → davidkhala_ai-0.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "davidkhala.ai"
-version = "0.0.8"
+version = "0.1.0"
 description = "misc AI modules"
 readme = "README.md"
 requires-python = ">=3.13"