ai-prompter 0.2.1__tar.gz → 0.2.2__tar.gz

{ai_prompter-0.2.1 → ai_prompter-0.2.2}/PKG-INFO

@@ -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.1 → ai_prompter-0.2.2}/notebooks/prompter_usage.ipynb

@@ -16,7 +16,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 6,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -36,7 +36,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 7,
    "metadata": {},
    "outputs": [
     {
@@ -45,7 +45,7 @@
        "'Write an article about AI.'"
       ]
      },
-     "execution_count": 2,
+     "execution_count": 7,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -71,7 +71,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 8,
    "metadata": {},
    "outputs": [
     {
@@ -80,7 +80,7 @@
        "'Write an article about AI.'"
       ]
      },
-     "execution_count": 3,
+     "execution_count": 8,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -103,7 +103,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 9,
    "metadata": {},
    "outputs": [
     {
@@ -112,7 +112,7 @@
        "'This is the outer file \\n\\nThis is the inner file\\n\\nValue: a\\n\\n\\n    You selected A\\n\\n\\nThis is the end of the outer file'"
       ]
      },
-     "execution_count": 4,
+     "execution_count": 9,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -136,7 +136,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 10,
    "metadata": {},
    "outputs": [
     {
@@ -145,7 +145,7 @@
        "ChatPromptTemplate(input_variables=['topic'], input_types={}, partial_variables={}, messages=[HumanMessagePromptTemplate(prompt=PromptTemplate(input_variables=['topic'], input_types={}, partial_variables={}, template='Write an article about {{topic}}.', template_format='jinja2'), additional_kwargs={})])"
       ]
      },
-     "execution_count": 5,
+     "execution_count": 10,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -162,7 +162,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 11,
    "metadata": {},
    "outputs": [
     {
@@ -171,7 +171,7 @@
        "ChatPromptTemplate(input_variables=['type'], input_types={}, partial_variables={}, messages=[HumanMessagePromptTemplate(prompt=PromptTemplate(input_variables=['type'], input_types={}, partial_variables={}, template=\"This is the outer file \\n\\nThis is the inner file\\n\\nValue: {{ type }}\\n\\n{% if type == 'a' %}\\n    You selected A\\n{% else %}\\n    You didn't select A\\n{% endif %}\\n\\n\\nThis is the end of the outer file\\n\", template_format='jinja2'), additional_kwargs={})])"
       ]
      },
-     "execution_count": 6,
+     "execution_count": 11,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -184,7 +184,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 12,
    "metadata": {},
    "outputs": [
     {
@@ -193,7 +193,7 @@
        "ChatPromptValue(messages=[HumanMessage(content='This is the outer file \\n\\nThis is the inner file\\n\\nValue: a\\n\\n\\n    You selected A\\n\\n\\n\\nThis is the end of the outer file', additional_kwargs={}, response_metadata={})])"
       ]
      },
-     "execution_count": 7,
+     "execution_count": 12,
      "metadata": {},
      "output_type": "execute_result"
     }

{ai_prompter-0.2.1 → ai_prompter-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ai-prompter"
-version = "0.2.1"
+version = "0.2.2"
 description = "A prompt management library using Jinja2 templates to build complex prompts easily."
 readme = "README.md"
 homepage = "https://github.com/lfnovo/ai-prompter"

{ai_prompter-0.2.1 → ai_prompter-0.2.2}/src/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 → ai_prompter-0.2.2}/tests/test_prompter.py

@@ -17,10 +17,15 @@ def test_raw_text_template():
 
 
 def test_missing_both_raises():
-    with pytest.raises(ValueError):
+    with pytest.raises(ValueError, match="Either prompt_template or template_text must be provided"):
         Prompter()
 
 
+def test_both_parameters_raises():
+    with pytest.raises(ValueError, match="Cannot provide both"):
+        Prompter(template_text="Hello", prompt_template="template")
+
+
 def test_base_model_data():
     class DataModel(BaseModel):
         foo: str

{ai_prompter-0.2.1 → ai_prompter-0.2.2}/uv.lock

@@ -8,7 +8,7 @@ resolution-markers = [
 
 [[package]]
 name = "ai-prompter"
-version = "0.2.1"
+version = "0.2.2"
 source = { editable = "." }
 dependencies = [
     { name = "jinja2" },