|
| 1 | +from datetime import timedelta |
1 | 2 | from typing import List, Union |
2 | 3 |
|
3 | 4 | import pytest |
|
11 | 12 | Vectorizers, |
12 | 13 | _CollectionConfigCreate, |
13 | 14 | _GenerativeProvider, |
| 15 | + _ObjectTTLConfig, |
14 | 16 | _RerankerProvider, |
15 | 17 | _VectorizerConfigCreate, |
16 | 18 | ) |
|
19 | 21 | Multi2VecField, |
20 | 22 | VectorDistances, |
21 | 23 | ) |
22 | | - |
23 | 24 | from weaviate.collections.classes.config_vectors import _VectorConfigCreate |
24 | 25 |
|
25 | 26 | DEFAULTS = { |
@@ -2118,6 +2119,24 @@ def test_config_with_named_vectors( |
2118 | 2119 | } |
2119 | 2120 | }, |
2120 | 2121 | ), |
| 2122 | + ( |
| 2123 | + [Configure.MultiVectors.multi2vec_weaviate(name="test", image_field="prop")], |
| 2124 | + { |
| 2125 | + "test": { |
| 2126 | + "vectorizer": { |
| 2127 | + "multi2multivec-weaviate": { |
| 2128 | + "imageFields": ["prop"], |
| 2129 | + } |
| 2130 | + }, |
| 2131 | + "vectorIndexConfig": { |
| 2132 | + "multivector": { |
| 2133 | + "enabled": True, |
| 2134 | + }, |
| 2135 | + }, |
| 2136 | + "vectorIndexType": "hnsw", |
| 2137 | + } |
| 2138 | + }, |
| 2139 | + ), |
2121 | 2140 | ( |
2122 | 2141 | [Configure.Vectors.text2vec_gpt4all(name="test", source_properties=["prop"])], |
2123 | 2142 | { |
@@ -2502,3 +2521,87 @@ def test_config_with_vectors(vector_config: List[_VectorConfigCreate], expected: |
2502 | 2521 | "class": "Test", |
2503 | 2522 | "vectorConfig": expected, |
2504 | 2523 | } |
| 2524 | + |
| 2525 | + |
| 2526 | +TEST_OBJECT_TTL_CONFIG_TO_DICT_PARAMETERS = [ |
| 2527 | + # delete_by_creation_time |
| 2528 | + ( |
| 2529 | + _ObjectTTLConfig( |
| 2530 | + enabled=True, |
| 2531 | + time_to_live=timedelta(hours=24), |
| 2532 | + filter_expired_objects=True, |
| 2533 | + delete_on="creationTime", |
| 2534 | + ), |
| 2535 | + { |
| 2536 | + "enabled": True, |
| 2537 | + "timeToLive": 86400, |
| 2538 | + "filterExpiredObjects": True, |
| 2539 | + "deleteOn": "creationTime", |
| 2540 | + }, |
| 2541 | + ), |
| 2542 | + # delete_by_update_time |
| 2543 | + ( |
| 2544 | + _ObjectTTLConfig( |
| 2545 | + enabled=True, |
| 2546 | + time_to_live=timedelta(days=7), |
| 2547 | + filter_expired_objects=False, |
| 2548 | + delete_on="updateTime", |
| 2549 | + ), |
| 2550 | + { |
| 2551 | + "enabled": True, |
| 2552 | + "timeToLive": 604800, |
| 2553 | + "filterExpiredObjects": False, |
| 2554 | + "deleteOn": "updateTime", |
| 2555 | + }, |
| 2556 | + ), |
| 2557 | + # delete_by_date_property |
| 2558 | + ( |
| 2559 | + _ObjectTTLConfig( |
| 2560 | + enabled=True, |
| 2561 | + time_to_live=timedelta(hours=1, minutes=30), |
| 2562 | + filter_expired_objects=True, |
| 2563 | + delete_on="releaseDate", |
| 2564 | + ), |
| 2565 | + { |
| 2566 | + "enabled": True, |
| 2567 | + "timeToLive": 5400, |
| 2568 | + "filterExpiredObjects": True, |
| 2569 | + "deleteOn": "releaseDate", |
| 2570 | + }, |
| 2571 | + ), |
| 2572 | + # None time_to_live |
| 2573 | + ( |
| 2574 | + _ObjectTTLConfig( |
| 2575 | + enabled=True, |
| 2576 | + time_to_live=None, |
| 2577 | + filter_expired_objects=False, |
| 2578 | + delete_on="creationTime", |
| 2579 | + ), |
| 2580 | + { |
| 2581 | + "enabled": True, |
| 2582 | + "filterExpiredObjects": False, |
| 2583 | + "deleteOn": "creationTime", |
| 2584 | + }, |
| 2585 | + ), |
| 2586 | + # negative offset (delete_by_date_property with offset before date) |
| 2587 | + ( |
| 2588 | + _ObjectTTLConfig( |
| 2589 | + enabled=True, |
| 2590 | + time_to_live=timedelta(seconds=-3600), |
| 2591 | + filter_expired_objects=True, |
| 2592 | + delete_on="eventDate", |
| 2593 | + ), |
| 2594 | + { |
| 2595 | + "enabled": True, |
| 2596 | + "timeToLive": -3600, |
| 2597 | + "filterExpiredObjects": True, |
| 2598 | + "deleteOn": "eventDate", |
| 2599 | + }, |
| 2600 | + ), |
| 2601 | +] |
| 2602 | + |
| 2603 | + |
| 2604 | +@pytest.mark.parametrize("ttl_config,expected", TEST_OBJECT_TTL_CONFIG_TO_DICT_PARAMETERS) |
| 2605 | +def test_object_ttl_config_to_dict(ttl_config: _ObjectTTLConfig, expected: dict) -> None: |
| 2606 | + """Test that _ObjectTTLConfig.to_dict() properly converts timedelta to seconds.""" |
| 2607 | + assert ttl_config.to_dict() == expected |
0 commit comments