ai-prompter 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

ai_prompter/__init__.py

@@ -5,7 +5,7 @@ A prompt management module using Jinja to generate complex prompts with simple t
 import os
 from dataclasses import dataclass
 from datetime import datetime
-from typing import Any, Dict, Optional, Union
+from typing import Any, Dict, Optional, Union, Callable
 
 from jinja2 import Environment, FileSystemLoader, Template
 from pydantic import BaseModel
@@ -39,24 +39,36 @@ class Prompter:
 
     def __init__(
         self,
-        prompt_template: Optional[str] = None,
-        model: Optional[Union[str, Any]] = None,
-        template_text: Optional[str] = None,
-        prompt_dir: Optional[str] = None,
+        prompt_template: str | None = None,
+        model: str | Any | None = None,
+        prompt_variation: str = "default",
+        prompt_dir: str | None = None,
+        template_text: str | None = None,
+        parser: Callable[[str], dict[str, Any]] | None = None,
+        *args,
+        **kwargs,
     ) -> None:
         """Initialize the Prompter with a template name, model, and optional custom directory.
 
         Args:
             prompt_template (str, optional): The name of the prompt template (without .jinja extension).
             model (Union[str, Any], optional): The model to use for generation.
-            template_text (str, optional): The raw text of the template.
+            prompt_variation (str, optional): The variation of the prompt template.
             prompt_dir (str, optional): Custom directory to search for templates.
+            template_text (str, optional): The raw text of the template.
+            parser (Callable[[str], dict[str, Any]], optional): The parser to use for generation.
         """
+        if template_text is not None and prompt_template is not None:
+            raise ValueError(
+                "Cannot provide both template_text and prompt_template. Choose one or the other."
+            )
         self.prompt_template = prompt_template
-        self.template = None
+        self.prompt_variation = prompt_variation
+        self.prompt_dir = prompt_dir
         self.template_text = template_text
+        self.parser = parser
+        self.template: Template | None = None
         self.model = model or os.getenv("OPENAI_MODEL", "gpt-4-turbo")
-        self.prompt_dir = prompt_dir
         self._setup_template(template_text, prompt_dir)
 
     def _setup_template(

{ai_prompter-0.2.1.dist-info → ai_prompter-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-prompter
-Version: 0.2.1
+Version: 0.2.2
 Summary: A prompt management library using Jinja2 templates to build complex prompts easily.
 Author-email: LUIS NOVO <lfnovo@gmail.com>
 License: MIT

ai_prompter-0.2.2.dist-info/RECORD

@@ -0,0 +1,6 @@
+ai_prompter/__init__.py,sha256=p3AyJ9R3C3c8BL2DdMxwYFEv0UMcH-Bz_d6nQ-YjKKA,10744
+ai_prompter/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+ai_prompter-0.2.2.dist-info/METADATA,sha256=wKeWaoaiYixNYEAjEmkhZYp8FfO68hwjtfjApHgtvIQ,6950
+ai_prompter-0.2.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ai_prompter-0.2.2.dist-info/licenses/LICENSE,sha256=cS0_fa_8BoP0PvVG8D19pn_HDJrG96hd4PyEm9nkRo8,1066
+ai_prompter-0.2.2.dist-info/RECORD,,

ai_prompter-0.2.1.dist-info/RECORD

@@ -1,6 +0,0 @@
-ai_prompter/__init__.py,sha256=ehtuhOKP0ze3sW1cYO4OyNhoB8CY926TmkPqvzz4ilA,10124
-ai_prompter/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-ai_prompter-0.2.1.dist-info/METADATA,sha256=c4Wh6qs7JVcBwWf1FCe2MZtsw5HjygYuXyJzhuKFTUc,6950
-ai_prompter-0.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ai_prompter-0.2.1.dist-info/licenses/LICENSE,sha256=cS0_fa_8BoP0PvVG8D19pn_HDJrG96hd4PyEm9nkRo8,1066
-ai_prompter-0.2.1.dist-info/RECORD,,