Moved rendered prompt debug logging to Prompt class

Author: christo-olivier

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "modelsmith"
-version = "0.3.0"
+version = "0.4.0"
 description = "Get Pydantic models and Python types as LLM responses from Google Vertex AI models."
 authors = ["Christo Olivier <mail@christoolivier.com>"]
 maintainers = ["Christo Olivier <mail@christoolivier.com>"]

src/modelsmith/forge.py

@@ -98,7 +98,6 @@ def generate(
             for attempt in Retrying(stop=stop_after_attempt(self.max_retries)):
                 with attempt:
                     try:
-                        logger.debug(f"Prompt sent to LLM:\n{prepared_prompt}")
                         response = self.model.send(prepared_prompt, model_settings)
                         model_response = self._process_response(response)
                     except (CombinedException, PatternNotFound) as e:

src/modelsmith/prompt.py

@@ -1,10 +1,13 @@
 import inspect
 import json
+import logging
 from typing import Any
 
 from jinja2 import BaseLoader, Environment, StrictUndefined, meta, select_autoescape
 from pydantic import BaseModel
 
+logger = logging.getLogger(__name__)
+
 RESPONSE_MODEL_TEXT = inspect.cleandoc("""
     Your output MUST be a JSON object that conforms to the JSON Schema below. All
     JSON object property names MUST be enclosed in double quotes.
@@ -112,7 +115,13 @@ def render(self, **kwargs: Any) -> str:
         if "user_input" in prompt_kwargs and "user_input" not in self.prompt_variables:
             prompt_to_render += f"\n{prompt_kwargs['user_input']}\n"
 
-        return self._env.from_string(prompt_to_render).render(**prompt_kwargs)
+        # Render the prompt and log it for debugging purposes.
+        rendered_prompt = self._env.from_string(prompt_to_render).render(
+            **prompt_kwargs
+        )
+        logger.debug(f"Rendered prompt:\n{rendered_prompt}")
+
+        return rendered_prompt
 
     def _process_kwargs(self, prompt_values: dict[str, Any]) -> dict[str, Any]:
         """