commit 0c00abec83397b854556ba8193875344b6472e6a · pyrox.dev/nixpkgs

+34 -36
nixos/modules/services/search/qdrant.nix
···

       4
       4
        
         pkgs,

     

       5
       5
        
         ...

     

       6
       6
        
       }:

     

       7
       7
       -
       

     

       8
       8
       -
       with lib;

     

       9
       7
        
       let

     

       10
       8
        
       

     

       11
       9
        
         cfg = config.services.qdrant;

     
···

       17
       15
        
       

     

       18
       16
        
         options = {

     

       19
       17
        
           services.qdrant = {

     

       20
       20
       -
             enable = mkEnableOption "Vector Search Engine for the next generation of AI applications";

     

       18
       18
       +
             enable = lib.mkEnableOption "Vector Search Engine for the next generation of AI applications";

     

       21
       19
        
       

     

       22
       22
       -
             settings = mkOption {

     

       20
       20
       +
             settings = lib.mkOption {

     

       23
       21
        
               description = ''

     

       24
       22
        
                 Configuration for Qdrant

     

       25
       23
        
                 Refer to <https://github.com/qdrant/qdrant/blob/master/config/config.yaml> for details on supported values.

     
···

       43
       41
        
                 telemetry_disabled = true;

     

       44
       42
        
               };

     

       45
       43
        
       

     

       46
       46
       -
               defaultText = literalExpression ''

     

       44
       44
       +
               defaultText = lib.literalExpression ''

     

       47
       45
        
                 {

     

       48
       46
        
                   storage = {

     

       49
       47
        
                     storage_path = "/var/lib/qdrant/storage";

     
···

       64
       62
        
           };

     

       65
       63
        
         };

     

       66
       64
        
       

     

       67
       67
       -
         config = mkIf cfg.enable {

     

       65
       65
       +
         config = lib.mkIf cfg.enable {

     

       68
       66
        
           services.qdrant.settings = {

     

       69
       69
       -
             service.static_content_dir = mkDefault pkgs.qdrant-web-ui;

     

       70
       70
       -
             storage.storage_path = mkDefault "/var/lib/qdrant/storage";

     

       71
       71
       -
             storage.snapshots_path = mkDefault "/var/lib/qdrant/snapshots";

     

       67
       67
       +
             service.static_content_dir = lib.mkDefault pkgs.qdrant-web-ui;

     

       68
       68
       +
             storage.storage_path = lib.mkDefault "/var/lib/qdrant/storage";

     

       69
       69
       +
             storage.snapshots_path = lib.mkDefault "/var/lib/qdrant/snapshots";

     

       72
       70
        
             # The following default values are the same as in the default config,

     

       73
       71
        
             # they are just written here for convenience.

     

       74
       74
       -
             storage.on_disk_payload = mkDefault true;

     

       75
       75
       -
             storage.wal.wal_capacity_mb = mkDefault 32;

     

       76
       76
       -
             storage.wal.wal_segments_ahead = mkDefault 0;

     

       77
       77
       -
             storage.performance.max_search_threads = mkDefault 0;

     

       78
       78
       -
             storage.performance.max_optimization_threads = mkDefault 1;

     

       79
       79
       -
             storage.optimizers.deleted_threshold = mkDefault 0.2;

     

       80
       80
       -
             storage.optimizers.vacuum_min_vector_number = mkDefault 1000;

     

       81
       81
       -
             storage.optimizers.default_segment_number = mkDefault 0;

     

       82
       82
       -
             storage.optimizers.max_segment_size_kb = mkDefault null;

     

       83
       83
       -
             storage.optimizers.memmap_threshold_kb = mkDefault null;

     

       84
       84
       -
             storage.optimizers.indexing_threshold_kb = mkDefault 20000;

     

       85
       85
       -
             storage.optimizers.flush_interval_sec = mkDefault 5;

     

       86
       86
       -
             storage.optimizers.max_optimization_threads = mkDefault 1;

     

       87
       87
       -
             storage.hnsw_index.m = mkDefault 16;

     

       88
       88
       -
             storage.hnsw_index.ef_construct = mkDefault 100;

     

       89
       89
       -
             storage.hnsw_index.full_scan_threshold_kb = mkDefault 10000;

     

       90
       90
       -
             storage.hnsw_index.max_indexing_threads = mkDefault 0;

     

       91
       91
       -
             storage.hnsw_index.on_disk = mkDefault false;

     

       92
       92
       -
             storage.hnsw_index.payload_m = mkDefault null;

     

       93
       93
       -
             service.max_request_size_mb = mkDefault 32;

     

       94
       94
       -
             service.max_workers = mkDefault 0;

     

       95
       95
       -
             service.http_port = mkDefault 6333;

     

       96
       96
       -
             service.grpc_port = mkDefault 6334;

     

       97
       97
       -
             service.enable_cors = mkDefault true;

     

       98
       98
       -
             cluster.enabled = mkDefault false;

     

       72
       72
       +
             storage.on_disk_payload = lib.mkDefault true;

     

       73
       73
       +
             storage.wal.wal_capacity_mb = lib.mkDefault 32;

     

       74
       74
       +
             storage.wal.wal_segments_ahead = lib.mkDefault 0;

     

       75
       75
       +
             storage.performance.max_search_threads = lib.mkDefault 0;

     

       76
       76
       +
             storage.performance.max_optimization_threads = lib.mkDefault 1;

     

       77
       77
       +
             storage.optimizers.deleted_threshold = lib.mkDefault 0.2;

     

       78
       78
       +
             storage.optimizers.vacuum_min_vector_number = lib.mkDefault 1000;

     

       79
       79
       +
             storage.optimizers.default_segment_number = lib.mkDefault 0;

     

       80
       80
       +
             storage.optimizers.max_segment_size_kb = lib.mkDefault null;

     

       81
       81
       +
             storage.optimizers.memmap_threshold_kb = lib.mkDefault null;

     

       82
       82
       +
             storage.optimizers.indexing_threshold_kb = lib.mkDefault 20000;

     

       83
       83
       +
             storage.optimizers.flush_interval_sec = lib.mkDefault 5;

     

       84
       84
       +
             storage.optimizers.max_optimization_threads = lib.mkDefault 1;

     

       85
       85
       +
             storage.hnsw_index.m = lib.mkDefault 16;

     

       86
       86
       +
             storage.hnsw_index.ef_construct = lib.mkDefault 100;

     

       87
       87
       +
             storage.hnsw_index.full_scan_threshold_kb = lib.mkDefault 10000;

     

       88
       88
       +
             storage.hnsw_index.max_indexing_threads = lib.mkDefault 0;

     

       89
       89
       +
             storage.hnsw_index.on_disk = lib.mkDefault false;

     

       90
       90
       +
             storage.hnsw_index.payload_m = lib.mkDefault null;

     

       91
       91
       +
             service.max_request_size_mb = lib.mkDefault 32;

     

       92
       92
       +
             service.max_workers = lib.mkDefault 0;

     

       93
       93
       +
             service.http_port = lib.mkDefault 6333;

     

       94
       94
       +
             service.grpc_port = lib.mkDefault 6334;

     

       95
       95
       +
             service.enable_cors = lib.mkDefault true;

     

       96
       96
       +
             cluster.enabled = lib.mkDefault false;

     

       99
       97
        
             # the following have been altered for security

     

       100
       100
       -
             service.host = mkDefault "127.0.0.1";

     

       101
       101
       -
             telemetry_disabled = mkDefault true;

     

       98
       98
       +
             service.host = lib.mkDefault "127.0.0.1";

     

       99
       99
       +
             telemetry_disabled = lib.mkDefault true;

     

       102
       100
        
           };

     

       103
       101
        
       

     

       104
       102
        
           systemd.services.qdrant = {