Hugging Face's logo

Spaces:
AUXteam
/
tiny_factory
Paused

App Files Community
tiny_factory / htmlcov /z_13b615ba9b634dc5_tiny_story_py.html
AUXteam's picture
AUXteam
Upload folder using huggingface_hub
7a0719c verified
raw
history blame contribute delete
40.9 kB
<!DOCTYPE html>
<html lang="en">
<head>
 <meta http-equiv="Content-Type" content="text/html; charset=utf-8">
 <title>Coverage for tinytroupe/steering/tiny_story.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&#8201;/&#8201;steering&#8201;/&#8201;tiny_story.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>
 &nbsp; toggle line displays
 </p>
 <p>
 <kbd>j</kbd>
 <kbd>k</kbd>
 &nbsp; next/prev highlighted chunk
 </p>
 <p>
 <kbd>0</kbd> &nbsp; (zero) top of page
 </p>
 <p>
 <kbd>1</kbd> &nbsp; (one) first highlighted chunk
 </p>
 <p>
 <kbd>[</kbd>
 <kbd>]</kbd>
 &nbsp; prev/next file
 </p>
 <p>
 <kbd>u</kbd> &nbsp; up to the index
 </p>
 <p>
 <kbd>?</kbd> &nbsp; show/hide this help
 </p>
 </div>
 </div>
 </aside>
 <h2>
 <span class="text">42 statements &nbsp;</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">42<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_13b615ba9b634dc5_intervention_py.html">&#xab; prev</a> &nbsp; &nbsp;
 <a id="indexLink" class="nav" href="index.html">&Hat; index</a> &nbsp; &nbsp;
 <a id="nextFileLink" class="nav" href="z_8ff660fc2daa80e3___init___py.html">&#xbb; next</a>
 &nbsp; &nbsp; &nbsp;
 <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">typing</span> <span class="key">import</span> <span class="nam">List</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t2" href="#t2">2</a></span><span class="t">&nbsp;</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">extraction</span> <span class="key">import</span> <span class="nam">logger</span>&nbsp;</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">from</span> <span class="nam">tinytroupe</span><span class="op">.</span><span class="nam">agent</span> <span class="key">import</span> <span class="nam">TinyPerson</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t5" href="#t5">5</a></span><span class="t"><span class="key">from</span> <span class="nam">tinytroupe</span><span class="op">.</span><span class="nam">environment</span> <span class="key">import</span> <span class="nam">TinyWorld</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t6" href="#t6">6</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>&nbsp;</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">from</span> <span class="nam">tinytroupe</span> <span class="key">import</span> <span class="nam">openai_utils</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t8" href="#t8">8</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t9" href="#t9">9</a></span><span class="t"><span class="key">class</span> <span class="nam">TinyStory</span><span class="op">:</span>&nbsp;</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">"""</span>&nbsp;</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"> Every simulation tells a story. This class provides helper mechanisms to help with crafting appropriate stories in TinyTroupe.</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t12" href="#t12">12</a></span><span class="t"><span class="str"> """</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t13" href="#t13">13</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t14" href="#t14">14</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t15" href="#t15">15</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">environment</span><span class="op">:</span><span class="nam">TinyWorld</span><span class="op">=</span><span class="key">None</span><span class="op">,</span> <span class="nam">agent</span><span class="op">:</span><span class="nam">TinyPerson</span><span class="op">=</span><span class="key">None</span><span class="op">,</span> <span class="nam">purpose</span><span class="op">:</span><span class="nam">str</span><span class="op">=</span><span class="str">"Be a realistic simulation."</span><span class="op">,</span> <span class="nam">context</span><span class="op">:</span><span class="nam">str</span><span class="op">=</span><span class="str">""</span><span class="op">,</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t16" href="#t16">16</a></span><span class="t"> <span class="nam">first_n</span><span class="op">=</span><span class="num">10</span><span class="op">,</span> <span class="nam">last_n</span><span class="op">=</span><span class="num">20</span><span class="op">,</span> <span class="nam">include_omission_info</span><span class="op">:</span><span class="nam">bool</span><span class="op">=</span><span class="key">True</span><span class="op">)</span> <span class="op">-></span> <span class="key">None</span><span class="op">:</span>&nbsp;</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">"""</span>&nbsp;</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"> Initialize the story. The story can be about an environment or an agent. It also has a purpose, which</span>&nbsp;</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"> is used to guide the story generation. Stories are aware that they are related to simulations, so one can</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t20" href="#t20">20</a></span><span class="t"><span class="str"> specify simulation-related purposes.</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t21" href="#t21">21</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t22" href="#t22">22</a></span><span class="t"><span class="str"> Args:</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t23" href="#t23">23</a></span><span class="t"><span class="str"> environment (TinyWorld, optional): The environment in which the story takes place. Defaults to None.</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t24" href="#t24">24</a></span><span class="t"><span class="str"> agent (TinyPerson, optional): The agent in the story. Defaults to None.</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t25" href="#t25">25</a></span><span class="t"><span class="str"> purpose (str, optional): The purpose of the story. Defaults to "Be a realistic simulation.".</span>&nbsp;</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"> context (str, optional): The current story context. Defaults to "". The actual story will be appended to this context.</span>&nbsp;</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"> first_n (int, optional): The number of first interactions to include in the story. Defaults to 10.</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t28" href="#t28">28</a></span><span class="t"><span class="str"> last_n (int, optional): The number of last interactions to include in the story. Defaults to 20.</span>&nbsp;</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"> include_omission_info (bool, optional): Whether to include information about omitted interactions. Defaults to True.</span>&nbsp;</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"> """</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t31" href="#t31">31</a></span><span class="t">&nbsp;</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="com"># exactly one of these must be provided</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t33" href="#t33">33</a></span><span class="t"> <span class="key">if</span> <span class="nam">environment</span> <span class="key">and</span> <span class="nam">agent</span><span class="op">:</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t34" href="#t34">34</a></span><span class="t"> <span class="key">raise</span> <span class="nam">Exception</span><span class="op">(</span><span class="str">"Either 'environment' or 'agent' should be provided, not both"</span><span class="op">)</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t35" href="#t35">35</a></span><span class="t"> <span class="key">if</span> <span class="key">not</span> <span class="op">(</span><span class="nam">environment</span> <span class="key">or</span> <span class="nam">agent</span><span class="op">)</span><span class="op">:</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t36" href="#t36">36</a></span><span class="t"> <span class="key">raise</span> <span class="nam">Exception</span><span class="op">(</span><span class="str">"At least one of the parameters should be provided"</span><span class="op">)</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t37" href="#t37">37</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t38" href="#t38">38</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">environment</span> <span class="op">=</span> <span class="nam">environment</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t39" href="#t39">39</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">agent</span> <span class="op">=</span> <span class="nam">agent</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t40" href="#t40">40</a></span><span class="t">&nbsp;</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="nam">self</span><span class="op">.</span><span class="nam">purpose</span> <span class="op">=</span> <span class="nam">purpose</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t42" href="#t42">42</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t43" href="#t43">43</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">current_story</span> <span class="op">=</span> <span class="nam">context</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t44" href="#t44">44</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t45" href="#t45">45</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">first_n</span> <span class="op">=</span> <span class="nam">first_n</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t46" href="#t46">46</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">last_n</span> <span class="op">=</span> <span class="nam">last_n</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t47" href="#t47">47</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">include_omission_info</span> <span class="op">=</span> <span class="nam">include_omission_info</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t48" href="#t48">48</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t49" href="#t49">49</a></span><span class="t"> <span class="key">def</span> <span class="nam">start_story</span><span class="op">(</span><span class="nam">self</span><span class="op">,</span> <span class="nam">requirements</span><span class="op">=</span><span class="str">"Start some interesting story about the agents."</span><span class="op">,</span> <span class="nam">number_of_words</span><span class="op">:</span><span class="nam">int</span><span class="op">=</span><span class="num">100</span><span class="op">,</span> <span class="nam">include_plot_twist</span><span class="op">:</span><span class="nam">bool</span><span class="op">=</span><span class="key">False</span><span class="op">)</span> <span class="op">-></span> <span class="nam">str</span><span class="op">:</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t50" href="#t50">50</a></span><span class="t"> <span class="str">"""</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t51" href="#t51">51</a></span><span class="t"><span class="str"> Start a new story.</span>&nbsp;</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="str"> """</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t53" href="#t53">53</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t54" href="#t54">54</a></span><span class="t"> <span class="nam">rendering_configs</span> <span class="op">=</span> <span class="op">{</span>&nbsp;</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="str">"purpose"</span><span class="op">:</span> <span class="nam">self</span><span class="op">.</span><span class="nam">purpose</span><span class="op">,</span>&nbsp;</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="str">"requirements"</span><span class="op">:</span> <span class="nam">requirements</span><span class="op">,</span>&nbsp;</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="str">"current_simulation_trace"</span><span class="op">:</span> <span class="nam">self</span><span class="op">.</span><span class="nam">_current_story</span><span class="op">(</span><span class="op">)</span><span class="op">,</span>&nbsp;</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="str">"number_of_words"</span><span class="op">:</span> <span class="nam">number_of_words</span><span class="op">,</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t59" href="#t59">59</a></span><span class="t"> <span class="str">"include_plot_twist"</span><span class="op">:</span> <span class="nam">include_plot_twist</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t60" href="#t60">60</a></span><span class="t"> <span class="op">}</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t61" href="#t61">61</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t62" href="#t62">62</a></span><span class="t"> <span class="nam">messages</span> <span class="op">=</span> <span class="nam">utils</span><span class="op">.</span><span class="nam">compose_initial_LLM_messages_with_templates</span><span class="op">(</span><span class="str">"story.start.system.mustache"</span><span class="op">,</span> <span class="str">"story.start.user.mustache"</span><span class="op">,</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t63" href="#t63">63</a></span><span class="t"> <span class="nam">base_module_folder</span><span class="op">=</span><span class="str">"steering"</span><span class="op">,</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t64" href="#t64">64</a></span><span class="t"> <span class="nam">rendering_configs</span><span class="op">=</span><span class="nam">rendering_configs</span><span class="op">)</span>&nbsp;</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="nam">next_message</span> <span class="op">=</span> <span class="nam">openai_utils</span><span class="op">.</span><span class="nam">client</span><span class="op">(</span><span class="op">)</span><span class="op">.</span><span class="nam">send_message</span><span class="op">(</span><span class="nam">messages</span><span class="op">,</span> <span class="nam">temperature</span><span class="op">=</span><span class="num">1.5</span><span class="op">)</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t66" href="#t66">66</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t67" href="#t67">67</a></span><span class="t"> <span class="nam">start</span> <span class="op">=</span> <span class="nam">next_message</span><span class="op">[</span><span class="str">"content"</span><span class="op">]</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t68" href="#t68">68</a></span><span class="t">&nbsp;</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="nam">self</span><span class="op">.</span><span class="nam">current_story</span> <span class="op">+=</span> <span class="nam">utils</span><span class="op">.</span><span class="nam">dedent</span><span class="op">(</span><span class="xx">\</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t70" href="#t70">70</a></span><span class="t"> <span class="fst">f"""</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t71" href="#t71">71</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t72" href="#t72">72</a></span><span class="t"><span class="fst"> ## The story begins</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t73" href="#t73">73</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t74" href="#t74">74</a></span><span class="t"><span class="fst"> </span><span class="op">{</span><span class="nam">start</span><span class="op">}</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t75" href="#t75">75</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t76" href="#t76">76</a></span><span class="t"><span class="fst"> </span><span class="fst">"""</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t77" href="#t77">77</a></span><span class="t"> <span class="op">)</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t78" href="#t78">78</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t79" href="#t79">79</a></span><span class="t"> <span class="key">return</span> <span class="nam">start</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t80" href="#t80">80</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t81" href="#t81">81</a></span><span class="t"> <span class="key">def</span> <span class="nam">continue_story</span><span class="op">(</span><span class="nam">self</span><span class="op">,</span> <span class="nam">requirements</span><span class="op">=</span><span class="str">"Continue the story in an interesting way."</span><span class="op">,</span> <span class="nam">number_of_words</span><span class="op">:</span><span class="nam">int</span><span class="op">=</span><span class="num">100</span><span class="op">,</span> <span class="nam">include_plot_twist</span><span class="op">:</span><span class="nam">bool</span><span class="op">=</span><span class="key">False</span><span class="op">)</span> <span class="op">-></span> <span class="nam">str</span><span class="op">:</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t82" href="#t82">82</a></span><span class="t"> <span class="str">"""</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t83" href="#t83">83</a></span><span class="t"><span class="str"> Propose a continuation of the story.</span>&nbsp;</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="str"> """</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t85" href="#t85">85</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t86" href="#t86">86</a></span><span class="t"> <span class="nam">rendering_configs</span> <span class="op">=</span> <span class="op">{</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t87" href="#t87">87</a></span><span class="t"> <span class="str">"purpose"</span><span class="op">:</span> <span class="nam">self</span><span class="op">.</span><span class="nam">purpose</span><span class="op">,</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t88" href="#t88">88</a></span><span class="t"> <span class="str">"requirements"</span><span class="op">:</span> <span class="nam">requirements</span><span class="op">,</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t89" href="#t89">89</a></span><span class="t"> <span class="str">"current_simulation_trace"</span><span class="op">:</span> <span class="nam">self</span><span class="op">.</span><span class="nam">_current_story</span><span class="op">(</span><span class="op">)</span><span class="op">,</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t90" href="#t90">90</a></span><span class="t"> <span class="str">"number_of_words"</span><span class="op">:</span> <span class="nam">number_of_words</span><span class="op">,</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t91" href="#t91">91</a></span><span class="t"> <span class="str">"include_plot_twist"</span><span class="op">:</span> <span class="nam">include_plot_twist</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t92" href="#t92">92</a></span><span class="t"> <span class="op">}</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t93" href="#t93">93</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t94" href="#t94">94</a></span><span class="t"> <span class="nam">messages</span> <span class="op">=</span> <span class="nam">utils</span><span class="op">.</span><span class="nam">compose_initial_LLM_messages_with_templates</span><span class="op">(</span><span class="str">"story.continuation.system.mustache"</span><span class="op">,</span> <span class="str">"story.continuation.user.mustache"</span><span class="op">,</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t95" href="#t95">95</a></span><span class="t"> <span class="nam">base_module_folder</span><span class="op">=</span><span class="str">"steering"</span><span class="op">,</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t96" href="#t96">96</a></span><span class="t"> <span class="nam">rendering_configs</span><span class="op">=</span><span class="nam">rendering_configs</span><span class="op">)</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t97" href="#t97">97</a></span><span class="t"> <span class="nam">next_message</span> <span class="op">=</span> <span class="nam">openai_utils</span><span class="op">.</span><span class="nam">client</span><span class="op">(</span><span class="op">)</span><span class="op">.</span><span class="nam">send_message</span><span class="op">(</span><span class="nam">messages</span><span class="op">)</span><span class="com">#, temperature=1.5)</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t98" href="#t98">98</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t99" href="#t99">99</a></span><span class="t"> <span class="nam">continuation</span> <span class="op">=</span> <span class="nam">next_message</span><span class="op">[</span><span class="str">"content"</span><span class="op">]</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t100" href="#t100">100</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t101" href="#t101">101</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">current_story</span> <span class="op">+=</span> <span class="nam">utils</span><span class="op">.</span><span class="nam">dedent</span><span class="op">(</span><span class="xx">\</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t102" href="#t102">102</a></span><span class="t"> <span class="fst">f"""</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t103" href="#t103">103</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t104" href="#t104">104</a></span><span class="t"><span class="fst"> ## The story continues</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t105" href="#t105">105</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t106" href="#t106">106</a></span><span class="t"><span class="fst"> </span><span class="op">{</span><span class="nam">continuation</span><span class="op">}</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t107" href="#t107">107</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t108" href="#t108">108</a></span><span class="t"><span class="fst"> </span><span class="fst">"""</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t109" href="#t109">109</a></span><span class="t"> <span class="op">)</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t110" href="#t110">110</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t111" href="#t111">111</a></span><span class="t"> <span class="key">return</span> <span class="nam">continuation</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t112" href="#t112">112</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t113" href="#t113">113</a></span><span class="t"> <span class="key">def</span> <span class="nam">_current_story</span><span class="op">(</span><span class="nam">self</span><span class="op">)</span> <span class="op">-></span> <span class="nam">str</span><span class="op">:</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t114" href="#t114">114</a></span><span class="t"> <span class="str">"""</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t115" href="#t115">115</a></span><span class="t"><span class="str"> Get the current story.</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t116" href="#t116">116</a></span><span class="t"><span class="str"> """</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t117" href="#t117">117</a></span><span class="t"> <span class="nam">interaction_history</span> <span class="op">=</span> <span class="str">""</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t118" href="#t118">118</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t119" href="#t119">119</a></span><span class="t"> <span class="key">if</span> <span class="nam">self</span><span class="op">.</span><span class="nam">agent</span> <span class="key">is</span> <span class="key">not</span> <span class="key">None</span><span class="op">:</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t120" href="#t120">120</a></span><span class="t"> <span class="nam">interaction_history</span> <span class="op">+=</span> <span class="nam">self</span><span class="op">.</span><span class="nam">agent</span><span class="op">.</span><span class="nam">pretty_current_interactions</span><span class="op">(</span><span class="nam">first_n</span><span class="op">=</span><span class="nam">self</span><span class="op">.</span><span class="nam">first_n</span><span class="op">,</span> <span class="nam">last_n</span><span class="op">=</span><span class="nam">self</span><span class="op">.</span><span class="nam">last_n</span><span class="op">,</span> <span class="nam">include_omission_info</span><span class="op">=</span><span class="nam">self</span><span class="op">.</span><span class="nam">include_omission_info</span><span class="op">)</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t121" href="#t121">121</a></span><span class="t"> <span class="key">elif</span> <span class="nam">self</span><span class="op">.</span><span class="nam">environment</span> <span class="key">is</span> <span class="key">not</span> <span class="key">None</span><span class="op">:</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t122" href="#t122">122</a></span><span class="t"> <span class="nam">interaction_history</span> <span class="op">+=</span> <span class="nam">self</span><span class="op">.</span><span class="nam">environment</span><span class="op">.</span><span class="nam">pretty_current_interactions</span><span class="op">(</span><span class="nam">first_n</span><span class="op">=</span><span class="nam">self</span><span class="op">.</span><span class="nam">first_n</span><span class="op">,</span> <span class="nam">last_n</span><span class="op">=</span><span class="nam">self</span><span class="op">.</span><span class="nam">last_n</span><span class="op">,</span> <span class="nam">include_omission_info</span><span class="op">=</span><span class="nam">self</span><span class="op">.</span><span class="nam">include_omission_info</span><span class="op">)</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t123" href="#t123">123</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t124" href="#t124">124</a></span><span class="t"> <span class="nam">tmp_current_story</span> <span class="op">=</span> <span class="nam">self</span><span class="op">.</span><span class="nam">current_story</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t125" href="#t125">125</a></span><span class="t"> <span class="nam">tmp_current_story</span> <span class="op">+=</span> <span class="nam">utils</span><span class="op">.</span><span class="nam">dedent</span><span class="op">(</span><span class="xx">\</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t126" href="#t126">126</a></span><span class="t"> <span class="fst">f"""</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t127" href="#t127">127</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t128" href="#t128">128</a></span><span class="t"><span class="fst"> ## New simulation interactions to consider</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t129" href="#t129">129</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t130" href="#t130">130</a></span><span class="t"><span class="fst"> </span><span class="op">{</span><span class="nam">interaction_history</span><span class="op">}</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t131" href="#t131">131</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t132" href="#t132">132</a></span><span class="t"><span class="fst"> </span><span class="fst">"""</span>&nbsp;</span><span class="r"></span></p>
 <p class="mis mis2 show_mis"><span class="n"><a id="t133" href="#t133">133</a></span><span class="t"> <span class="op">)</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t134" href="#t134">134</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="mis show_mis"><span class="n"><a id="t135" href="#t135">135</a></span><span class="t"> <span class="key">return</span> <span class="nam">tmp_current_story</span>&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t136" href="#t136">136</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t137" href="#t137">137</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
 <p class="pln"><span class="n"><a id="t138" href="#t138">138</a></span><span class="t">&nbsp;</span><span class="r"></span></p>
</main>
<footer>
 <div class="content">
 <p>
 <a class="nav" href="z_13b615ba9b634dc5_intervention_py.html">&#xab; prev</a> &nbsp; &nbsp;
 <a class="nav" href="index.html">&Hat; index</a> &nbsp; &nbsp;
 <a class="nav" href="z_8ff660fc2daa80e3___init___py.html">&#xbb; next</a>
 &nbsp; &nbsp; &nbsp;
 <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>