chatan 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

chatan/__init__.py

@@ -1,6 +1,6 @@
 """Minos: Create synthetic datasets with LLM generators and samplers."""
 
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 
 from .dataset import dataset
 from .generator import generator

chatan/generator.py

@@ -58,14 +58,24 @@ class AnthropicGenerator(BaseGenerator):
 
 class GeneratorFunction:
     """Callable generator function for use in dataset schemas."""
-    
-    def __init__(self, generator: BaseGenerator, prompt_template: str):
+
+    def __init__(
+        self,
+        generator: BaseGenerator,
+        prompt_template: str,
+        variables: Optional[Dict[str, Any]] = None,
+    ):
         self.generator = generator
         self.prompt_template = prompt_template
-    
+        self.variables = variables or {}
+
     def __call__(self, context: Dict[str, Any]) -> str:
         """Generate content with context substitution."""
-        prompt = self.prompt_template.format(**context)
+        merged = dict(context)
+        for key, value in self.variables.items():
+            merged[key] = value(context) if callable(value) else value
+
+        prompt = self.prompt_template.format(**merged)
         result = self.generator.generate(prompt)
         return result.strip() if isinstance(result, str) else result
 
@@ -81,9 +91,19 @@ class GeneratorClient:
         else:
             raise ValueError(f"Unsupported provider: {provider}")
     
-    def __call__(self, prompt_template: str) -> GeneratorFunction:
-        """Create a generator function."""
-        return GeneratorFunction(self._generator, prompt_template)
+    def __call__(self, prompt_template: str, **variables) -> GeneratorFunction:
+        """Create a generator function.
+
+        Parameters
+        ----------
+        prompt_template:
+            Template string for the prompt.
+        **variables:
+            Optional variables to include when formatting the prompt. If a value
+            is callable it will be invoked with the row context when the
+            generator function is executed.
+        """
+        return GeneratorFunction(self._generator, prompt_template, variables)
 
 
 # Factory function

{chatan-0.1.0.dist-info → chatan-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chatan
-Version: 0.1.0
+Version: 0.1.1
 Summary: Create synthetic datasets with LLM generators and samplers
 Project-URL: Documentation, https://github.com/cdreetz/chatan#readme
 Project-URL: Issues, https://github.com/cdreetz/chatan/issues

chatan-0.1.1.dist-info/RECORD

@@ -0,0 +1,7 @@
+chatan/__init__.py,sha256=SebE8sKfiXl3FC0iwYi7ft9piJgOVdRXnj6g17z5rzo,233
+chatan/dataset.py,sha256=t6RrrQchsD2dROD886IfnlXWnn-F-IAWMcEIK0SS3xg,4358
+chatan/generator.py,sha256=zAOVxd2iY_KMK21dCMeFhvsJu2q8NEwch5jGJP-Me9s,3914
+chatan/sampler.py,sha256=0X6AVQK20py4SwKnsppZC2yAZnP_jBhRxt9MfT1e-k4,4812
+chatan-0.1.1.dist-info/METADATA,sha256=cCZgPzZaGvLWBDqDBj3dS7Fzh-l4HFFqGDoy01n5Sw8,2549
+chatan-0.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+chatan-0.1.1.dist-info/RECORD,,

chatan-0.1.0.dist-info/RECORD

@@ -1,7 +0,0 @@
-chatan/__init__.py,sha256=GdCxiObHomq4-2TyqS2dfEd2EuDYGpi2AkvC5QIb3mU,233
-chatan/dataset.py,sha256=t6RrrQchsD2dROD886IfnlXWnn-F-IAWMcEIK0SS3xg,4358
-chatan/generator.py,sha256=kj8axCWI00F8R0DJL831AUtK41Bvv2VL4nOweTgGfGc,3286
-chatan/sampler.py,sha256=0X6AVQK20py4SwKnsppZC2yAZnP_jBhRxt9MfT1e-k4,4812
-chatan-0.1.0.dist-info/METADATA,sha256=A5L6T8zlmoO5tKG30kmjQKQnOcCivSypaw8TFd4fS_k,2549
-chatan-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-chatan-0.1.0.dist-info/RECORD,,