context-dedup 0.1.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,375 @@
1
+ Mozilla Public License Version 2.0
2
+ ==================================
3
+
4
+ 1. Definitions
5
+ --------------
6
+
7
+ 1.1. "Contributor"
8
+ means each individual or legal entity that creates, contributes to
9
+ the creation of, or owns Covered Software.
10
+
11
+ 1.2. "Contributor Version"
12
+ means the combination of the Contributions of others (if any) used
13
+ by a Contributor and that particular Contributor's Contribution.
14
+
15
+ 1.3. "Contribution"
16
+ means Covered Software of a particular Contributor.
17
+
18
+ 1.4. "Covered Software"
19
+ means Source Code Form to which the initial Contributor has attached
20
+ the notice in Exhibit A, the Executable Form of such Source Code
21
+ Form, and Modifications of such Source Code Form, in each case
22
+ including portions thereof.
23
+
24
+ 1.5. "Incompatible With Secondary Licenses"
25
+ means
26
+
27
+ (a) that the initial Contributor has attached the notice described
28
+ in Exhibit B to the Covered Software; or
29
+
30
+ (b) that the Covered Software was made available under the terms of
31
+ version 1.1 or earlier of the License, but not also under the
32
+ terms of a Secondary License.
33
+
34
+ 1.6. "Executable Form"
35
+ means any form of the work other than Source Code Form.
36
+
37
+ 1.7. "Larger Work"
38
+ means a work that combines Covered Software with other material, in
39
+ a separate file or files, that is not Covered Software.
40
+
41
+ 1.8. "License"
42
+ means this document.
43
+
44
+ 1.9. "Licensable"
45
+ means having the right to grant, to the maximum extent possible,
46
+ whether at the time of the initial grant or subsequently, any and
47
+ all of the rights conveyed by this License.
48
+
49
+ 1.10. "Modifications"
50
+ means any of the following:
51
+
52
+ (a) any file in Source Code Form that results from an addition to,
53
+ deletion from, or modification of the contents of Covered
54
+ Software; or
55
+
56
+ (b) any new file in Source Code Form that contains any Covered
57
+ Software.
58
+
59
+ 1.11. "Patent Claims" of a Contributor
60
+ means any patent claim(s), including without limitation, method,
61
+ process, and apparatus claims, in any patent Licensable by such
62
+ Contributor that would be infringed, but for the grant of the
63
+ License, by the making, using, selling, offering for sale, having
64
+ made, import, or transfer of either its Contributions or its
65
+ Contributor Version.
66
+
67
+ 1.12. "Secondary License"
68
+ means either the GNU General Public License, Version 2.0, the GNU
69
+ Lesser General Public License, Version 2.1, the GNU Affero General
70
+ Public License, Version 3.0, or any later versions of those
71
+ licenses.
72
+
73
+ 1.13. "Source Code Form"
74
+ means the form of the work preferred for making modifications.
75
+
76
+ 1.14. "You" (or "Your")
77
+ means an individual or a legal entity exercising rights under this
78
+ License. For legal entities, "You" includes any entity that
79
+ controls, is controlled by, or is under common control with You. For
80
+ purposes of this definition, "control" means (a) the power, direct
81
+ or indirect, to cause the direction or management of such entity,
82
+ whether by contract or otherwise, or (b) ownership of more than
83
+ fifty percent (50%) of the outstanding shares or beneficial
84
+ ownership of such entity.
85
+
86
+ 2. License Grants and Conditions
87
+ --------------------------------
88
+
89
+ 2.1. Grants
90
+
91
+ Each Contributor hereby grants You a world-wide, royalty-free,
92
+ non-exclusive license:
93
+
94
+ (a) under intellectual property rights (other than patent or trademark)
95
+ Licensable by such Contributor to use, reproduce, make available,
96
+ modify, display, perform, distribute, and otherwise exploit its
97
+ Contributions, either on an unmodified basis, with Modifications, or
98
+ as part of a Larger Work; and
99
+
100
+ (b) under Patent Claims of such Contributor to make, use, sell, offer
101
+ for sale, have made, import, and otherwise transfer either its
102
+ Contributions or its Contributor Version.
103
+
104
+ 2.2. Effective Date
105
+
106
+ The licenses granted in Section 2.1 with respect to any Contribution
107
+ become effective for each Contribution on the date the Contributor first
108
+ distributes such Contribution.
109
+
110
+ 2.3. Limitations on Grant Scope
111
+
112
+ The licenses granted in this Section 2 are the only rights granted under
113
+ this License. No additional rights or licenses will be implied from the
114
+ distribution or licensing of Covered Software under this License.
115
+ Notwithstanding Section 2.1(b) above, no patent license is granted by a
116
+ Contributor:
117
+
118
+ (a) for any code that a Contributor has removed from Covered Software;
119
+ or
120
+
121
+ (b) for infringements caused by: (i) Your and any other third party's
122
+ modifications of Covered Software, or (ii) the combination of its
123
+ Contributions with other software (except as part of its Contributor
124
+ Version); or
125
+
126
+ (c) under Patent Claims infringed by Covered Software in the absence of
127
+ its Contributions.
128
+
129
+ This License does not grant any rights in the trademarks, service marks,
130
+ or logos of any Contributor (except as may be necessary to comply with
131
+ the notice requirements in Section 3.4).
132
+
133
+ 2.4. Subsequent Licenses
134
+
135
+ No Contributor makes additional grants as a result of Your choice to
136
+ distribute the Covered Software under a subsequent version of this
137
+ License (see Section 10.2) or under the terms of a Secondary License (if
138
+ permitted under the terms of Section 3.3).
139
+
140
+ 2.5. Representation
141
+
142
+ Each Contributor represents that the Contributor believes its
143
+ Contributions are its original creation(s) or it has sufficient rights
144
+ to grant the rights to its Contributions conveyed by this License.
145
+
146
+ 2.6. Fair Use
147
+
148
+ This License is not intended to limit any rights You have under
149
+ applicable copyright doctrines of fair use, fair dealing, or other
150
+ equivalents.
151
+
152
+ 2.7. Conditions
153
+
154
+ Sections 3.1, 3.2, 3.3, and 3.4 are conditions of the licenses granted
155
+ in Section 2.1.
156
+
157
+ 3. Responsibilities
158
+ -------------------
159
+
160
+ 3.1. Distribution of Source Form
161
+
162
+ All distribution of Covered Software in Source Code Form, including any
163
+ Modifications that You create or to which You contribute, must be under
164
+ the terms of this License. You must inform recipients that the Source
165
+ Code Form of the Covered Software is governed by the terms of this
166
+ License, and how they can obtain a copy of this License. You may not
167
+ attempt to alter or restrict the recipients' rights in the Source Code
168
+ Form.
169
+
170
+ 3.2. Distribution of Executable Form
171
+
172
+ If You distribute Covered Software in Executable Form then:
173
+
174
+ (a) such Covered Software must also be made available in Source Code
175
+ Form, as described in Section 3.1, and You must inform recipients of
176
+ the Executable Form how they can obtain a copy of such Source Code
177
+ Form by reasonable means in a timely manner, at a charge no more
178
+ than the cost of distribution to the recipient; and
179
+
180
+ (b) You may distribute such Executable Form under the terms of this
181
+ License, or sublicense it under different terms, provided that the
182
+ license for the Executable Form does not attempt to limit or alter
183
+ the recipients' rights in the Source Code Form under this License.
184
+
185
+ 3.3. Distribution of a Larger Work
186
+
187
+ You may create and distribute a Larger Work under terms of Your choice,
188
+ provided that You also comply with the requirements of this License for
189
+ the Covered Software. If the Larger Work is a combination of Covered
190
+ Software with a work governed by one or more Secondary Licenses, and the
191
+ Covered Software is not Incompatible With Secondary Licenses, this
192
+ License permits You to additionally distribute such Covered Software
193
+ under the terms of such Secondary License(s), so that the recipient of
194
+ the Larger Work may, at their option, further distribute the Covered
195
+ Software under the terms of either this License or such Secondary
196
+ License(s).
197
+
198
+ 3.4. Notices
199
+
200
+ You may not remove or alter the substance of any license notices
201
+ (including copyright notices, patent notices, disclaimers of warranty,
202
+ or limitations of liability) contained within the Source Code Form of
203
+ the Covered Software, except that You may alter any license notices to
204
+ the extent required to remedy known factual inaccuracies.
205
+
206
+ 3.5. Application of Additional Terms
207
+
208
+ You may choose to offer, and to charge a fee for, warranty, support,
209
+ indemnity or liability obligations to one or more recipients of Covered
210
+ Software. However, You may do so only on Your own behalf, and not on
211
+ behalf of any Contributor. You must make it absolutely clear that any
212
+ such warranty, support, indemnity, or liability obligation is offered by
213
+ You alone, and You hereby agree to indemnify every Contributor for any
214
+ liability incurred by such Contributor as a result of warranty, support,
215
+ indemnity or liability terms You offer. You may include additional
216
+ disclaimers of warranty and limitations of liability specific to any
217
+ jurisdiction.
218
+
219
+ 4. Inability to Comply Due to Statute or Regulation
220
+ ---------------------------------------------------
221
+
222
+ If it is impossible for You to comply with any of the terms of this
223
+ License with respect to some or all of the Covered Software due to
224
+ statute, judicial order, or regulation then You must: (a) comply with
225
+ the terms of this License to the maximum extent possible; and (b)
226
+ describe the limitations and the code they affect. Such description must
227
+ be placed in a text file included with all distributions of the Covered
228
+ Software under this License. Except to the extent prohibited by statute
229
+ or regulation, such description must be sufficiently detailed for a
230
+ recipient of ordinary skill to be able to understand it.
231
+
232
+ 5. Termination
233
+ --------------
234
+
235
+ 5.1. The rights granted under this License will terminate automatically
236
+ if You fail to comply with any of its terms. However, if You become
237
+ compliant, then the rights granted under this License from a particular
238
+ Contributor are reinstated (a) provisionally, unless and until such
239
+ Contributor explicitly and finally terminates Your grants, and (b) on an
240
+ ongoing basis, if such Contributor fails to notify You of the
241
+ non-compliance by some reasonable means prior to 60 days after You have
242
+ come back into compliance. Moreover, Your grants from a particular
243
+ Contributor are reinstated on an ongoing basis if such Contributor
244
+ notifies You of the non-compliance by some reasonable means, this is the
245
+ first time You have received notice of non-compliance with this License
246
+ from such Contributor, and You become compliant prior to 30 days after
247
+ Your receipt of the notice.
248
+
249
+ 5.2. If You initiate litigation against any entity by asserting a patent
250
+ infringement claim (excluding declaratory judgment actions,
251
+ counter-claims, and cross-claims) alleging that a Contributor Version
252
+ directly or indirectly infringes any patent, then the rights granted to
253
+ You by any and all Contributors for the Covered Software under Section
254
+ 2.1 of this License shall terminate.
255
+
256
+ 5.3. In the event of termination under Sections 5.1 or 5.2 above, all
257
+ end user license agreements (excluding distributors and resellers) which
258
+ have been validly granted by You or Your distributors under this License
259
+ prior to termination shall survive termination.
260
+
261
+ ************************************************************************
262
+ * *
263
+ * 6. Disclaimer of Warranty *
264
+ * ------------------------- *
265
+ * *
266
+ * Covered Software is provided under this License on an "as is" *
267
+ * basis, without warranty of any kind, either expressed, implied, or *
268
+ * statutory, including, without limitation, warranties that the *
269
+ * Covered Software is free of defects, merchantable, fit for a *
270
+ * particular purpose or non-infringing. The entire risk as to the *
271
+ * quality and performance of the Covered Software is with You. *
272
+ * Should any Covered Software prove defective in any respect, You *
273
+ * (not any Contributor) assume the cost of any necessary servicing, *
274
+ * repair, or correction. This disclaimer of warranty constitutes an *
275
+ * essential part of this License. No use of any Covered Software is *
276
+ * authorized under this License except under this disclaimer. *
277
+ * *
278
+ ************************************************************************
279
+
280
+ ************************************************************************
281
+ * *
282
+ * 7. Limitation of Liability *
283
+ * -------------------------- *
284
+ * *
285
+ * Under no circumstances and under no legal theory, whether tort *
286
+ * (including negligence), contract, or otherwise, shall any *
287
+ * Contributor, or anyone who distributes Covered Software as *
288
+ * permitted above, be liable to You for any direct, indirect, *
289
+ * special, incidental, or consequential damages of any character *
290
+ * including, without limitation, damages for lost profits, loss of *
291
+ * goodwill, work stoppage, computer failure or malfunction, or any *
292
+ * and all other commercial damages or losses, even if such party *
293
+ * shall have been informed of the possibility of such damages. This *
294
+ * limitation of liability shall not apply to liability for death or *
295
+ * personal injury resulting from such party's negligence to the *
296
+ * extent applicable law prohibits such limitation. Some *
297
+ * jurisdictions do not allow the exclusion or limitation of *
298
+ * incidental or consequential damages, so this exclusion and *
299
+ * limitation may not apply to You. *
300
+ * *
301
+ ************************************************************************
302
+
303
+ 8. Litigation
304
+ -------------
305
+
306
+ Any litigation relating to this License may be brought only in the
307
+ courts of a jurisdiction where the defendant maintains its principal
308
+ place of business and such litigation shall be governed by laws of that
309
+ jurisdiction, without reference to its conflict-of-law provisions.
310
+ Nothing in this Section shall prevent a party's ability to bring
311
+ cross-claims or counter-claims.
312
+
313
+ 9. Miscellaneous
314
+ ----------------
315
+
316
+ This License represents the complete agreement concerning the subject
317
+ matter hereof. If any provision of this License is held to be
318
+ unenforceable, such provision shall be reformed only to the extent
319
+ necessary to make it enforceable. Any law or regulation which provides
320
+ that the language of a contract shall be construed against the drafter
321
+ shall not be used to construe this License against a Contributor.
322
+
323
+ 10. Versions of the License
324
+ ---------------------------
325
+
326
+ 10.1. New Versions
327
+
328
+ Mozilla Foundation is the license steward. Except as provided in Section
329
+ 10.3, no one other than the license steward has the right to modify or
330
+ publish new versions of this License. Each version will be given a
331
+ distinguishing version number.
332
+
333
+ 10.2. Effect of New Versions
334
+
335
+ You may distribute the Covered Software under the terms of the version
336
+ of the License under which You originally received the Covered Software,
337
+ or under the terms of any subsequent version published by the license
338
+ steward.
339
+
340
+ 10.3. Modified Versions
341
+
342
+ If you create software not governed by this License, and you want to
343
+ create a new license for such software, you may create and use a
344
+ modified version of this License if you rename the license and remove
345
+ any references to the name of the license steward (except to note that
346
+ such modified license differs from this License).
347
+
348
+ 10.4. Distributing Source Code Form that is Incompatible With Secondary
349
+ Licenses
350
+
351
+ If You choose to distribute Source Code Form that is Incompatible With
352
+ Secondary Licenses under the terms of this version of the License, the
353
+ notice described in Exhibit B of this License must be attached.
354
+
355
+ Exhibit A - Source Code Form License Notice
356
+ -------------------------------------------
357
+
358
+ This Source Code Form is subject to the terms of the Mozilla Public
359
+ License, v. 2.0. If a copy of the MPL was not distributed with this
360
+ file, You can obtain one at http://mozilla.org/MPL/2.0/.
361
+
362
+ If it is not possible or desirable to put the notice in a particular
363
+ file, then You may include the notice in a location (such as a LICENSE
364
+ file in a relevant directory) where a recipient would be likely to look
365
+ for such a notice.
366
+
367
+ You may add additional accurate notices of copyright ownership.
368
+
369
+ Exhibit B - "Incompatible With Secondary Licenses" Notice
370
+ ---------------------------------------------------------
371
+
372
+ This Source Code Form is "Incompatible With Secondary Licenses", as
373
+ defined by the Mozilla Public License, v. 2.0.
374
+
375
+ Copyright (c) 2026 Eduardo J. Barrios
@@ -0,0 +1,146 @@
1
+ Metadata-Version: 2.4
2
+ Name: context-dedup
3
+ Version: 0.1.0
4
+ Summary: Detect and remove redundant context before sending it to an LLM.
5
+ Author-email: "Eduardo J. Barrios" <edujbarrios@outlook.com>
6
+ License-Expression: MPL-2.0
7
+ Project-URL: Homepage, https://github.com/edujbarrios/context-dedup
8
+ Project-URL: Source, https://github.com/edujbarrios/context-dedup
9
+ Project-URL: Issues, https://github.com/edujbarrios/context-dedup/issues
10
+ Keywords: llm,rag,context,deduplication,retrieval,ai,prompt,context-window
11
+ Classifier: Development Status :: 3 - Alpha
12
+ Classifier: Intended Audience :: Developers
13
+ Classifier: Programming Language :: Python :: 3
14
+ Classifier: Programming Language :: Python :: 3.10
15
+ Classifier: Programming Language :: Python :: 3.11
16
+ Classifier: Programming Language :: Python :: 3.12
17
+ Classifier: Programming Language :: Python :: 3.13
18
+ Classifier: Operating System :: OS Independent
19
+ Requires-Python: >=3.10
20
+ Description-Content-Type: text/markdown
21
+ License-File: LICENSE
22
+ Provides-Extra: dev
23
+ Requires-Dist: pytest; extra == "dev"
24
+ Requires-Dist: build; extra == "dev"
25
+ Requires-Dist: twine; extra == "dev"
26
+ Dynamic: license-file
27
+
28
+ # context-dedup
29
+
30
+ [![PyPI version](https://img.shields.io/pypi/v/context-dedup.svg?label=PyPI&logo=pypi&cacheSeconds=300)](https://pypi.org/project/context-dedup/)
31
+ [![License: MPL 2.0](https://img.shields.io/badge/license-MPL%202.0-blue.svg)](https://github.com/edujbarrios/context-dedup/blob/main/LICENSE)
32
+
33
+ Detect and remove redundant context before sending it to an LLM.
34
+
35
+ context-dedup is a small, deterministic Python library for finding exact and near-duplicate chunks in RAG results, agent pipelines, and assembled prompts. It uses lexical word n-gram overlap: no embeddings, no LLM calls, and no runtime dependencies.
36
+
37
+ ## Installation
38
+
39
+ ```bash
40
+ pip install context-dedup
41
+ ```
42
+
43
+ ## Usage
44
+
45
+ ```python
46
+ from context_dedup import deduplicate, inspect_context, similarity
47
+
48
+ chunks = [
49
+ "The Eiffel Tower is located in Paris.",
50
+ "The Eiffel Tower is located in Paris. It was completed in 1889.",
51
+ "Madrid is the capital of Spain.",
52
+ ]
53
+
54
+ print(similarity(chunks[0], chunks[1]))
55
+ print(inspect_context(chunks))
56
+ print(deduplicate(chunks))
57
+ ```
58
+
59
+ ## Why context-dedup?
60
+
61
+ Retrieval and prompt assembly can repeat the same facts across overlapping passages, wasting context-window space and attention. context-dedup provides a transparent heuristic between raw retrieval and an LLM call, without adding a model, service, database, or provider SDK.
62
+
63
+ It is intentionally focused:
64
+
65
+ * No embeddings or transformer models
66
+ * No LLM calls or external APIs
67
+ * No runtime dependencies or infrastructure
68
+
69
+ ## Inspect context
70
+
71
+ `inspect_context` returns an ordinary, serializable dictionary with pair scores, connected duplicate groups, recommended representatives, removable indices, and aggregate redundancy estimates:
72
+
73
+ ```python
74
+ report = inspect_context(chunks)
75
+
76
+ print(report["redundant_pairs"])
77
+ print(report["groups"])
78
+ print(report["estimated_redundant_words"])
79
+ ```
80
+
81
+ Thresholds are configurable. Their defaults are practical heuristics, not statistically universal values:
82
+
83
+ ```python
84
+ report = inspect_context(
85
+ chunks,
86
+ similarity_threshold=0.75,
87
+ containment_threshold=0.85,
88
+ n=3,
89
+ )
90
+ ```
91
+
92
+ ## Deduplicate context
93
+
94
+ The default strategy keeps the longest chunk in each duplicate group, preserving more information. Use `first` to keep the earliest chunk instead:
95
+
96
+ ```python
97
+ clean_chunks = deduplicate(chunks)
98
+ first_chunks = deduplicate(chunks, strategy="first")
99
+ ```
100
+
101
+ Objects with metadata are supported through `key`; returned objects are the originals:
102
+
103
+ ```python
104
+ retrieved = [
105
+ {"text": "Refunds are available within 30 days.", "source": "policy.pdf", "page": 4},
106
+ {"text": "Refunds are available within 30 days. Contact support to begin.", "source": "faq.pdf", "page": 2},
107
+ ]
108
+
109
+ clean = deduplicate(retrieved, key=lambda item: item["text"])
110
+ ```
111
+
112
+ ## Algorithm
113
+
114
+ Text is lowercased, stripped, and normalized to single spaces. The library builds sets of word trigrams by default, then calculates Jaccard similarity and containment in both directions. A pair is redundant when either configured threshold is reached. Connected components combine transitive pairs into groups, and selection is deterministic.
115
+
116
+ ## Limitations
117
+
118
+ This is a lexical heuristic, not a semantic-equivalence detector. Paraphrases with different wording may not match, while repeated wording can match despite different meaning. Punctuation remains part of word tokens. Version 0.1.0 compares every pair in `O(n²)` time, which is appropriate for contexts with tens or hundreds of chunks but not large document collections.
119
+
120
+ ## Use cases
121
+
122
+ * Remove overlap from RAG retrieval results before prompt construction
123
+ * Inspect context redundancy in LLM and agent pipelines
124
+ * Deduplicate passages assembled from multiple document sources
125
+ * Reduce repeated prompt content without provider-specific infrastructure
126
+
127
+ ## Features
128
+
129
+ * Deterministic word n-gram Jaccard similarity and directional containment
130
+ * Transitive duplicate groups and serializable inspection reports
131
+ * `longest` and `first` representative strategies
132
+ * Metadata-preserving `key` support
133
+ * Configurable thresholds and n-gram size
134
+ * Standard-library implementation with no runtime dependencies
135
+
136
+ ## Issues
137
+
138
+ Report issues in the [GitHub issue tracker](https://github.com/edujbarrios/context-dedup/issues).
139
+
140
+ ## Author
141
+
142
+ Eduardo J. Barrios — [edujbarrios@outlook.com](mailto:edujbarrios@outlook.com)
143
+
144
+ ## License
145
+
146
+ [Mozilla Public License 2.0](https://github.com/edujbarrios/context-dedup/blob/main/LICENSE)
@@ -0,0 +1,119 @@
1
+ # context-dedup
2
+
3
+ [![PyPI version](https://img.shields.io/pypi/v/context-dedup.svg?label=PyPI&logo=pypi&cacheSeconds=300)](https://pypi.org/project/context-dedup/)
4
+ [![License: MPL 2.0](https://img.shields.io/badge/license-MPL%202.0-blue.svg)](https://github.com/edujbarrios/context-dedup/blob/main/LICENSE)
5
+
6
+ Detect and remove redundant context before sending it to an LLM.
7
+
8
+ context-dedup is a small, deterministic Python library for finding exact and near-duplicate chunks in RAG results, agent pipelines, and assembled prompts. It uses lexical word n-gram overlap: no embeddings, no LLM calls, and no runtime dependencies.
9
+
10
+ ## Installation
11
+
12
+ ```bash
13
+ pip install context-dedup
14
+ ```
15
+
16
+ ## Usage
17
+
18
+ ```python
19
+ from context_dedup import deduplicate, inspect_context, similarity
20
+
21
+ chunks = [
22
+ "The Eiffel Tower is located in Paris.",
23
+ "The Eiffel Tower is located in Paris. It was completed in 1889.",
24
+ "Madrid is the capital of Spain.",
25
+ ]
26
+
27
+ print(similarity(chunks[0], chunks[1]))
28
+ print(inspect_context(chunks))
29
+ print(deduplicate(chunks))
30
+ ```
31
+
32
+ ## Why context-dedup?
33
+
34
+ Retrieval and prompt assembly can repeat the same facts across overlapping passages, wasting context-window space and attention. context-dedup provides a transparent heuristic between raw retrieval and an LLM call, without adding a model, service, database, or provider SDK.
35
+
36
+ It is intentionally focused:
37
+
38
+ * No embeddings or transformer models
39
+ * No LLM calls or external APIs
40
+ * No runtime dependencies or infrastructure
41
+
42
+ ## Inspect context
43
+
44
+ `inspect_context` returns an ordinary, serializable dictionary with pair scores, connected duplicate groups, recommended representatives, removable indices, and aggregate redundancy estimates:
45
+
46
+ ```python
47
+ report = inspect_context(chunks)
48
+
49
+ print(report["redundant_pairs"])
50
+ print(report["groups"])
51
+ print(report["estimated_redundant_words"])
52
+ ```
53
+
54
+ Thresholds are configurable. Their defaults are practical heuristics, not statistically universal values:
55
+
56
+ ```python
57
+ report = inspect_context(
58
+ chunks,
59
+ similarity_threshold=0.75,
60
+ containment_threshold=0.85,
61
+ n=3,
62
+ )
63
+ ```
64
+
65
+ ## Deduplicate context
66
+
67
+ The default strategy keeps the longest chunk in each duplicate group, preserving more information. Use `first` to keep the earliest chunk instead:
68
+
69
+ ```python
70
+ clean_chunks = deduplicate(chunks)
71
+ first_chunks = deduplicate(chunks, strategy="first")
72
+ ```
73
+
74
+ Objects with metadata are supported through `key`; returned objects are the originals:
75
+
76
+ ```python
77
+ retrieved = [
78
+ {"text": "Refunds are available within 30 days.", "source": "policy.pdf", "page": 4},
79
+ {"text": "Refunds are available within 30 days. Contact support to begin.", "source": "faq.pdf", "page": 2},
80
+ ]
81
+
82
+ clean = deduplicate(retrieved, key=lambda item: item["text"])
83
+ ```
84
+
85
+ ## Algorithm
86
+
87
+ Text is lowercased, stripped, and normalized to single spaces. The library builds sets of word trigrams by default, then calculates Jaccard similarity and containment in both directions. A pair is redundant when either configured threshold is reached. Connected components combine transitive pairs into groups, and selection is deterministic.
88
+
89
+ ## Limitations
90
+
91
+ This is a lexical heuristic, not a semantic-equivalence detector. Paraphrases with different wording may not match, while repeated wording can match despite different meaning. Punctuation remains part of word tokens. Version 0.1.0 compares every pair in `O(n²)` time, which is appropriate for contexts with tens or hundreds of chunks but not large document collections.
92
+
93
+ ## Use cases
94
+
95
+ * Remove overlap from RAG retrieval results before prompt construction
96
+ * Inspect context redundancy in LLM and agent pipelines
97
+ * Deduplicate passages assembled from multiple document sources
98
+ * Reduce repeated prompt content without provider-specific infrastructure
99
+
100
+ ## Features
101
+
102
+ * Deterministic word n-gram Jaccard similarity and directional containment
103
+ * Transitive duplicate groups and serializable inspection reports
104
+ * `longest` and `first` representative strategies
105
+ * Metadata-preserving `key` support
106
+ * Configurable thresholds and n-gram size
107
+ * Standard-library implementation with no runtime dependencies
108
+
109
+ ## Issues
110
+
111
+ Report issues in the [GitHub issue tracker](https://github.com/edujbarrios/context-dedup/issues).
112
+
113
+ ## Author
114
+
115
+ Eduardo J. Barrios — [edujbarrios@outlook.com](mailto:edujbarrios@outlook.com)
116
+
117
+ ## License
118
+
119
+ [Mozilla Public License 2.0](https://github.com/edujbarrios/context-dedup/blob/main/LICENSE)
@@ -0,0 +1,40 @@
1
+ [build-system]
2
+ requires = ["setuptools>=77"]
3
+ build-backend = "setuptools.build_meta"
4
+
5
+ [project]
6
+ name = "context-dedup"
7
+ version = "0.1.0"
8
+ description = "Detect and remove redundant context before sending it to an LLM."
9
+ readme = "README.md"
10
+ requires-python = ">=3.10"
11
+ license = "MPL-2.0"
12
+ license-files = ["LICENSE"]
13
+ authors = [{ name = "Eduardo J. Barrios", email = "edujbarrios@outlook.com" }]
14
+ keywords = ["llm", "rag", "context", "deduplication", "retrieval", "ai", "prompt", "context-window"]
15
+ classifiers = [
16
+ "Development Status :: 3 - Alpha",
17
+ "Intended Audience :: Developers",
18
+ "Programming Language :: Python :: 3",
19
+ "Programming Language :: Python :: 3.10",
20
+ "Programming Language :: Python :: 3.11",
21
+ "Programming Language :: Python :: 3.12",
22
+ "Programming Language :: Python :: 3.13",
23
+ "Operating System :: OS Independent",
24
+ ]
25
+
26
+ [project.optional-dependencies]
27
+ dev = ["pytest", "build", "twine"]
28
+
29
+ [project.urls]
30
+ Homepage = "https://github.com/edujbarrios/context-dedup"
31
+ Source = "https://github.com/edujbarrios/context-dedup"
32
+ Issues = "https://github.com/edujbarrios/context-dedup/issues"
33
+
34
+ [tool.setuptools.packages.find]
35
+ where = ["src"]
36
+
37
+ [tool.pytest.ini_options]
38
+ testpaths = ["tests"]
39
+ addopts = "-ra"
40
+
@@ -0,0 +1,4 @@
1
+ [egg_info]
2
+ tag_build =
3
+ tag_date = 0
4
+
@@ -0,0 +1,5 @@
1
+ """Small, deterministic tools for removing redundant LLM context."""
2
+
3
+ from .core import deduplicate, inspect_context, similarity
4
+
5
+ __all__ = ["similarity", "inspect_context", "deduplicate"]
@@ -0,0 +1,229 @@
1
+ """Core implementation for context-dedup."""
2
+
3
+ from __future__ import annotations
4
+
5
+ import math
6
+ from collections.abc import Callable, Iterable
7
+ from numbers import Real
8
+ from typing import Any
9
+
10
+ DEFAULT_SIMILARITY_THRESHOLD = 0.8
11
+ DEFAULT_CONTAINMENT_THRESHOLD = 0.65
12
+
13
+
14
+ def _normalize(text: str) -> str:
15
+ if not isinstance(text, str):
16
+ raise TypeError("text must be a string")
17
+ return " ".join(text.lower().split())
18
+
19
+
20
+ def _validate_n(n: int) -> int:
21
+ if isinstance(n, bool) or not isinstance(n, int):
22
+ raise TypeError("n must be an integer")
23
+ if n < 1:
24
+ raise ValueError("n must be at least 1")
25
+ return n
26
+
27
+
28
+ def _validate_threshold(value: float, label: str) -> float:
29
+ if isinstance(value, bool) or not isinstance(value, Real):
30
+ raise TypeError(f"{label} must be a number between 0.0 and 1.0")
31
+ result = float(value)
32
+ if not math.isfinite(result) or not 0.0 <= result <= 1.0:
33
+ raise ValueError(f"{label} must be between 0.0 and 1.0 and finite")
34
+ return result
35
+
36
+
37
+ def _word_ngrams(text: str, n: int = 3) -> set[tuple[str, ...]]:
38
+ size = _validate_n(n)
39
+ words = _normalize(text).split()
40
+ if not words:
41
+ return set()
42
+ if len(words) < size:
43
+ return {tuple(words)}
44
+ return {tuple(words[index : index + size]) for index in range(len(words) - size + 1)}
45
+
46
+
47
+ def _scores(left: set[tuple[str, ...]], right: set[tuple[str, ...]]) -> tuple[float, float, float]:
48
+ if not left and not right:
49
+ return 1.0, 1.0, 1.0
50
+ shared = len(left & right)
51
+ union = len(left | right)
52
+ jaccard = shared / union if union else 1.0
53
+ left_containment = shared / len(left) if left else 0.0
54
+ right_containment = shared / len(right) if right else 0.0
55
+ return jaccard, left_containment, right_containment
56
+
57
+
58
+ def similarity(text_a: str, text_b: str, *, n: int = 3) -> float:
59
+ """Return word n-gram Jaccard similarity from 0.0 to 1.0.
60
+
61
+ Text is lowercased and whitespace is collapsed. Trigrams are used by
62
+ default; a non-empty text shorter than ``n`` is treated as one n-gram.
63
+ Two empty or whitespace-only strings have similarity 1.0.
64
+ """
65
+ return _scores(_word_ngrams(text_a, n), _word_ngrams(text_b, n))[0]
66
+
67
+
68
+ def _find_groups(
69
+ chunks: Iterable[Any],
70
+ *,
71
+ key: Callable[[Any], str],
72
+ similarity_threshold: float,
73
+ containment_threshold: float,
74
+ n: int,
75
+ ) -> tuple[list[Any], list[str], list[dict[str, Any]], list[list[int]]]:
76
+ items = list(chunks)
77
+ texts = [key(item) for item in items]
78
+ if any(not isinstance(text, str) for text in texts):
79
+ raise TypeError("key must return a string for every chunk")
80
+ grams = [_word_ngrams(text, n) for text in texts]
81
+ parents = list(range(len(items)))
82
+
83
+ def find(index: int) -> int:
84
+ while parents[index] != index:
85
+ parents[index] = parents[parents[index]]
86
+ index = parents[index]
87
+ return index
88
+
89
+ def union(left: int, right: int) -> None:
90
+ left_root, right_root = find(left), find(right)
91
+ if left_root != right_root:
92
+ parents[right_root] = left_root
93
+
94
+ pairs = []
95
+ for left in range(len(items)):
96
+ for right in range(left + 1, len(items)):
97
+ jaccard, left_containment, right_containment = _scores(grams[left], grams[right])
98
+ if jaccard >= similarity_threshold or max(left_containment, right_containment) >= containment_threshold:
99
+ pairs.append({
100
+ "indices": [left, right],
101
+ "similarity": jaccard,
102
+ "containment": [left_containment, right_containment],
103
+ })
104
+ union(left, right)
105
+
106
+ components: dict[int, list[int]] = {}
107
+ for index in range(len(items)):
108
+ components.setdefault(find(index), []).append(index)
109
+ groups = [indices for indices in components.values() if len(indices) > 1]
110
+ return items, texts, pairs, groups
111
+
112
+
113
+ def _identity(value: Any) -> Any:
114
+ return value
115
+
116
+
117
+ def _prepare(
118
+ chunks: Iterable[Any],
119
+ key: Callable[[Any], str] | None,
120
+ similarity_threshold: float,
121
+ containment_threshold: float,
122
+ n: int,
123
+ ) -> tuple[list[Any], list[str], list[dict[str, Any]], list[list[int]]]:
124
+ if isinstance(chunks, (str, bytes)):
125
+ raise TypeError("chunks must be a non-string iterable")
126
+ try:
127
+ iter(chunks)
128
+ except TypeError as error:
129
+ raise TypeError("chunks must be a non-string iterable") from error
130
+ extractor = _identity if key is None else key
131
+ if not callable(extractor):
132
+ raise TypeError("key must be callable or None")
133
+ similarity_limit = _validate_threshold(similarity_threshold, "similarity_threshold")
134
+ containment_limit = _validate_threshold(containment_threshold, "containment_threshold")
135
+ items, texts, pairs, groups = _find_groups(
136
+ chunks,
137
+ key=extractor,
138
+ similarity_threshold=similarity_limit,
139
+ containment_threshold=containment_limit,
140
+ n=n,
141
+ )
142
+ return items, texts, pairs, groups
143
+
144
+
145
+ def _representative(indices: list[int], texts: list[str], strategy: str) -> int:
146
+ if strategy == "first":
147
+ return indices[0]
148
+ if strategy == "longest":
149
+ return max(indices, key=lambda index: len(texts[index]))
150
+ raise ValueError("strategy must be 'longest' or 'first'")
151
+
152
+
153
+ def inspect_context(
154
+ chunks: Iterable[Any],
155
+ *,
156
+ strategy: str = "longest",
157
+ key: Callable[[Any], str] | None = None,
158
+ similarity_threshold: float = DEFAULT_SIMILARITY_THRESHOLD,
159
+ containment_threshold: float = DEFAULT_CONTAINMENT_THRESHOLD,
160
+ n: int = 3,
161
+ ) -> dict[str, Any]:
162
+ """Inspect chunks and return a serializable lexical-redundancy report.
163
+
164
+ A pair is redundant when its Jaccard score reaches ``similarity_threshold``
165
+ or either directional containment score reaches ``containment_threshold``.
166
+ The defaults are practical heuristics, not universal statistical cutoffs.
167
+ """
168
+ if strategy not in {"longest", "first"}:
169
+ raise ValueError("strategy must be 'longest' or 'first'")
170
+ items, texts, pairs, components = _prepare(
171
+ chunks, key, similarity_threshold, containment_threshold, n
172
+ )
173
+ groups = []
174
+ removable: list[int] = []
175
+ for indices in components:
176
+ keep = _representative(indices, texts, strategy)
177
+ remove = [index for index in indices if index != keep]
178
+ removable.extend(remove)
179
+ group_pairs = [
180
+ pair for pair in pairs if pair["indices"][0] in indices and pair["indices"][1] in indices
181
+ ]
182
+ groups.append({
183
+ "indices": indices,
184
+ "representative": keep,
185
+ "remove_indices": remove,
186
+ "max_similarity": max(pair["similarity"] for pair in group_pairs),
187
+ "max_containment": max(max(pair["containment"]) for pair in group_pairs),
188
+ })
189
+ removable.sort()
190
+ total = len(items)
191
+ return {
192
+ "total_chunks": total,
193
+ "redundant_pairs": len(pairs),
194
+ "redundancy_groups": len(groups),
195
+ "redundant_chunks": len(removable),
196
+ "redundancy_ratio": len(removable) / total if total else 0.0,
197
+ "estimated_redundant_characters": sum(len(texts[index]) for index in removable),
198
+ "estimated_redundant_words": sum(len(texts[index].split()) for index in removable),
199
+ "removable_indices": removable,
200
+ "pairs": pairs,
201
+ "groups": groups,
202
+ }
203
+
204
+
205
+ def deduplicate(
206
+ chunks: Iterable[Any],
207
+ *,
208
+ strategy: str = "longest",
209
+ key: Callable[[Any], str] | None = None,
210
+ similarity_threshold: float = DEFAULT_SIMILARITY_THRESHOLD,
211
+ containment_threshold: float = DEFAULT_CONTAINMENT_THRESHOLD,
212
+ n: int = 3,
213
+ ) -> list[Any]:
214
+ """Return chunks with lexical duplicates removed, preserving input order.
215
+
216
+ ``longest`` keeps the longest text in each group (and the first on ties).
217
+ ``first`` keeps the earliest chunk. Original objects are returned unchanged.
218
+ """
219
+ items = list(chunks) if not isinstance(chunks, (str, bytes)) else chunks
220
+ report = inspect_context(
221
+ items,
222
+ strategy=strategy,
223
+ key=key,
224
+ similarity_threshold=similarity_threshold,
225
+ containment_threshold=containment_threshold,
226
+ n=n,
227
+ )
228
+ remove = set(report["removable_indices"])
229
+ return [item for index, item in enumerate(items) if index not in remove]
@@ -0,0 +1,146 @@
1
+ Metadata-Version: 2.4
2
+ Name: context-dedup
3
+ Version: 0.1.0
4
+ Summary: Detect and remove redundant context before sending it to an LLM.
5
+ Author-email: "Eduardo J. Barrios" <edujbarrios@outlook.com>
6
+ License-Expression: MPL-2.0
7
+ Project-URL: Homepage, https://github.com/edujbarrios/context-dedup
8
+ Project-URL: Source, https://github.com/edujbarrios/context-dedup
9
+ Project-URL: Issues, https://github.com/edujbarrios/context-dedup/issues
10
+ Keywords: llm,rag,context,deduplication,retrieval,ai,prompt,context-window
11
+ Classifier: Development Status :: 3 - Alpha
12
+ Classifier: Intended Audience :: Developers
13
+ Classifier: Programming Language :: Python :: 3
14
+ Classifier: Programming Language :: Python :: 3.10
15
+ Classifier: Programming Language :: Python :: 3.11
16
+ Classifier: Programming Language :: Python :: 3.12
17
+ Classifier: Programming Language :: Python :: 3.13
18
+ Classifier: Operating System :: OS Independent
19
+ Requires-Python: >=3.10
20
+ Description-Content-Type: text/markdown
21
+ License-File: LICENSE
22
+ Provides-Extra: dev
23
+ Requires-Dist: pytest; extra == "dev"
24
+ Requires-Dist: build; extra == "dev"
25
+ Requires-Dist: twine; extra == "dev"
26
+ Dynamic: license-file
27
+
28
+ # context-dedup
29
+
30
+ [![PyPI version](https://img.shields.io/pypi/v/context-dedup.svg?label=PyPI&logo=pypi&cacheSeconds=300)](https://pypi.org/project/context-dedup/)
31
+ [![License: MPL 2.0](https://img.shields.io/badge/license-MPL%202.0-blue.svg)](https://github.com/edujbarrios/context-dedup/blob/main/LICENSE)
32
+
33
+ Detect and remove redundant context before sending it to an LLM.
34
+
35
+ context-dedup is a small, deterministic Python library for finding exact and near-duplicate chunks in RAG results, agent pipelines, and assembled prompts. It uses lexical word n-gram overlap: no embeddings, no LLM calls, and no runtime dependencies.
36
+
37
+ ## Installation
38
+
39
+ ```bash
40
+ pip install context-dedup
41
+ ```
42
+
43
+ ## Usage
44
+
45
+ ```python
46
+ from context_dedup import deduplicate, inspect_context, similarity
47
+
48
+ chunks = [
49
+ "The Eiffel Tower is located in Paris.",
50
+ "The Eiffel Tower is located in Paris. It was completed in 1889.",
51
+ "Madrid is the capital of Spain.",
52
+ ]
53
+
54
+ print(similarity(chunks[0], chunks[1]))
55
+ print(inspect_context(chunks))
56
+ print(deduplicate(chunks))
57
+ ```
58
+
59
+ ## Why context-dedup?
60
+
61
+ Retrieval and prompt assembly can repeat the same facts across overlapping passages, wasting context-window space and attention. context-dedup provides a transparent heuristic between raw retrieval and an LLM call, without adding a model, service, database, or provider SDK.
62
+
63
+ It is intentionally focused:
64
+
65
+ * No embeddings or transformer models
66
+ * No LLM calls or external APIs
67
+ * No runtime dependencies or infrastructure
68
+
69
+ ## Inspect context
70
+
71
+ `inspect_context` returns an ordinary, serializable dictionary with pair scores, connected duplicate groups, recommended representatives, removable indices, and aggregate redundancy estimates:
72
+
73
+ ```python
74
+ report = inspect_context(chunks)
75
+
76
+ print(report["redundant_pairs"])
77
+ print(report["groups"])
78
+ print(report["estimated_redundant_words"])
79
+ ```
80
+
81
+ Thresholds are configurable. Their defaults are practical heuristics, not statistically universal values:
82
+
83
+ ```python
84
+ report = inspect_context(
85
+ chunks,
86
+ similarity_threshold=0.75,
87
+ containment_threshold=0.85,
88
+ n=3,
89
+ )
90
+ ```
91
+
92
+ ## Deduplicate context
93
+
94
+ The default strategy keeps the longest chunk in each duplicate group, preserving more information. Use `first` to keep the earliest chunk instead:
95
+
96
+ ```python
97
+ clean_chunks = deduplicate(chunks)
98
+ first_chunks = deduplicate(chunks, strategy="first")
99
+ ```
100
+
101
+ Objects with metadata are supported through `key`; returned objects are the originals:
102
+
103
+ ```python
104
+ retrieved = [
105
+ {"text": "Refunds are available within 30 days.", "source": "policy.pdf", "page": 4},
106
+ {"text": "Refunds are available within 30 days. Contact support to begin.", "source": "faq.pdf", "page": 2},
107
+ ]
108
+
109
+ clean = deduplicate(retrieved, key=lambda item: item["text"])
110
+ ```
111
+
112
+ ## Algorithm
113
+
114
+ Text is lowercased, stripped, and normalized to single spaces. The library builds sets of word trigrams by default, then calculates Jaccard similarity and containment in both directions. A pair is redundant when either configured threshold is reached. Connected components combine transitive pairs into groups, and selection is deterministic.
115
+
116
+ ## Limitations
117
+
118
+ This is a lexical heuristic, not a semantic-equivalence detector. Paraphrases with different wording may not match, while repeated wording can match despite different meaning. Punctuation remains part of word tokens. Version 0.1.0 compares every pair in `O(n²)` time, which is appropriate for contexts with tens or hundreds of chunks but not large document collections.
119
+
120
+ ## Use cases
121
+
122
+ * Remove overlap from RAG retrieval results before prompt construction
123
+ * Inspect context redundancy in LLM and agent pipelines
124
+ * Deduplicate passages assembled from multiple document sources
125
+ * Reduce repeated prompt content without provider-specific infrastructure
126
+
127
+ ## Features
128
+
129
+ * Deterministic word n-gram Jaccard similarity and directional containment
130
+ * Transitive duplicate groups and serializable inspection reports
131
+ * `longest` and `first` representative strategies
132
+ * Metadata-preserving `key` support
133
+ * Configurable thresholds and n-gram size
134
+ * Standard-library implementation with no runtime dependencies
135
+
136
+ ## Issues
137
+
138
+ Report issues in the [GitHub issue tracker](https://github.com/edujbarrios/context-dedup/issues).
139
+
140
+ ## Author
141
+
142
+ Eduardo J. Barrios — [edujbarrios@outlook.com](mailto:edujbarrios@outlook.com)
143
+
144
+ ## License
145
+
146
+ [Mozilla Public License 2.0](https://github.com/edujbarrios/context-dedup/blob/main/LICENSE)
@@ -0,0 +1,11 @@
1
+ LICENSE
2
+ README.md
3
+ pyproject.toml
4
+ src/context_dedup/__init__.py
5
+ src/context_dedup/core.py
6
+ src/context_dedup.egg-info/PKG-INFO
7
+ src/context_dedup.egg-info/SOURCES.txt
8
+ src/context_dedup.egg-info/dependency_links.txt
9
+ src/context_dedup.egg-info/requires.txt
10
+ src/context_dedup.egg-info/top_level.txt
11
+ tests/test_core.py
@@ -0,0 +1,5 @@
1
+
2
+ [dev]
3
+ pytest
4
+ build
5
+ twine
@@ -0,0 +1 @@
1
+ context_dedup
@@ -0,0 +1,164 @@
1
+ import math
2
+
3
+ import pytest
4
+
5
+ from context_dedup import deduplicate, inspect_context, similarity
6
+ from context_dedup.core import _find_groups
7
+
8
+
9
+ def test_exact_capitalization_and_whitespace_duplicates():
10
+ assert similarity("Python is a language", "Python is a language") == 1.0
11
+ assert similarity(" PYTHON is\na language ", "python is a language") == 1.0
12
+
13
+
14
+ def test_different_strings_and_symmetry():
15
+ left = "Madrid is the capital of Spain"
16
+ right = "Whales live in the ocean"
17
+ assert similarity(left, right) == 0.0
18
+ assert similarity(left, right) == similarity(right, left)
19
+
20
+
21
+ def test_near_duplicate_has_partial_similarity():
22
+ score = similarity("one two three four five", "one two three four six")
23
+ assert 0.0 < score < 1.0
24
+
25
+
26
+ def test_empty_and_short_strings():
27
+ assert similarity("", " \n") == 1.0
28
+ assert similarity("", "content") == 0.0
29
+ assert similarity("short text", "SHORT TEXT") == 1.0
30
+
31
+
32
+ def test_containment_and_transitive_groups():
33
+ chunks = [
34
+ "alpha beta gamma delta",
35
+ "alpha beta gamma delta epsilon zeta",
36
+ "gamma delta epsilon zeta eta",
37
+ ]
38
+ _, _, pairs, groups = _find_groups(
39
+ chunks,
40
+ key=lambda value: value,
41
+ similarity_threshold=0.9,
42
+ containment_threshold=0.65,
43
+ n=3,
44
+ )
45
+ assert [pair["indices"] for pair in pairs] == [[0, 1], [1, 2]]
46
+ assert groups == [[0, 1, 2]]
47
+
48
+
49
+ @pytest.mark.parametrize("n,error", [(0, ValueError), (-1, ValueError), (1.5, TypeError), (True, TypeError)])
50
+ def test_invalid_ngram_size(n, error):
51
+ with pytest.raises(error):
52
+ similarity("a", "a", n=n)
53
+
54
+
55
+ @pytest.mark.parametrize("value", [None, 1, []])
56
+ def test_non_string_input(value):
57
+ with pytest.raises(TypeError):
58
+ similarity(value, "text")
59
+
60
+
61
+ def test_report_consistency_and_longest_selection():
62
+ chunks = [
63
+ "The Eiffel Tower is located in Paris.",
64
+ "The Eiffel Tower is located in Paris. It was completed in 1889.",
65
+ "Madrid is the capital of Spain.",
66
+ ]
67
+ report = inspect_context(chunks)
68
+ assert report["total_chunks"] == 3
69
+ assert report["redundant_pairs"] == 1
70
+ assert report["redundancy_groups"] == 1
71
+ assert report["redundant_chunks"] == 1
72
+ assert report["removable_indices"] == [0]
73
+ assert report["groups"][0]["representative"] == 1
74
+ assert report["redundancy_ratio"] == pytest.approx(1 / 3)
75
+ assert report["estimated_redundant_characters"] == len(chunks[0])
76
+ assert report["estimated_redundant_words"] == len(chunks[0].split())
77
+ assert deduplicate(chunks) == chunks[1:]
78
+
79
+
80
+ def test_default_containment_matches_overlapping_rag_chunks():
81
+ chunks = [
82
+ "Python is a programming language.",
83
+ "Python is a programming language used for data science.",
84
+ "Madrid is the capital of Spain.",
85
+ ]
86
+ assert inspect_context(chunks)["redundant_pairs"] == 1
87
+ assert deduplicate(chunks) == chunks[1:]
88
+
89
+
90
+ def test_first_strategy_and_no_mutation():
91
+ chunks = ["alpha beta gamma", "alpha beta gamma delta"]
92
+ original = chunks.copy()
93
+ assert deduplicate(chunks, strategy="first") == [chunks[0]]
94
+ assert chunks == original
95
+
96
+
97
+ def test_configurable_thresholds():
98
+ chunks = ["one two three four five", "one two three four six"]
99
+ assert deduplicate(chunks, similarity_threshold=0.49, containment_threshold=1.0) == [chunks[0]]
100
+ assert deduplicate(chunks, similarity_threshold=0.51, containment_threshold=1.0) == chunks
101
+
102
+
103
+ def test_metadata_key_preserves_original_objects():
104
+ chunks = [
105
+ {"text": "same useful context", "page": 1},
106
+ {"text": "same useful context with detail", "page": 2},
107
+ ]
108
+ result = deduplicate(chunks, key=lambda item: item["text"])
109
+ assert result == [chunks[1]]
110
+ assert result[0] is chunks[1]
111
+
112
+
113
+ def test_key_is_called_once_per_item():
114
+ calls = []
115
+
116
+ def extract(item):
117
+ calls.append(item)
118
+ return item["text"]
119
+
120
+ chunks = [{"text": "one"}, {"text": "two"}]
121
+ inspect_context(chunks, key=extract)
122
+ assert calls == chunks
123
+
124
+
125
+ def test_transitive_deduplicate_is_deterministic():
126
+ chunks = [
127
+ "alpha beta gamma delta",
128
+ "alpha beta gamma delta epsilon zeta",
129
+ "gamma delta epsilon zeta eta",
130
+ ]
131
+ options = {"similarity_threshold": 0.9, "containment_threshold": 0.65}
132
+ first = inspect_context(chunks, **options)
133
+ assert first == inspect_context(chunks, **options)
134
+ assert first["groups"][0]["indices"] == [0, 1, 2]
135
+ assert deduplicate(chunks, **options) == [chunks[1]]
136
+
137
+
138
+ def test_empty_context_report():
139
+ report = inspect_context([])
140
+ assert report["total_chunks"] == 0
141
+ assert report["redundancy_ratio"] == 0.0
142
+ assert report["groups"] == []
143
+ assert deduplicate([]) == []
144
+
145
+
146
+ @pytest.mark.parametrize("strategy", ["last", "shortest", ""])
147
+ def test_invalid_strategy(strategy):
148
+ with pytest.raises(ValueError):
149
+ deduplicate(["a"], strategy=strategy)
150
+
151
+
152
+ @pytest.mark.parametrize("threshold", [-0.1, 1.1, math.nan, math.inf])
153
+ def test_invalid_threshold(threshold):
154
+ with pytest.raises(ValueError):
155
+ inspect_context(["a"], similarity_threshold=threshold)
156
+
157
+
158
+ def test_invalid_chunks_and_key():
159
+ with pytest.raises(TypeError):
160
+ inspect_context("not a chunk collection")
161
+ with pytest.raises(TypeError):
162
+ inspect_context([{"text": "x"}], key=None)
163
+ with pytest.raises(TypeError):
164
+ inspect_context(["x"], key="text")