Merge 'fix/tests' into 'main'

fix: tests, version, readme

See merge request: !18

Author: ZaynJarvis

README.md

@@ -88,9 +88,9 @@ print("search results:", result)
 
 ### Example Guides
 
-#### Vector Examples (Pytest)
+#### Vector Examples
 
-The integration guides under `examples/vector` mirror the Go SDK walkthroughs (`E1`–`E5`). Each test connects to a live VikingDB environment and exercises a specific workflow.
+The integration guides under `examples/vector` mirror the Go SDK walkthroughs (`1`–`6`). Each test connects to a live VikingDB environment and exercises a specific workflow.
 
 1. Set the required environment variables (or create a `.env` file in the project root):
 
@@ -114,13 +114,13 @@ The integration guides under `examples/vector` mirror the Go SDK walkthroughs (`
 2. Install pytest (if not already available):
 
    ```bash
-   pip install pytest
+   uv add --dev pytest
    ```
 
 3. Execute the guides:
 
    ```bash
-   pytest examples/vector -k guide
+   uv run pytest examples/vector -k scenario
    ```
 
 Each scenario writes temporary documents using unique session tags and cleans them up afterwards.
@@ -185,9 +185,9 @@ vikingdb/
 
 examples/
 ├── vector/              # Vector integration guides (pytest)
-│   ├── E1_connectivity_test.py
-│   ├── E2_collection_lifecycle_test.py
-│   ├── E3_*_test.py     # Search and indexing examples
+│   ├── 1_connectivity_test.py
+│   ├── 2_collection_lifecycle_test.py
+│   ├── 3_*_test.py     # Search and indexing examples
 │   └── ...
 └── memory/              # Memory usage examples
     ├── 01_init_client_and_collection.py

examples/vector/6_exception_handling.py

@@ -15,61 +15,32 @@
 
 from __future__ import annotations
 
-import sys
+import os
 
 from vikingdb import IAM
-from vikingdb.vector import VikingVector
-from vikingdb.vector.exceptions import VikingVectorException, VikingConnectionException
+from vikingdb.vector import SearchByRandomRequest, VikingVector
 
-from .test_helper import EnvConfig, load_config
 
-
-def _build_vector_client(*, host_override: str | None = None, timeout: int | None = None) -> tuple[EnvConfig, VikingVector]:
-    """Create a VikingVector client using the shared guide configuration."""
-    config = load_config()
-    auth = IAM(ak=config.access_key, sk=config.secret_key)
+def main() -> None:
+    """Connectivity quickstart mirroring the snippet test."""
+    auth = IAM(
+        ak=os.environ["VIKINGDB_AK"],
+        sk=os.environ["VIKINGDB_SK"],
+    )
     client = VikingVector(
-        host=host_override if host_override is not None else config.host,
-        region=config.region,
-        scheme=config.scheme,
+        host=os.environ["VIKINGDB_HOST"],
+        region=os.environ["VIKINGDB_REGION"],
         auth=auth,
-        timeout=timeout if timeout is not None else 30,
+        scheme="https",
+        timeout=30,
+    )
+    index_client = client.index(
+        collection_name="unknown",
+        index_name="unknown",
     )
-    return config, client
-
-
-def demo_collection_not_exist() -> None:
-    """Attempt to fetch from a non-existent collection to trigger an SDK exception."""
-    try:
-        config, client = _build_vector_client()
-        missing_collection = f"{config.collection}-missing"
-        collection_client = client.collection(
-            collection_name=missing_collection,
-            project_name=config.project_name,
-            resource_id=config.resource_id,
-        )
-        # Intentionally request a fake ID
-        collection_client.fetch({"ids": ["non-existent-id"]})
-    except VikingVectorException as e:
-        print(f"Caught VikingVectorException: code={getattr(e, 'code', None)} message={e}")
-
-
-def demo_wrong_host_network_error() -> None:
-    """Use an invalid host to demonstrate a network connection error."""
-    try:
-        _build_vector_client(host_override="in-v-alid.io", timeout=1)
-    except VikingConnectionException as e:
-        print(f"Caught VikingConnectionException: message={e}")
-
-
 
-def main() -> int:
-    print("-- Scenario 6: Exception handling demos --")
-    demo_collection_not_exist()
-    demo_wrong_host_network_error()
-    print("-- Done --")
-    return 0
+    index_client.search_by_random(SearchByRandomRequest(limit=1))
 
 
 if __name__ == "__main__":
-    raise SystemExit(main())
+    main()

examples/vector/viking_test.py

@@ -28,24 +28,26 @@
     VikingVector,
 )
 
-from .test_helper import (
-    Clients,
-    EnvConfig,
-    DEFAULT_REGION,
-    EMBEDDING_MODEL_NAME,
-    TEXT_COLLECTION,
-    TEXT_INDEX,
+# Ensure local helper can be imported when running via pytest path
+import sys
+from pathlib import Path
+sys.path.insert(0, str(Path(__file__).parent))
+
+from test_helper import (
     VECTOR_COLLECTION,
     VECTOR_INDEX,
-    require_env_vars,
     load_config,
-    build_clients,
     embed_dense_vectors,
-    StoryChapter,
     build_story_chapters,
     chapters_to_upsert,
     assign_chapter_ids_via_search,
-    search_chapter_by_narrative,
+    new_session_tag,
+    build_request_options,
+    cleanup_chapters,
+    bool_and_filters,
+    session_paragraph_bounds,
+    score_at_least_filter,
+    assert_keyword_hit_titles,
 )
 
 

vikingdb/version.py

@@ -1,4 +1,4 @@
 # Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd.
 # SPDX-License-Identifier: Apache-2.0
 
-__version__ = '0.1.1'
+__version__ = '0.1.2'