fix: cap pymilvus<2.6.10, add milvus_token param, sync version (#19)

- Cap pymilvus to <2.6.10 to fix MilvusClient hanging with milvus-lite 2.5.1
- Add milvus_token parameter to VectorGraphRAG and create_rag for Zilliz Cloud support
- Sync __version__ in __init__.py to match pyproject.toml (0.1.1)
- Add milvus_token column to Milvus backend table in README
- Add Zilliz Cloud section and deployment modes table to docs/getting-started.md

Signed-off-by: Cheney Zhang <chen.zhang@zilliz.com>

Author: zc277584121

README.md

@@ -183,11 +183,11 @@ Evaluated on three multi-hop QA benchmarks (Recall@5):
 
 Vector Graph RAG supports three Milvus deployment modes — just change `milvus_uri`:
 
-| Mode | `milvus_uri` | Best for |
-|------|-------------|----------|
-| **Milvus Lite** (default) | `./vector_graph_rag.db` | Personal use, dev — zero config |
-| **Milvus Server** | `http://localhost:19530` | Multi-dataset, team environments |
-| ⭐ **Zilliz Cloud** | `https://in03-xxx.api.gcp-us-west1.zillizcloud.com` | Production, fully managed — [free tier available](https://cloud.zilliz.com/signup?utm_source=github&utm_medium=referral&utm_campaign=vector-graph-rag-readme) |
+| Mode | `milvus_uri` | `milvus_token` | Best for |
+|------|-------------|----------------|----------|
+| **Milvus Lite** (default) | `./vector_graph_rag.db` | — | Personal use, dev — zero config |
+| **Milvus Server** | `http://localhost:19530` | Optional | Multi-dataset, team environments |
+| ⭐ **Zilliz Cloud** | `https://in03-xxx.api.gcp-us-west1.zillizcloud.com` | API key | Production, fully managed — [free tier available](https://cloud.zilliz.com/signup?utm_source=github&utm_medium=referral&utm_campaign=vector-graph-rag-readme) |
 
 > **Recommended:** [Zilliz Cloud](https://cloud.zilliz.com/signup?utm_source=github&utm_medium=referral&utm_campaign=vector-graph-rag-readme) gives you zero-config, zero-ops Milvus with concurrent access and real-time indexing — no Docker needed.
 

docs/getting-started.md

@@ -65,6 +65,26 @@ rag = VectorGraphRAG(
 )
 ```
 
+### With Zilliz Cloud
+
+```python
+rag = VectorGraphRAG(
+    milvus_uri="https://in03-xxx.api.gcp-us-west1.zillizcloud.com",
+    milvus_token="your-api-key",
+)
+```
+
+!!! tip "Free tier available"
+    [Sign up for Zilliz Cloud](https://cloud.zilliz.com/signup?utm_source=github&utm_medium=referral&utm_campaign=vector-graph-rag-docs) to get a free cluster. Use the cluster endpoint as `milvus_uri` and your API key as `milvus_token`.
+
+### Milvus Deployment Modes
+
+| Mode | `milvus_uri` | `milvus_token` | Best for |
+|------|-------------|----------------|----------|
+| **Milvus Lite** (default) | `./vector_graph_rag.db` | — | Personal use, dev — zero config |
+| **Milvus Server** | `http://localhost:19530` | Optional | Multi-dataset, team environments |
+| ⭐ **Zilliz Cloud** | `https://in03-xxx.api.gcp-us-west1.zillizcloud.com` | API key | Production, fully managed |
+
 !!! info "Collection naming"
     With `collection_prefix="my_project"`, collections are named `my_project_vgrag_entities`, `my_project_vgrag_relations`, `my_project_vgrag_passages`.
 

pyproject.toml

@@ -22,7 +22,7 @@ classifiers = [
 ]
 
 dependencies = [
-    "pymilvus>=2.4.0",
+    "pymilvus>=2.4.0,<2.6.10",
     "milvus-lite>=2.4.0",
     "numpy>=1.24.0",
     "scipy>=1.10.0",

src/vector_graph_rag/__init__.py

@@ -18,7 +18,7 @@
 from vector_graph_rag.rag import VectorGraphRAG, create_rag
 from vector_graph_rag.llm.cache import LLMCache, get_llm_cache
 
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 
 __all__ = [
     "Settings",

src/vector_graph_rag/rag.py

@@ -55,6 +55,7 @@ def __init__(
         self,
         settings: Optional[Settings] = None,
         milvus_uri: Optional[str] = None,
+        milvus_token: Optional[str] = None,
         milvus_db: Optional[str] = None,
         collection_prefix: Optional[str] = None,
         openai_api_key: Optional[str] = None,
@@ -67,6 +68,7 @@ def __init__(
         Args:
             settings: Full settings object (overrides other parameters).
             milvus_uri: Milvus connection URI. Defaults to local file.
+            milvus_token: Milvus authentication token (for Zilliz Cloud).
             milvus_db: Milvus database name. Defaults to None (use server default).
             collection_prefix: Prefix for collection names (e.g., graph/dataset name).
             openai_api_key: OpenAI API key. Uses environment variable if not provided.
@@ -83,6 +85,12 @@ def __init__(
             ...     llm_model="gpt-4o",
             ... )
             >>>
+            >>> # Zilliz Cloud
+            >>> rag = VectorGraphRAG(
+            ...     milvus_uri="https://in03-xxx.api.gcp-us-west1.zillizcloud.com",
+            ...     milvus_token="your-api-key",
+            ... )
+            >>>
             >>> # Separate graphs with collection prefix
             >>> rag = VectorGraphRAG(
             ...     milvus_uri="http://localhost:19530",
@@ -97,6 +105,8 @@ def __init__(
             settings_kwargs = {}
             if milvus_uri:
                 settings_kwargs["milvus_uri"] = milvus_uri
+            if milvus_token:
+                settings_kwargs["milvus_token"] = milvus_token
             if milvus_db:
                 settings_kwargs["milvus_db"] = milvus_db
             if collection_prefix:
@@ -712,6 +722,7 @@ def reset(self) -> None:
 
 def create_rag(
     milvus_uri: Optional[str] = None,
+    milvus_token: Optional[str] = None,
     milvus_db: Optional[str] = None,
     collection_prefix: Optional[str] = None,
     openai_api_key: Optional[str] = None,
@@ -725,6 +736,7 @@ def create_rag(
 
     Args:
         milvus_uri: Milvus connection URI. Defaults to local file.
+        milvus_token: Milvus authentication token (for Zilliz Cloud).
         milvus_db: Milvus database name. Defaults to None (use server default).
         collection_prefix: Prefix for collection names (e.g., graph/dataset name).
         openai_api_key: OpenAI API key. Uses environment variable if not provided.
@@ -741,6 +753,7 @@ def create_rag(
     """
     return VectorGraphRAG(
         milvus_uri=milvus_uri,
+        milvus_token=milvus_token,
         milvus_db=milvus_db,
         collection_prefix=collection_prefix,
         openai_api_key=openai_api_key,