Skip to content

Commit be8f8db

Browse files
khuyentran1401khuyentran1401
committed
add timegpt
1 parent 8298c12 commit be8f8db

10 files changed

+768
-33
lines changed

‎Chapter5/llm.ipynb

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -378,7 +378,7 @@
378378
},
379379
{
380380
"cell_type": "code",
381-
"execution_count": 117,
381+
"execution_count": 131,
382382
"id": "620f3cf4",
383383
"metadata": {
384384
"tags": [
@@ -445,17 +445,17 @@
445445
},
446446
{
447447
"cell_type": "code",
448-
"execution_count": 129,
448+
"execution_count": 137,
449449
"id": "da316438",
450450
"metadata": {},
451451
"outputs": [
452452
{
453453
"data": {
454454
"text/plain": [
455-
"'The price for apple is 0.75.'"
455+
"'The price for banana is 0.5.'"
456456
]
457457
},
458-
"execution_count": 129,
458+
"execution_count": 137,
459459
"metadata": {},
460460
"output_type": "execute_result"
461461
}
@@ -468,22 +468,22 @@
468468
"conn = sqlite3.connect(\"grocery.db\")\n",
469469
"\n",
470470
"\n",
471-
"def get_item_info(table_name: str, item_name: str, info: str) -> dict:\n",
472-
" \"\"\"Get `info` from the `table_name` database table based on `name`.\"\"\"\n",
471+
"def get_item_info(table: str, item: str, info: str) -> dict:\n",
472+
" \"\"\"Get `info` from the `table` table based on `item`.\"\"\"\n",
473473
" cursor = conn.cursor()\n",
474474
" try:\n",
475475
" row = cursor.execute(\n",
476-
" f\"SELECT {info} FROM {table_name} WHERE name = ?\", (item_name,)\n",
476+
" f\"SELECT {info} FROM {table} WHERE name = ?\", (item,)\n",
477477
" ).fetchone()\n",
478-
" return f\"The {info} for {item_name} is {row[0]}.\"\n",
478+
" return f\"The {info} for {item} is {row[0]}.\"\n",
479479
" except TypeError:\n",
480-
" return f\"Sorry but {item_name} doesn't exist in the database.\"\n",
480+
" return f\"Sorry but {item} doesn't exist in the database.\"\n",
481481
"\n",
482482
"\n",
483483
"class GroceryItemQuery(OpenAICall):\n",
484484
" prompt_template = \"\"\"\n",
485485
" SYSTEM:\n",
486-
" Your task is properly query a database given a user's input.\n",
486+
" Your task is to query a database given a user's input.\n",
487487
"\n",
488488
" USER:\n",
489489
" {input}\n",
@@ -492,7 +492,7 @@
492492
" call_params = OpenAICallParams(tools=[get_item_info])\n",
493493
"\n",
494494
"\n",
495-
"text = \"What's the price for apple in the grocery_items table?\"\n",
495+
"text = \"What's the price for banana in the grocery_items table?\"\n",
496496
"query_tool = GroceryItemQuery(input=text).call().tool\n",
497497
"result = query_tool.fn(**query_tool.args)\n",
498498
"result"
@@ -523,7 +523,7 @@
523523
"name": "python",
524524
"nbconvert_exporter": "python",
525525
"pygments_lexer": "ipython3",
526-
"version": "3.11.6"
526+
"version": "3.11.4"
527527
}
528528
},
529529
"nbformat": 4,

‎Chapter5/time_series.ipynb

Lines changed: 274 additions & 0 deletions
Large diffs are not rendered by default.

‎docs/Chapter2/typing.html

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -234,6 +234,7 @@
234234
<li class="toctree-l2"><a class="reference internal" href="dataclasses.html">3.7. Data Classes</a></li>
235235
<li class="toctree-l2 current active"><a class="current reference internal" href="#">3.8. Typing</a></li>
236236
<li class="toctree-l2"><a class="reference internal" href="pathlib.html">3.9. pathlib</a></li>
237+
<li class="toctree-l2"><a class="reference internal" href="pydantic.html">3.10. Pydantic</a></li>
237238
</ul>
238239
</li>
239240
<li class="toctree-l1 has-children"><a class="reference internal" href="../Chapter3/Chapter3.html">4. Pandas</a><input class="toctree-checkbox" id="toctree-checkbox-4" name="toctree-checkbox-4" type="checkbox"/><label class="toctree-toggle" for="toctree-checkbox-4"><i class="fa-solid fa-chevron-down"></i></label><ul>

‎docs/Chapter5/llm.html

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -817,22 +817,22 @@ <h2><span class="section-number">6.16.4. </span>Maximize Accuracy and Relevance
817817
<span class="n">conn</span> <span class="o">=</span> <span class="n">sqlite3</span><span class="o">.</span><span class="n">connect</span><span class="p">(</span><span class="s2">&quot;grocery.db&quot;</span><span class="p">)</span>
818818

819819

820-
<span class="k">def</span> <span class="nf">get_item_info</span><span class="p">(</span><span class="n">table_name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">item_name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">info</span><span class="p">:</span> <span class="nb">str</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">dict</span><span class="p">:</span>
821-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;Get `info` from the `table_name` database table based on `name`.&quot;&quot;&quot;</span>
820+
<span class="k">def</span> <span class="nf">get_item_info</span><span class="p">(</span><span class="n">table</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">item</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">info</span><span class="p">:</span> <span class="nb">str</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">dict</span><span class="p">:</span>
821+
<span class="w"> </span><span class="sd">&quot;&quot;&quot;Get `info` from the `table` table based on `item`.&quot;&quot;&quot;</span>
822822
<span class="n">cursor</span> <span class="o">=</span> <span class="n">conn</span><span class="o">.</span><span class="n">cursor</span><span class="p">()</span>
823823
<span class="k">try</span><span class="p">:</span>
824824
<span class="n">row</span> <span class="o">=</span> <span class="n">cursor</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span>
825-
<span class="sa">f</span><span class="s2">&quot;SELECT </span><span class="si">{</span><span class="n">info</span><span class="si">}</span><span class="s2"> FROM </span><span class="si">{</span><span class="n">table_name</span><span class="si">}</span><span class="s2"> WHERE name = ?&quot;</span><span class="p">,</span> <span class="p">(</span><span class="n">item_name</span><span class="p">,)</span>
825+
<span class="sa">f</span><span class="s2">&quot;SELECT </span><span class="si">{</span><span class="n">info</span><span class="si">}</span><span class="s2"> FROM </span><span class="si">{</span><span class="n">table</span><span class="si">}</span><span class="s2"> WHERE name = ?&quot;</span><span class="p">,</span> <span class="p">(</span><span class="n">item</span><span class="p">,)</span>
826826
<span class="p">)</span><span class="o">.</span><span class="n">fetchone</span><span class="p">()</span>
827-
<span class="k">return</span> <span class="sa">f</span><span class="s2">&quot;The </span><span class="si">{</span><span class="n">info</span><span class="si">}</span><span class="s2"> for </span><span class="si">{</span><span class="n">item_name</span><span class="si">}</span><span class="s2"> is </span><span class="si">{</span><span class="n">row</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="si">}</span><span class="s2">.&quot;</span>
827+
<span class="k">return</span> <span class="sa">f</span><span class="s2">&quot;The </span><span class="si">{</span><span class="n">info</span><span class="si">}</span><span class="s2"> for </span><span class="si">{</span><span class="n">item</span><span class="si">}</span><span class="s2"> is </span><span class="si">{</span><span class="n">row</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="si">}</span><span class="s2">.&quot;</span>
828828
<span class="k">except</span> <span class="ne">TypeError</span><span class="p">:</span>
829-
<span class="k">return</span> <span class="sa">f</span><span class="s2">&quot;Sorry but </span><span class="si">{</span><span class="n">item_name</span><span class="si">}</span><span class="s2"> doesn&#39;t exist in the database.&quot;</span>
829+
<span class="k">return</span> <span class="sa">f</span><span class="s2">&quot;Sorry but </span><span class="si">{</span><span class="n">item</span><span class="si">}</span><span class="s2"> doesn&#39;t exist in the database.&quot;</span>
830830

831831

832832
<span class="k">class</span> <span class="nc">GroceryItemQuery</span><span class="p">(</span><span class="n">OpenAICall</span><span class="p">):</span>
833833
<span class="n">prompt_template</span> <span class="o">=</span> <span class="s2">&quot;&quot;&quot;</span>
834834
<span class="s2"> SYSTEM:</span>
835-
<span class="s2"> Your task is properly query a database given a user&#39;s input.</span>
835+
<span class="s2"> Your task is to query a database given a user&#39;s input.</span>
836836

837837
<span class="s2"> USER:</span>
838838
<span class="s2"> </span><span class="si">{input}</span>
@@ -841,15 +841,15 @@ <h2><span class="section-number">6.16.4. </span>Maximize Accuracy and Relevance
841841
<span class="n">call_params</span> <span class="o">=</span> <span class="n">OpenAICallParams</span><span class="p">(</span><span class="n">tools</span><span class="o">=</span><span class="p">[</span><span class="n">get_item_info</span><span class="p">])</span>
842842

843843

844-
<span class="n">text</span> <span class="o">=</span> <span class="s2">&quot;What&#39;s the price for apple in the grocery_items table?&quot;</span>
844+
<span class="n">text</span> <span class="o">=</span> <span class="s2">&quot;What&#39;s the price for banana in the grocery_items table?&quot;</span>
845845
<span class="n">query_tool</span> <span class="o">=</span> <span class="n">GroceryItemQuery</span><span class="p">(</span><span class="nb">input</span><span class="o">=</span><span class="n">text</span><span class="p">)</span><span class="o">.</span><span class="n">call</span><span class="p">()</span><span class="o">.</span><span class="n">tool</span>
846846
<span class="n">result</span> <span class="o">=</span> <span class="n">query_tool</span><span class="o">.</span><span class="n">fn</span><span class="p">(</span><span class="o">**</span><span class="n">query_tool</span><span class="o">.</span><span class="n">args</span><span class="p">)</span>
847847
<span class="n">result</span>
848848
</pre></div>
849849
</div>
850850
</div>
851851
<div class="cell_output docutils container">
852-
<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>&#39;The price for apple is 0.75.&#39;
852+
<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>&#39;The price for banana is 0.5.&#39;
853853
</pre></div>
854854
</div>
855855
</div>

0 commit comments

Comments
 (0)