refactor chat templates

Author: Samoed

autointent/generation/utterances/basic/chat_templates/_synthesizer_en.py

@@ -70,7 +70,7 @@ class EnglishSynthesizerTemplate(BaseSynthesizerTemplate):
         ),
         Message(
             role=Role.ASSISTANT,
-            content=("1. Can you tell me the forecast for tomorrow?\n" "2. Is it going to rain this weekend?"),
+            content="1. Can you tell me the forecast for tomorrow?\n2. Is it going to rain this weekend?",
         ),
         Message(
             role=Role.USER,

autointent/generation/utterances/basic/utterance_generator.py

@@ -8,6 +8,7 @@
 
 from autointent import Dataset
 from autointent.custom_types import Split
+from autointent.generation.utterances.basic.chat_templates import BaseSynthesizerTemplate
 from autointent.generation.utterances.generator import Generator
 from autointent.generation.utterances.schemas import Message
 from autointent.schemas import Intent, Sample
@@ -23,7 +24,7 @@ class UtteranceGenerator:
     """
 
     def __init__(
-        self, generator: Generator, prompt_maker: Callable[[Intent, int], list[Message]], async_mode: bool = False
+        self, generator: Generator, prompt_maker: BaseSynthesizerTemplate, async_mode: bool = False
     ) -> None:
         """Initialize."""
         self.generator = generator

autointent/generation/utterances/evolution/evolver.py

@@ -13,6 +13,7 @@
 
 from autointent import Dataset
 from autointent.custom_types import Split
+from autointent.generation.utterances import EvolutionChatTemplate
 from autointent.generation.utterances.generator import Generator
 from autointent.generation.utterances.schemas import Message
 from autointent.schemas import Intent
@@ -29,7 +30,7 @@ class UtteranceEvolver:
     def __init__(
         self,
         generator: Generator,
-        prompt_makers: Sequence[Callable[[str, Intent], list[Message]]],
+        prompt_makers: Sequence[EvolutionChatTemplate],
         seed: int = 0,
         async_mode: bool = False,
     ) -> None:

autointent/generation/utterances/evolution/incremental_evolver.py

@@ -14,6 +14,7 @@
 
 from autointent import Dataset, Pipeline
 from autointent.custom_types import Split
+from autointent.generation.utterances import EvolutionChatTemplate
 from autointent.generation.utterances.evolution.evolver import UtteranceEvolver
 from autointent.generation.utterances.generator import Generator
 from autointent.generation.utterances.schemas import Message
@@ -47,7 +48,7 @@ class IncrementalUtteranceEvolver(UtteranceEvolver):
     def __init__(
         self,
         generator: Generator,
-        prompt_makers: Sequence[Callable[[str, Intent], list[Message]]],
+        prompt_makers: Sequence[EvolutionChatTemplate],
         seed: int = 0,
         async_mode: bool = False,
         search_space: str | None = None,

pyproject.toml

@@ -30,7 +30,7 @@ classifiers=[
     'Framework :: Sphinx',
     'Typing :: Typed',
 ]
-requires-python = ">=3.10,<=4.0"
+requires-python = ">=3.10,<4.0"
 dependencies = [
     "sentence-transformers (>=3,<4)",
     "scikit-learn (>=1.5,<2.0)",