Rename ChunkOverlapTask -> ChunkOverlap

Aleph-Alpha · Feb 29, 2024 · 501589a · 501589a
1 parent f96c738
commit 501589a
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/src/intelligence_layer/core/__init__.py b/src/intelligence_layer/core/__init__.py
@@ -7,7 +7,7 @@
 from .chunk import Chunk as Chunk
 from .chunk import ChunkInput as ChunkInput
 from .chunk import ChunkOutput as ChunkOutput
-from .chunk import ChunkOverlapTask as ChunkOverlapTask
+from .chunk import ChunkOverlap as ChunkOverlap
 from .chunk import TextChunk as TextChunk
 from .detect_language import DetectLanguage as DetectLanguage
 from .detect_language import DetectLanguageInput as DetectLanguageInput

diff --git a/src/intelligence_layer/core/chunk.py b/src/intelligence_layer/core/chunk.py
@@ -66,7 +66,7 @@ def do_run(self, input: ChunkInput, task_span: TaskSpan) -> ChunkOutput:
         return ChunkOutput(chunks=chunks)
 
 
-class ChunkOverlapTask(Task[ChunkInput, ChunkOutput]):
+class ChunkOverlap(Task[ChunkInput, ChunkOutput]):
     """Splits a longer text into smaller text chunks, where every chunk overlaps
     with the previous chunk by `overlap_length_tokens` number of tokens.
 

diff --git a/src/intelligence_layer/use_cases/summarize/steerable_long_context_summarize.py b/src/intelligence_layer/use_cases/summarize/steerable_long_context_summarize.py
@@ -1,7 +1,7 @@
 from typing import Mapping
 
 from intelligence_layer.core import Chunk, ChunkInput, Task, TaskSpan
-from intelligence_layer.core.chunk import ChunkOutput, ChunkOverlapTask
+from intelligence_layer.core.chunk import ChunkOutput, ChunkOverlap
 from intelligence_layer.core.detect_language import Language
 from intelligence_layer.core.model import ControlModel, LuminousControlModel
 from intelligence_layer.use_cases.summarize.steerable_single_chunk_summarize import (
@@ -52,7 +52,7 @@ def __init__(
         if overlap_length_tokens == 0:
             self._chunk_task = Chunk(model, max_tokens_per_chunk)
         else:
-            self._chunk_task = ChunkOverlapTask(
+            self._chunk_task = ChunkOverlap(
                 model,
                 max_tokens_per_chunk,
                 overlap_length_tokens,

diff --git a/tests/core/test_chunk.py b/tests/core/test_chunk.py
@@ -2,7 +2,7 @@
 
 from intelligence_layer.core import (
     ChunkInput,
-    ChunkOverlapTask,
+    ChunkOverlap,
     InMemoryTracer,
     LuminousControlModel,
 )
@@ -24,7 +24,7 @@ def test_overlapped_chunking(
     MAX_TOKENS = 16
 
     tracer = InMemoryTracer()
-    task = ChunkOverlapTask(
+    task = ChunkOverlap(
         model=luminous_control_model,
         max_tokens_per_chunk=MAX_TOKENS,
         overlap_length_tokens=OVERLAP,