openviking 0.2.4.dev20__tar.gz → 0.2.4.dev33__tar.gz

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/.github/workflows/_build.yml

@@ -79,6 +79,9 @@ jobs:
       run: |
         git reset --hard HEAD
         git clean -fd
+        # For sdist, ensure local runtime binaries are not packaged even if present
+        rm -rf openviking/bin openviking/lib third_party/agfs/bin || true
+        rm -f openviking/storage/vectordb/*.so openviking/storage/vectordb/*.dylib openviking/storage/vectordb/*.dll openviking/storage/vectordb/*.exe || true
         rm -rf openviking/_version.py openviking.egg-info
         # Ignore uv.lock changes to avoid dirty state in setuptools_scm
         git update-index --assume-unchanged uv.lock || true
@@ -99,7 +102,7 @@ jobs:
       run: uv run --frozen python -m build --sdist
 
     - name: Store the distribution packages
-      uses: actions/upload-artifact@v6
+      uses: actions/upload-artifact@v7
       with:
         name: python-package-distributions-sdist
         path: dist/*.tar.gz
@@ -264,7 +267,7 @@ jobs:
         rmdir dist_fixed
 
     - name: Store the distribution packages
-      uses: actions/upload-artifact@v6
+      uses: actions/upload-artifact@v7
       with:
         name: python-package-distributions-linux-${{ matrix.python-version }}
         path: dist/
@@ -402,7 +405,7 @@ jobs:
       run: uv run --frozen python -m build --wheel
 
     - name: Store the distribution packages
-      uses: actions/upload-artifact@v6
+      uses: actions/upload-artifact@v7
       with:
         name: python-package-distributions-${{ matrix.os == 'macos-latest' && 'macos-arm64' || matrix.os == 'macos-15-intel' && 'macos-x86_64' || matrix.os == 'windows-latest' && 'windows-x86_64' || matrix.os }}-${{ matrix.python-version }}
         path: dist/

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/.github/workflows/_publish.yml

@@ -80,7 +80,7 @@ jobs:
     steps:
     - name: Download all the dists (Same Run)
       if: inputs.build_run_id == ''
-      uses: actions/download-artifact@v7
+      uses: actions/download-artifact@v8
       with:
         pattern: python-package-distributions-*
         path: dist/
@@ -88,7 +88,7 @@ jobs:
 
     - name: Download all the dists (Cross Run)
       if: inputs.build_run_id != ''
-      uses: actions/download-artifact@v7
+      uses: actions/download-artifact@v8
       with:
         run-id: ${{ inputs.build_run_id }}
         github-token: ${{ secrets.GITHUB_TOKEN }}
@@ -127,7 +127,7 @@ jobs:
     steps:
     - name: Download all the dists (Same Run)
       if: inputs.build_run_id == ''
-      uses: actions/download-artifact@v7
+      uses: actions/download-artifact@v8
       with:
         pattern: python-package-distributions-*
         path: dist/
@@ -135,7 +135,7 @@ jobs:
 
     - name: Download all the dists (Cross Run)
       if: inputs.build_run_id != ''
-      uses: actions/download-artifact@v7
+      uses: actions/download-artifact@v8
       with:
         run-id: ${{ inputs.build_run_id }}
         github-token: ${{ secrets.GITHUB_TOKEN }}

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/.github/workflows/build-docker-image.yml

@@ -33,7 +33,7 @@ jobs:
           submodules: recursive
 
       - name: Log in to the Container registry
-        uses: docker/login-action@v3
+        uses: docker/login-action@v4
         with:
           registry: ${{ env.REGISTRY }}
           username: ${{ github.actor }}
@@ -46,10 +46,10 @@ jobs:
           images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
 
       - name: Set up QEMU
-        uses: docker/setup-qemu-action@v3
+        uses: docker/setup-qemu-action@v4
 
       - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v3
+        uses: docker/setup-buildx-action@v4
 
       - name: Build and push Docker image
         id: push

openviking-0.2.4.dev33/.github/workflows/pr-review.yml

@@ -0,0 +1,41 @@
+name: PR Review (Doubao)
+
+on:
+  pull_request:
+    types: [opened, synchronize, reopened, ready_for_review]
+  issue_comment:
+
+jobs:
+  pr_review:
+    if: ${{ github.event.sender.type != 'Bot' }}
+    runs-on: ubuntu-latest
+    permissions:
+      issues: write
+      pull-requests: write
+      contents: write
+    steps:
+      - name: PR Agent
+        uses: qodo-ai/pr-agent@main
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          OPENAI_KEY: ${{ secrets.DOUBAO_API_KEY }}
+          OPENAI.API_BASE: "https://ark.cn-beijing.volces.com/api/v3"
+          config.model: "openai/doubao-seed-2-0-code-preview-260215"
+          config.custom_model_max_tokens: 131072
+          github_action_config.auto_review: "true"
+          github_action_config.auto_describe: "false"
+          github_action_config.auto_improve: "true"
+          pr_reviewer.extra_instructions: >
+            Classify each finding with a severity tag:
+            [Bug] = Must fix: behavioral regression, logic error, security vulnerability, data loss/corruption, resource leak, API contract violation.
+            [Suggestion] = Recommended: missing tests, dead code, inconsistency, poor observability, unclear intent, unrelated changes, performance issues (N+1 queries, redundant computation in loops, unnecessary large object copies, unbounded collection growth).
+
+            Review focus areas:
+            1. Control Flow - are all branches reachable? errors propagated correctly? unintended fall-through?
+            2. Data Flow - input validated at boundaries? implicit type coercion? null/empty handled?
+            3. Integration Points - API changes backward compatible? config defaults sensible? dependencies pinned?
+            4. Concurrency - shared state synchronized? execution order assumptions valid? cleanup in all paths?
+
+            Be specific: reference exact variable names, function calls, conditions.
+            When suggesting a fix, include a code block with the recommended change.
+            Match the language of the PR (Chinese PR = Chinese comments, English PR = English comments).

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/.github/workflows/release.yml

@@ -101,7 +101,7 @@ jobs:
       actions: read
     steps:
     - name: Download all the dists (Same Run)
-      uses: actions/download-artifact@v7
+      uses: actions/download-artifact@v8
       with:
         pattern: python-package-distributions-*
         path: dist/
@@ -136,7 +136,7 @@ jobs:
       actions: read
     steps:
     - name: Download all the dists (Same Run)
-      uses: actions/download-artifact@v7
+      uses: actions/download-artifact@v8
       with:
         pattern: python-package-distributions-*
         path: dist/
@@ -174,7 +174,7 @@ jobs:
           submodules: recursive
 
       - name: Log in to the Container registry
-        uses: docker/login-action@v3
+        uses: docker/login-action@v4
         with:
           registry: ghcr.io
           username: ${{ github.actor }}
@@ -187,10 +187,10 @@ jobs:
           images: ghcr.io/${{ github.repository }}
 
       - name: Set up QEMU
-        uses: docker/setup-qemu-action@v3
+        uses: docker/setup-qemu-action@v4
 
       - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v3
+        uses: docker/setup-buildx-action@v4
 
       - name: Build and push Docker image
         uses: docker/build-push-action@v6

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/.github/workflows/rust-cli.yml

@@ -123,7 +123,7 @@ jobs:
         fi
 
     - name: Upload artifacts
-      uses: actions/upload-artifact@v6
+      uses: actions/upload-artifact@v7
       with:
         name: ${{ matrix.artifact_name }}
         path: artifacts/*
@@ -144,7 +144,7 @@ jobs:
       run: echo "version=${GITHUB_REF#refs/tags/cli@}" >> $GITHUB_OUTPUT
 
     - name: Download all artifacts
-      uses: actions/download-artifact@v7
+      uses: actions/download-artifact@v8
       with:
         path: artifacts
 

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/CONTRIBUTING.md

@@ -14,7 +14,7 @@ Thank you for your interest in OpenViking! We welcome contributions of all kinds
 ### Prerequisites
 
 - **Python**: 3.10+
-- **Go**: 1.19.1+ (Required for building AGFS components from source)
+- **Go**: 1.22+ (Required for building AGFS components from source)
 - **C++ Compiler**: GCC 9+ or Clang 11+ (Required for building core extensions, must support C++17)
 - **CMake**: 3.12+
 
@@ -51,7 +51,7 @@ source .venv/bin/activate  # Linux/macOS
 
 #### Local Development & AGFS Compilation
 
-If you modify the **AGFS (Go)** code or **C++ extensions**, you need to re-compile and re-install them to make the changes take effect in your environment. Run the following command in the project root:
+OpenViking defaults to `binding-client` mode for AGFS, which requires a pre-built shared library. If you modify the **AGFS (Go)** code or **C++ extensions**, or if the pre-built library is not found, you need to re-compile and re-install them. Run the following command in the project root:
 
 ```bash
 uv pip install -e . --force-reinstall

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/CONTRIBUTING_CN.md

@@ -14,7 +14,7 @@
 ### 前置要求
 
 - **Python**: 3.10+
-- **Go**: 1.19+ (从源码构建 AGFS 组件需要)
+- **Go**: 1.22+ (从源码构建 AGFS 组件需要)
 - **C++ 编译器**: GCC 9+ 或 Clang 11+ (构建核心扩展需要，必须支持 C++17)
 - **CMake**: 3.12+
 
@@ -51,7 +51,7 @@ source .venv/bin/activate  # Linux/macOS
 
 #### 本地开发与 AGFS 编译
 
-如果你修改了 **AGFS (Go)** 代码或 **C++ 扩展**，你需要重新编译并安装它们，以使更改在本地环境中生效。在项目根目录下运行以下命令：
+OpenViking 默认使用 `binding-client` 模式，这需要一个预先构建的共享库。如果你修改了 **AGFS (Go)** 代码或 **C++ 扩展**，或者预编译库未找到，你需要重新编译并安装它们，以使更改在本地环境中生效。在项目根目录下运行以下命令：
 
 ```bash
 uv pip install -e . --force-reinstall

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/MANIFEST.in

@@ -11,6 +11,11 @@ include README.md
 include pyproject.toml
 include setup.py
 recursive-include openviking *.yaml
+
+# sdist should be source-only: never ship runtime binaries from working tree
+prune openviking/bin
+prune openviking/lib
+recursive-exclude openviking *.so *.dylib *.dll *.exe
 global-exclude *.py[cod]
 global-exclude __pycache__
 global-exclude .git*

{openviking-0.2.4.dev20/openviking.egg-info → openviking-0.2.4.dev33}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openviking
-Version: 0.2.4.dev20
+Version: 0.2.4.dev33
 Summary: An Agent-native context database
 Author-email: ByteDance <noreply@bytedance.com>
 License: Apache-2.0
@@ -141,6 +141,8 @@ With OpenViking, developers can build an Agent's brain just like managing local
 Before starting with OpenViking, please ensure your environment meets the following requirements:
 
 - **Python Version**: 3.10 or higher
+- **Go Version**: 1.22 or higher (Required for building AGFS components)
+- **C++ Compiler**: GCC 9+ or Clang 11+ (Required for building core extensions)
 - **Operating System**: Linux, macOS, Windows
 - **Network Connection**: A stable network connection is required (for downloading dependencies and accessing model services)
 

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/README.md

@@ -60,6 +60,8 @@ With OpenViking, developers can build an Agent's brain just like managing local
 Before starting with OpenViking, please ensure your environment meets the following requirements:
 
 - **Python Version**: 3.10 or higher
+- **Go Version**: 1.22 or higher (Required for building AGFS components)
+- **C++ Compiler**: GCC 9+ or Clang 11+ (Required for building core extensions)
 - **Operating System**: Linux, macOS, Windows
 - **Network Connection**: A stable network connection is required (for downloading dependencies and accessing model services)
 

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/README_CN.md

@@ -59,6 +59,8 @@
 在开始使用 OpenViking 之前，请确保您的环境满足以下要求：
 
 - **Python 版本**：3.10 或更高版本
+- **Go 版本**：1.22 或更高（从源码构建 AGFS 组件需要）
+- **C++ 编译器**：GCC 9+ 或 Clang 11+（构建核心扩展需要，必须支持 C++17）
 - **操作系统**：Linux、macOS、Windows
 - **网络连接**：需要稳定的网络连接（用于下载依赖和访问模型服务）
 

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/docs/design/multi-tenant-design.md

@@ -126,7 +126,7 @@ Request
 class APIKeyManager:
     """API Key 生命周期管理与解析"""
 
-    def __init__(self, root_key: Optional[str], agfs_url: str)
+    def __init__(self, root_key: str, agfs_client: AGFSClient)
     async def load()                                     # 加载所有 account 的 users.json 到内存
     async def save_account(account_id: str)              # 持久化指定 account 的 users.json
     def resolve(api_key: str) -> ResolvedIdentity        # Key → 身份 + 角色

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/docs/en/faq/faq.md

@@ -56,11 +56,25 @@ viking://
 ### What are the environment requirements?
 
 - **Python Version**: 3.10 or higher
+- **Build Tools** (if installing from source or on unsupported platforms): Go 1.19+, GCC 9+ or Clang 11+
 - **Required Dependencies**: Embedding model (Volcengine Doubao recommended)
 - **Optional Dependencies**:
   - VLM (Vision Language Model): For multimodal content processing and semantic extraction
   - Rerank model: For improved retrieval precision
 
+### What are `binding-client` and `http-client`? Which one should I choose?
+
+- **`binding-client` (Default)**: Runs AGFS logic directly within the Python process via CGO bindings. Advantages: extremely high performance, zero network latency; Disadvantages: requires a compiled AGFS shared library locally.
+- **`http-client`**: Communicates with a standalone `agfs-server` via HTTP. Advantages: decoupled deployment, no local Go compilation needed; Disadvantages: some network communication overhead.
+
+If your environment supports Go compilation or you've installed a Wheel package containing pre-compiled libraries, the default `binding-client` is recommended.
+
+### What should I do if I encounter "AGFS binding library not found"?
+
+This usually means the AGFS shared library is not pre-built in your environment. You can:
+1. **Re-compile and install**: Run `pip install -e . --force-reinstall` in the project root (requires Go environment).
+2. **Switch to HTTP mode**: Set `storage.agfs.mode = "http-client"` in your `ov.conf` and ensure an `agfs-server` is running.
+
 ### How do I install/upgrade OpenViking?
 
 ```bash

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/docs/zh/faq/faq.md

@@ -56,11 +56,25 @@ viking://
 ### 环境要求是什么？
 
 - **Python 版本**：3.10 或更高
+- **编译工具**（如果从源码安装或在不支持的平台上）：Go 1.19+, GCC 9+ 或 Clang 11+
 - **必需依赖**：Embedding 模型（推荐火山引擎 Doubao）
 - **可选依赖**：
   - VLM（视觉语言模型）：用于多模态内容处理和语义提取
   - Rerank 模型：用于提升检索精度
 
+### 什么是 `binding-client` 和 `http-client`？我该选哪个？
+
+- **`binding-client`（默认值）**：通过 CGO 绑定直接在 Python 进程内运行 AGFS 逻辑。优点是性能极高，无网络延迟；缺点是需要本地有编译好的 AGFS 共享库。
+- **`http-client`**：通过 HTTP 协议与独立的 `agfs-server` 通信。优点是部署解耦，不需要本地编译 Go 代码；缺点是有一定的网络通信开销。
+
+如果你的环境支持编译 Go 代码，或者安装了包含预编译库的 Wheel 包，推荐使用默认的 `binding-client`。
+
+### 遇到 "AGFS binding library not found" 错误怎么办？
+
+这通常是因为本地没有编译好的 AGFS 共享库。你可以：
+1. **重新编译安装**：在项目根目录运行 `pip install -e . --force-reinstall`（需要 Go 环境）。
+2. **切换到 HTTP 模式**：在 `ov.conf` 中设置 `storage.agfs.mode = "http-client"`，并确保有一个正在运行的 `agfs-server`。
+
 ### 如何安装 OpenViking？
 
 ```bash

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/openviking/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.2.4.dev20'
-__version_tuple__ = version_tuple = (0, 2, 4, 'dev20')
+__version__ = version = '0.2.4.dev33'
+__version_tuple__ = version_tuple = (0, 2, 4, 'dev33')
 
-__commit_id__ = commit_id = 'g847e86ac3'
+__commit_id__ = commit_id = 'g8c740fd56'

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/openviking/server/api_keys.py

@@ -52,9 +52,15 @@ class APIKeyManager:
     In-memory index for O(1) key lookup at runtime.
     """
 
-    def __init__(self, root_key: str, agfs_url: str):
+    def __init__(self, root_key: str, agfs_client: AGFSClient):
+        """Initialize APIKeyManager.
+
+        Args:
+            root_key: Global root API key for administrative access.
+            agfs_client: AGFS client for persistent storage of user keys.
+        """
         self._root_key = root_key
-        self._agfs = AGFSClient(agfs_url)
+        self._agfs = agfs_client
         self._accounts: Dict[str, AccountInfo] = {}
         self._user_keys: Dict[str, UserKeyEntry] = {}
 

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/openviking/server/app.py

@@ -68,7 +68,7 @@ def create_app(
         if config.root_api_key:
             api_key_manager = APIKeyManager(
                 root_key=config.root_api_key,
-                agfs_url=service._agfs_url,
+                agfs_client=service._agfs,
             )
             await api_key_manager.load()
             app.state.api_key_manager = api_key_manager
@@ -151,6 +151,7 @@ def create_app(
     # Configure Bot API if --with-bot is enabled
     if config.with_bot:
         import openviking.server.routers.bot as bot_module
+
         bot_module.set_bot_api_url(config.bot_api_url)
         logger.info(f"Bot API proxy enabled, forwarding to {config.bot_api_url}")
     else:

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/openviking/service/core.py

@@ -67,7 +67,6 @@ class OpenVikingService:
 
         # Infrastructure
         self._agfs_manager: Optional[AGFSManager] = None
-        self._agfs_url: Optional[str] = None
         self._agfs_client: Optional[Any] = None
         self._queue_manager: Optional[QueueManager] = None
         self._vikingdb_manager: Optional[VikingDBManager] = None
@@ -115,10 +114,8 @@ class OpenVikingService:
         if mode == "http-client":
             self._agfs_manager = AGFSManager(config=config.agfs)
             self._agfs_manager.start()
-            self._agfs_url = self._agfs_manager.url
-            config.agfs.url = self._agfs_url
-        else:
-            self._agfs_url = config.agfs.url
+            agfs_url = self._agfs_manager.url
+            config.agfs.url = agfs_url
 
         # Create AGFS client using utility
         self._agfs_client = create_agfs_client(config.agfs)
@@ -144,7 +141,12 @@ class OpenVikingService:
             self._queue_manager.setup_standard_queues(self._vikingdb_manager)
 
         # Initialize TransactionManager
-        self._transaction_manager = init_transaction_manager(agfs_config=config.agfs)
+        self._transaction_manager = init_transaction_manager(agfs=self._agfs_client)
+
+    @property
+    def _agfs(self) -> Any:
+        """Internal access to AGFS client for APIKeyManager."""
+        return self._agfs_client
 
     @property
     def viking_fs(self) -> Optional[VikingFS]:

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/openviking/storage/transaction/transaction_manager.py

@@ -334,14 +334,14 @@ class TransactionManager:
 
 
 def init_transaction_manager(
-    agfs_config: Any,
+    agfs: AGFSClient,
     tx_timeout: int = 3600,
     max_parallel_locks: int = 8,
 ) -> TransactionManager:
     """Initialize transaction manager singleton.
 
     Args:
-        agfs_config: AGFS configuration (url, timeout, etc.)
+        agfs: AGFS client instance
         tx_timeout: Transaction timeout in seconds (default: 3600)
         max_parallel_locks: Maximum number of parallel lock operations (default: 8)
 
@@ -355,16 +355,9 @@ def init_transaction_manager(
             logger.debug("TransactionManager already initialized")
             return _transaction_manager
 
-        # Get AGFS URL from config
-        agfs_url = getattr(agfs_config, "url", "http://localhost:8080")
-        agfs_timeout = getattr(agfs_config, "timeout", 10)
-
-        # Create AGFS client
-        agfs_client = AGFSClient(api_base_url=agfs_url, timeout=agfs_timeout)
-
         # Create transaction manager
         _transaction_manager = TransactionManager(
-            agfs_client=agfs_client,
+            agfs_client=agfs,
             timeout=tx_timeout,
             max_parallel_locks=max_parallel_locks,
         )

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/openviking/storage/vectordb/collection/collection.py

@@ -1,12 +1,23 @@
 # Copyright (c) 2026 Beijing Volcano Engine Technology Co., Ltd.
 # SPDX-License-Identifier: Apache-2.0
+import importlib
 from abc import ABC, abstractmethod
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, Type
 
 from openviking.storage.vectordb.collection.result import AggregateResult, SearchResult
 from openviking.storage.vectordb.index.index import IIndex
 
 
+def load_collection_class(class_path: str) -> Type["ICollection"]:
+    """Load collection class from string path"""
+    try:
+        module_name, class_name = class_path.rsplit(".", 1)
+        module = importlib.import_module(module_name)
+        return getattr(module, class_name)
+    except (ImportError, AttributeError) as e:
+        raise ImportError(f"Could not load collection class {class_path}: {e}")
+
+
 class ICollection(ABC):
     def __init__(self):
         pass

{openviking-0.2.4.dev20 → openviking-0.2.4.dev33}/openviking/storage/vectordb/project/vikingdb_project.py

@@ -2,12 +2,14 @@
 # SPDX-License-Identifier: Apache-2.0
 from typing import Any, Dict, List, Optional
 
-from openviking.storage.vectordb.collection.collection import Collection
+from openviking.storage.vectordb.collection.collection import (
+    Collection,
+    load_collection_class,
+)
 from openviking.storage.vectordb.collection.vikingdb_clients import (
     VIKINGDB_APIS,
     VikingDBClient,
 )
-from openviking.storage.vectordb.collection.vikingdb_collection import VikingDBCollection
 from openviking_cli.utils.logger import default_logger as logger
 
 
@@ -22,6 +24,8 @@ def get_or_create_vikingdb_project(
         config: Configuration dict with keys:
             - Host: VikingDB service host
             - Headers: Custom headers for authentication/context
+            - CollectionClass: Class path for collection implementation
+            - CollectionArgs: Optional dictionary of arguments to pass to collection constructor
 
     Returns:
         VikingDBProject instance
@@ -31,11 +35,23 @@ def get_or_create_vikingdb_project(
 
     host = config.get("Host")
     headers = config.get("Headers")
+    collection_class_path = config.get(
+        "CollectionClass",
+        "openviking.storage.vectordb.collection.vikingdb_collection.VikingDBCollection",
+    )
+    # Extract any other arguments that might be needed for collection initialization
+    collection_args = config.get("CollectionArgs", {})
 
     if not host:
         raise ValueError("config must contain 'Host'")
 
-    return VikingDBProject(host=host, headers=headers, project_name=project_name)
+    return VikingDBProject(
+        host=host,
+        headers=headers,
+        project_name=project_name,
+        collection_class_path=collection_class_path,
+        collection_args=collection_args,
+    )
 
 
 class VikingDBProject:
@@ -45,7 +61,12 @@ class VikingDBProject:
     """
 
     def __init__(
-        self, host: str, headers: Optional[Dict[str, str]] = None, project_name: str = "default"
+        self,
+        host: str,
+        headers: Optional[Dict[str, str]] = None,
+        project_name: str = "default",
+        collection_class_path: str = "openviking.storage.vectordb.collection.vikingdb_collection.VikingDBCollection",
+        collection_args: Optional[Dict[str, Any]] = None,
     ):
         """
         Initialize VikingDB project.
@@ -54,12 +75,19 @@ class VikingDBProject:
             host: VikingDB service host
             headers: Custom headers for requests
             project_name: Project name
+            collection_class_path: Python path to the collection class
+            collection_args: Optional dictionary of arguments to pass to collection constructor
         """
         self.host = host
         self.headers = headers
         self.project_name = project_name
+        self.collection_class_path = collection_class_path
+        self.CollectionClass = load_collection_class(self.collection_class_path)
+        self.collection_args = collection_args or {}
 
-        logger.info(f"Initialized VikingDB project: {project_name} with host {host}")
+        logger.info(
+            f"Initialized VikingDB project: {project_name} with host {host} and collection class {collection_class_path}"
+        )
 
     def close(self):
         """Close project"""
@@ -87,9 +115,17 @@ class VikingDBProject:
             meta_data = result.get("Result", {})
             if not meta_data:
                 return None
-            vikingdb_collection = VikingDBCollection(
-                host=self.host, headers=self.headers, meta_data=meta_data
-            )
+            # Prepare arguments for collection constructor
+            # Default arguments
+            kwargs = {
+                "host": self.host,
+                "headers": self.headers,
+                "meta_data": meta_data,
+            }
+            # Update with user-provided arguments (can override defaults if needed, though usually additive)
+            kwargs.update(self.collection_args)
+            
+            vikingdb_collection = self.CollectionClass(**kwargs)
             return Collection(vikingdb_collection)
         except Exception:
             return None
@@ -118,12 +154,24 @@ class VikingDBProject:
     def get_collections(self) -> Dict[str, Collection]:
         """Get all collections from server"""
         colls = self._get_collections()
-        return {
-            c["CollectionName"]: Collection(
-                VikingDBCollection(host=self.host, headers=self.headers, meta_data=c)
-            )
-            for c in colls
+        
+        # Prepare base arguments
+        base_kwargs = {
+            "host": self.host,
+            "headers": self.headers,
         }
+        
+        collections = {}
+        for c in colls:
+            kwargs = base_kwargs.copy()
+            kwargs["meta_data"] = c
+            kwargs.update(self.collection_args)
+            
+            collections[c["CollectionName"]] = Collection(
+                self.CollectionClass(**kwargs)
+            )
+            
+        return collections
 
     def create_collection(self, collection_name: str, meta_data: Dict[str, Any]) -> Collection:
         """collection should be pre-created"""