Spaces:
Paused
Paused
| <html lang="en"> | |
| <head> | |
| <meta http-equiv="Content-Type" content="text/html; charset=utf-8"> | |
| <title>Coverage for tinytroupe/enrichment/tiny_styler.py: 0%</title> | |
| <link rel="icon" sizes="32x32" href="favicon_32_cb_c827f16f.png"> | |
| <link rel="stylesheet" href="style_cb_9ff733b0.css" type="text/css"> | |
| <script src="coverage_html_cb_dd2e7eb5.js" defer></script> | |
| </head> | |
| <body class="pyfile"> | |
| <header> | |
| <div class="content"> | |
| <h1> | |
| <span class="text">Coverage for </span><b>tinytroupe / enrichment / tiny_styler.py</b>: | |
| <span class="pc_cov">0%</span> | |
| </h1> | |
| <aside id="help_panel_wrapper"> | |
| <input id="help_panel_state" type="checkbox"> | |
| <label for="help_panel_state"> | |
| <img id="keyboard_icon" src="keybd_closed_cb_900cfef5.png" alt="Show/hide keyboard shortcuts"> | |
| </label> | |
| <div id="help_panel"> | |
| <p class="legend">Shortcuts on this page</p> | |
| <div class="keyhelp"> | |
| <p> | |
| <kbd>r</kbd> | |
| <kbd>m</kbd> | |
| <kbd>x</kbd> | |
| toggle line displays | |
| </p> | |
| <p> | |
| <kbd>j</kbd> | |
| <kbd>k</kbd> | |
| next/prev highlighted chunk | |
| </p> | |
| <p> | |
| <kbd>0</kbd> (zero) top of page | |
| </p> | |
| <p> | |
| <kbd>1</kbd> (one) first highlighted chunk | |
| </p> | |
| <p> | |
| <kbd>[</kbd> | |
| <kbd>]</kbd> | |
| prev/next file | |
| </p> | |
| <p> | |
| <kbd>u</kbd> up to the index | |
| </p> | |
| <p> | |
| <kbd>?</kbd> show/hide this help | |
| </p> | |
| </div> | |
| </div> | |
| </aside> | |
| <h2> | |
| <span class="text">26 statements </span> | |
| <button type="button" class="run button_toggle_run" value="run" data-shortcut="r" title="Toggle lines run">0<span class="text"> run</span></button> | |
| <button type="button" class="mis show_mis button_toggle_mis" value="mis" data-shortcut="m" title="Toggle lines missing">26<span class="text"> missing</span></button> | |
| <button type="button" class="exc show_exc button_toggle_exc" value="exc" data-shortcut="x" title="Toggle lines excluded">0<span class="text"> excluded</span></button> | |
| </h2> | |
| <p class="text"> | |
| <a id="prevFileLink" class="nav" href="z_37eec20419a3fd46_tiny_enricher_py.html">« prev</a> | |
| <a id="indexLink" class="nav" href="index.html">^ index</a> | |
| <a id="nextFileLink" class="nav" href="z_93fcbcbc09a08040___init___py.html">» next</a> | |
| | |
| <a class="nav" href="https://coverage.readthedocs.io/en/7.13.4">coverage.py v7.13.4</a>, | |
| created at 2026-02-28 17:48 +0000 | |
| </p> | |
| <aside class="hidden"> | |
| <button type="button" class="button_next_chunk" data-shortcut="j"></button> | |
| <button type="button" class="button_prev_chunk" data-shortcut="k"></button> | |
| <button type="button" class="button_top_of_page" data-shortcut="0"></button> | |
| <button type="button" class="button_first_chunk" data-shortcut="1"></button> | |
| <button type="button" class="button_prev_file" data-shortcut="["></button> | |
| <button type="button" class="button_next_file" data-shortcut="]"></button> | |
| <button type="button" class="button_to_index" data-shortcut="u"></button> | |
| <button type="button" class="button_show_hide_help" data-shortcut="?"></button> | |
| </aside> | |
| </div> | |
| </header> | |
| <main id="source"> | |
| <p class="mis show_mis"><span class="n"><a id="t1" href="#t1">1</a></span><span class="t"><span class="key">from</span> <span class="nam">tinytroupe</span><span class="op">.</span><span class="nam">enrichment</span> <span class="key">import</span> <span class="nam">logger</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t2" href="#t2">2</a></span><span class="t"><span class="key">from</span> <span class="nam">tinytroupe</span><span class="op">.</span><span class="nam">utils</span> <span class="key">import</span> <span class="nam">JsonSerializableRegistry</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t3" href="#t3">3</a></span><span class="t"><span class="key">from</span> <span class="nam">tinytroupe</span><span class="op">.</span><span class="nam">utils</span><span class="op">.</span><span class="nam">llm</span> <span class="key">import</span> <span class="nam">LLMChat</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t4" href="#t4">4</a></span><span class="t"><span class="key">import</span> <span class="nam">tinytroupe</span><span class="op">.</span><span class="nam">utils</span> <span class="key">as</span> <span class="nam">utils</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t5" href="#t5">5</a></span><span class="t"> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t6" href="#t6">6</a></span><span class="t"> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t7" href="#t7">7</a></span><span class="t"><span class="key">class</span> <span class="nam">TinyStyler</span><span class="op">(</span><span class="nam">JsonSerializableRegistry</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t8" href="#t8">8</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t9" href="#t9">9</a></span><span class="t"><span class="str"> A class for applying a specified writing or speaking style to content while preserving</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t10" href="#t10">10</a></span><span class="t"><span class="str"> the original information.</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t11" href="#t11">11</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t12" href="#t12">12</a></span><span class="t"> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t13" href="#t13">13</a></span><span class="t"> <span class="key">def</span> <span class="nam">__init__</span><span class="op">(</span><span class="nam">self</span><span class="op">,</span> <span class="nam">use_past_results_in_context</span><span class="op">=</span><span class="key">False</span><span class="op">)</span> <span class="op">-></span> <span class="key">None</span><span class="op">:</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t14" href="#t14">14</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t15" href="#t15">15</a></span><span class="t"><span class="str"> Initialize the TinyStyler.</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t16" href="#t16">16</a></span><span class="t"> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t17" href="#t17">17</a></span><span class="t"><span class="str"> Args:</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t18" href="#t18">18</a></span><span class="t"><span class="str"> use_past_results_in_context (bool): Whether to use past styling results in the context.</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t19" href="#t19">19</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t20" href="#t20">20</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">use_past_results_in_context</span> <span class="op">=</span> <span class="nam">use_past_results_in_context</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t21" href="#t21">21</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">context_cache</span> <span class="op">=</span> <span class="op">[</span><span class="op">]</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t22" href="#t22">22</a></span><span class="t"> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t23" href="#t23">23</a></span><span class="t"> <span class="key">def</span> <span class="nam">apply_style</span><span class="op">(</span><span class="nam">self</span><span class="op">,</span> <span class="nam">content</span><span class="op">:</span> <span class="nam">str</span><span class="op">,</span> <span class="nam">style</span><span class="op">:</span> <span class="nam">str</span><span class="op">,</span> <span class="nam">content_type</span><span class="op">:</span> <span class="nam">str</span> <span class="op">=</span> <span class="key">None</span><span class="op">,</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t24" href="#t24">24</a></span><span class="t"> <span class="nam">context_info</span><span class="op">:</span> <span class="nam">str</span> <span class="op">=</span> <span class="str">""</span><span class="op">,</span> <span class="nam">context_cache</span><span class="op">:</span> <span class="nam">list</span> <span class="op">=</span> <span class="key">None</span><span class="op">,</span> <span class="nam">verbose</span><span class="op">:</span> <span class="nam">bool</span> <span class="op">=</span> <span class="key">False</span><span class="op">,</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t25" href="#t25">25</a></span><span class="t"> <span class="nam">temperature</span><span class="op">:</span> <span class="nam">float</span> <span class="op">=</span> <span class="num">0.7</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t26" href="#t26">26</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t27" href="#t27">27</a></span><span class="t"><span class="str"> Apply a specified style to the content while preserving all the original information.</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t28" href="#t28">28</a></span><span class="t"> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t29" href="#t29">29</a></span><span class="t"><span class="str"> Args:</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t30" href="#t30">30</a></span><span class="t"><span class="str"> content (str): The content to style.</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t31" href="#t31">31</a></span><span class="t"><span class="str"> style (str): The style to apply (e.g., "professional", "casual", "technical", etc.).</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t32" href="#t32">32</a></span><span class="t"><span class="str"> content_type (str, optional): The type of content (e.g., "email", "report", "conversation").</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t33" href="#t33">33</a></span><span class="t"><span class="str"> context_info (str, optional): Additional context information.</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t34" href="#t34">34</a></span><span class="t"><span class="str"> context_cache (list, optional): Previous styling results to use as context.</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t35" href="#t35">35</a></span><span class="t"><span class="str"> verbose (bool, optional): Whether to print debug information.</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t36" href="#t36">36</a></span><span class="t"><span class="str"> temperature (float, optional): The temperature to use for the LLM generation.</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t37" href="#t37">37</a></span><span class="t"> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t38" href="#t38">38</a></span><span class="t"><span class="str"> Returns:</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t39" href="#t39">39</a></span><span class="t"><span class="str"> str: The styled content.</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t40" href="#t40">40</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t41" href="#t41">41</a></span><span class="t"> <span class="key">if</span> <span class="nam">context_cache</span> <span class="key">is</span> <span class="key">None</span> <span class="key">and</span> <span class="nam">self</span><span class="op">.</span><span class="nam">use_past_results_in_context</span><span class="op">:</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t42" href="#t42">42</a></span><span class="t"> <span class="nam">context_cache</span> <span class="op">=</span> <span class="nam">self</span><span class="op">.</span><span class="nam">context_cache</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t43" href="#t43">43</a></span><span class="t"> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t44" href="#t44">44</a></span><span class="t"> <span class="nam">rendering_configs</span> <span class="op">=</span> <span class="op">{</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t45" href="#t45">45</a></span><span class="t"> <span class="str">"content"</span><span class="op">:</span> <span class="nam">content</span><span class="op">,</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t46" href="#t46">46</a></span><span class="t"> <span class="str">"style"</span><span class="op">:</span> <span class="nam">style</span><span class="op">,</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t47" href="#t47">47</a></span><span class="t"> <span class="str">"content_type"</span><span class="op">:</span> <span class="nam">content_type</span><span class="op">,</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t48" href="#t48">48</a></span><span class="t"> <span class="str">"context_info"</span><span class="op">:</span> <span class="nam">context_info</span><span class="op">,</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t49" href="#t49">49</a></span><span class="t"> <span class="str">"context_cache"</span><span class="op">:</span> <span class="nam">context_cache</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t50" href="#t50">50</a></span><span class="t"> <span class="op">}</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t51" href="#t51">51</a></span><span class="t"> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t52" href="#t52">52</a></span><span class="t"> <span class="com"># Initialize the LLMChat with appropriate templates</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t53" href="#t53">53</a></span><span class="t"> <span class="nam">chat</span> <span class="op">=</span> <span class="nam">LLMChat</span><span class="op">(</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t54" href="#t54">54</a></span><span class="t"> <span class="nam">system_template_name</span><span class="op">=</span><span class="str">"styler.system.mustache"</span><span class="op">,</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t55" href="#t55">55</a></span><span class="t"> <span class="nam">user_template_name</span><span class="op">=</span><span class="str">"styler.user.mustache"</span><span class="op">,</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t56" href="#t56">56</a></span><span class="t"> <span class="nam">base_module_folder</span><span class="op">=</span><span class="str">"enrichment"</span><span class="op">,</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t57" href="#t57">57</a></span><span class="t"> <span class="nam">temperature</span><span class="op">=</span><span class="nam">temperature</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t58" href="#t58">58</a></span><span class="t"> <span class="op">)</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t59" href="#t59">59</a></span><span class="t"> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t60" href="#t60">60</a></span><span class="t"> <span class="com"># Call the model and get the response</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t61" href="#t61">61</a></span><span class="t"> <span class="nam">result</span> <span class="op">=</span> <span class="nam">chat</span><span class="op">.</span><span class="nam">call</span><span class="op">(</span><span class="op">**</span><span class="nam">rendering_configs</span><span class="op">)</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t62" href="#t62">62</a></span><span class="t"> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t63" href="#t63">63</a></span><span class="t"> <span class="nam">debug_msg</span> <span class="op">=</span> <span class="fst">f"</span><span class="fst">Styling result: </span><span class="op">{</span><span class="nam">result</span><span class="op">}</span><span class="fst">"</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t64" href="#t64">64</a></span><span class="t"> <span class="nam">logger</span><span class="op">.</span><span class="nam">debug</span><span class="op">(</span><span class="nam">debug_msg</span><span class="op">)</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t65" href="#t65">65</a></span><span class="t"> <span class="key">if</span> <span class="nam">verbose</span><span class="op">:</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t66" href="#t66">66</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="nam">debug_msg</span><span class="op">)</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t67" href="#t67">67</a></span><span class="t"> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t68" href="#t68">68</a></span><span class="t"> <span class="com"># Extract the styled content from code blocks if present</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t69" href="#t69">69</a></span><span class="t"> <span class="key">if</span> <span class="nam">result</span> <span class="key">is</span> <span class="key">not</span> <span class="key">None</span><span class="op">:</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t70" href="#t70">70</a></span><span class="t"> <span class="nam">styled_content</span> <span class="op">=</span> <span class="nam">utils</span><span class="op">.</span><span class="nam">extract_code_block</span><span class="op">(</span><span class="nam">result</span><span class="op">)</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t71" href="#t71">71</a></span><span class="t"> <span class="com"># If no code block was found, use the raw result</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t72" href="#t72">72</a></span><span class="t"> <span class="key">if</span> <span class="key">not</span> <span class="nam">styled_content</span><span class="op">:</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t73" href="#t73">73</a></span><span class="t"> <span class="nam">styled_content</span> <span class="op">=</span> <span class="nam">result</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t74" href="#t74">74</a></span><span class="t"> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t75" href="#t75">75</a></span><span class="t"> <span class="com"># Add to context cache if enabled</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t76" href="#t76">76</a></span><span class="t"> <span class="key">if</span> <span class="nam">self</span><span class="op">.</span><span class="nam">use_past_results_in_context</span><span class="op">:</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t77" href="#t77">77</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">context_cache</span><span class="op">.</span><span class="nam">append</span><span class="op">(</span><span class="op">{</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t78" href="#t78">78</a></span><span class="t"> <span class="str">"original"</span><span class="op">:</span> <span class="nam">content</span><span class="op">,</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t79" href="#t79">79</a></span><span class="t"> <span class="str">"style"</span><span class="op">:</span> <span class="nam">style</span><span class="op">,</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t80" href="#t80">80</a></span><span class="t"> <span class="str">"styled"</span><span class="op">:</span> <span class="nam">styled_content</span> </span><span class="r"></span></p> | |
| <p class="mis mis2 show_mis"><span class="n"><a id="t81" href="#t81">81</a></span><span class="t"> <span class="op">}</span><span class="op">)</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t82" href="#t82">82</a></span><span class="t"> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t83" href="#t83">83</a></span><span class="t"> <span class="key">return</span> <span class="nam">styled_content</span> </span><span class="r"></span></p> | |
| <p class="pln"><span class="n"><a id="t84" href="#t84">84</a></span><span class="t"> <span class="key">else</span><span class="op">:</span> </span><span class="r"></span></p> | |
| <p class="mis show_mis"><span class="n"><a id="t85" href="#t85">85</a></span><span class="t"> <span class="key">return</span> <span class="key">None</span> </span><span class="r"></span></p> | |
| </main> | |
| <footer> | |
| <div class="content"> | |
| <p> | |
| <a class="nav" href="z_37eec20419a3fd46_tiny_enricher_py.html">« prev</a> | |
| <a class="nav" href="index.html">^ index</a> | |
| <a class="nav" href="z_93fcbcbc09a08040___init___py.html">» next</a> | |
| | |
| <a class="nav" href="https://coverage.readthedocs.io/en/7.13.4">coverage.py v7.13.4</a>, | |
| created at 2026-02-28 17:48 +0000 | |
| </p> | |
| </div> | |
| </footer> | |
| </body> | |
| </html> | |