feat: add reranker support to Python SDK

examples/python/.env.example

@@ -0,0 +1,3 @@
+MOSS_PROJECT_ID=your_project_id
+MOSS_PROJECT_KEY=your_project_key
+COHERE_API_KEY=your_cohere_api_key

examples/python/requirements.txt

@@ -1,3 +1,4 @@
-moss>=1.0.0
+-e sdks/python/sdk
+cohere>=5.0.0
 python-dotenv
 openai

examples/python/reranking_sample.py

@@ -0,0 +1,75 @@
+import asyncio
+import json
+import os
+
+from dotenv import load_dotenv
+from moss import MossClient, DocumentInfo, QueryOptions, RerankOptions
+
+load_dotenv()
+
+INDEX_NAME = "rerank-demo-full"
+
+
+async def setup_index(client):
+    """Delete old index, create fresh one with all FAQ data."""
+    faqs_path = os.path.join(os.path.dirname(__file__), "faqs.json")
+    with open(faqs_path, "r") as f:
+        faqs = json.load(f)
+
+    docs = [
+        DocumentInfo(
+            id=faq["id"],
+            text=faq["text"],
+            metadata={k: str(v) for k, v in faq.get("metadata", {}).items()},
+        )
+        for faq in faqs
+    ]
+
+    print(f"Setting up index '{INDEX_NAME}'")
+    try:
+        await client.delete_index(INDEX_NAME)
+        print("Deleted old index.")
+    except Exception:
+        pass
+
+    await client.create_index(INDEX_NAME, docs)
+    print("Index created.")
+
+    await client.load_index(INDEX_NAME)
+    print("Index loaded.\n")
+
+
+async def main():
+    client = MossClient(os.getenv("MOSS_PROJECT_ID"), os.getenv("MOSS_PROJECT_KEY"))
+
+    await setup_index(client)
+
+    print("Without Reranking")
+    results = await client.query(
+        INDEX_NAME,
+        "How to get discount?",
+        QueryOptions(top_k=5, alpha=0.8),
+    )
+    for i, doc in enumerate(results.docs):
+        print(f"  {i + 1}. [{doc.score:.3f}] {doc.text[:100]}...")
+
+    print("\nWith Cohere Reranking")
+    results = await client.query(
+        INDEX_NAME,
+        "How to get discount?",
+        QueryOptions(
+            top_k=10,
+            alpha=0.8,
+            rerank=RerankOptions(
+                provider="cohere",
+                api_key=os.getenv("COHERE_API_KEY"),
+                top_n=5,
+            ),
+        ),
+    )
+    for i, doc in enumerate(results.docs):
+        print(f"  {i + 1}. [{doc.score:.3f}] {doc.text[:100]}...")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

sdks/python/sdk/pyproject.toml

@@ -32,7 +32,11 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
+cohere = [
+    "cohere>=5.0.0",
+]
 dev = [
+    "cohere>=5.0.0",
     "pytest>=8.4.2",
     "pytest-asyncio>=1.2.0",
     "tox>=4.0.0",

sdks/python/sdk/src/moss/__init__.py

@@ -1,62 +1,64 @@
-"""
-Moss Semantic Search SDK
-
-Powerful Python SDK for semantic search using state-of-the-art embedding models.
-
-Example:
-    ```python
-    from moss import MossClient, DocumentInfo
-
-    client = MossClient('your-project-id', 'your-project-key')
-
-    docs = [DocumentInfo(id="1", text="Example document")]
-
-    result = await client.create_index('my-index', docs, 'moss-minilm')
-
-    await client.load_index('my-index')
-    results = await client.query('my-index', 'search query')
-    ```
-"""
-
-from moss_core import (
-    DocumentInfo,
-    GetDocumentsOptions,
-    IndexInfo,
-    IndexStatus,
-    IndexStatusValues,
-    ModelRef,
-    MutationOptions,
-    MutationResult,
-    JobStatus,
-    JobPhase,
-    JobProgress,
-    JobStatusResponse,
-    QueryOptions,
-    QueryResultDocumentInfo,
-    SearchResult,
-)
-
-from .client.moss_client import MossClient
-
-__version__ = "1.0.0b19"
-
-__all__ = [
-    "MossClient",
-    # Core data types
-    "DocumentInfo",
-    "GetDocumentsOptions",
-    "IndexInfo",
-    "SearchResult",
-    "QueryResultDocumentInfo",
-    "ModelRef",
-    "IndexStatus",
-    "IndexStatusValues",
-    "QueryOptions",
-    # Mutation types
-    "MutationResult",
-    "MutationOptions",
-    "JobStatus",
-    "JobPhase",
-    "JobProgress",
-    "JobStatusResponse",
-]
+"""
+Moss Semantic Search SDK
+
+Powerful Python SDK for semantic search using state-of-the-art embedding models.
+
+Example:
+    ```python
+    from moss import MossClient, DocumentInfo
+
+    client = MossClient('your-project-id', 'your-project-key')
+
+    docs = [DocumentInfo(id="1", text="Example document")]
+
+    result = await client.create_index('my-index', docs, 'moss-minilm')
+
+    await client.load_index('my-index')
+    results = await client.query('my-index', 'search query')
+    ```
+"""
+
+from moss_core import (
+    DocumentInfo,
+    GetDocumentsOptions,
+    IndexInfo,
+    IndexStatus,
+    IndexStatusValues,
+    ModelRef,
+    MutationOptions,
+    MutationResult,
+    JobStatus,
+    JobPhase,
+    JobProgress,
+    JobStatusResponse,
+    QueryResultDocumentInfo,
+    SearchResult,
+)
+
+from .client.models import QueryOptions, RerankOptions
+from .client.moss_client import MossClient
+
+__version__ = "1.0.0b19"
+
+__all__ = [
+    "MossClient",
+    # Core data types
+    "DocumentInfo",
+    "GetDocumentsOptions",
+    "IndexInfo",
+    "SearchResult",
+    "QueryResultDocumentInfo",
+    "ModelRef",
+    "IndexStatus",
+    "IndexStatusValues",
+    "QueryOptions",
+    # Mutation types
+    "MutationResult",
+    "MutationOptions",
+    "JobStatus",
+    "JobPhase",
+    "JobProgress",
+    "JobStatusResponse",
+    # Reranking
+    "RerankOptions",
+]