\n",
+ "
โ๏ธ Witness โข
Court Witness\n",
+ "
On March 15, 2023, I observed a series of messages exchanged between Doe and Wilson on social media, indicating a heated argument. The screenshots show Wilson making threatening remarks. Additionally, the therapist's report dated March 20, 2023, confirms Doe's emotional distress linked to these interactions. I can confirm the authenticity of these documents.
\n",
+ "
๐
Day 3: Final Verdict
"
+ ]
+ },
+ "metadata": {}
+ },
+ {
+ "output_type": "display_data",
+ "data": {
+ "text/plain": [
+ "\u001b[38;2;210;227;200mโญโ\u001b[0m\u001b[38;2;210;227;200m \u001b[0m\u001b[1;38;2;210;227;200mAgent Info\u001b[0m\u001b[38;2;210;227;200m \u001b[0m\u001b[38;2;210;227;200mโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ\u001b[0m\u001b[38;2;210;227;200mโโฎ\u001b[0m\n",
+ "\u001b[38;2;210;227;200mโ\u001b[0m \u001b[38;2;210;227;200mโ\u001b[0m\n",
+ "\u001b[38;2;210;227;200mโ\u001b[0m \u001b[1;38;2;255;155;155m๐ค Agent:\u001b[0m \u001b[38;2;255;229;229mJudge\u001b[0m \u001b[38;2;210;227;200mโ\u001b[0m\n",
+ "\u001b[38;2;210;227;200mโ\u001b[0m \u001b[1;38;2;180;180;179mRole:\u001b[0m \u001b[38;2;255;229;229mPreside over court proceedings\u001b[0m \u001b[38;2;210;227;200mโ\u001b[0m\n",
+ "\u001b[38;2;210;227;200mโ\u001b[0m \u001b[38;2;210;227;200mโ\u001b[0m\n",
+ "\u001b[38;2;210;227;200mโฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\u001b[0m\n"
+ ],
+ "text/html": [
+ "โญโ Agent Info โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฎ\n",
+ "โ โ\n",
+ "โ ๐ค Agent: Judge โ\n",
+ "โ Role: Preside over court proceedings โ\n",
+ "โ โ\n",
+ "โฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\n",
+ "
\n"
+ ]
+ },
+ "metadata": {}
+ },
+ {
+ "output_type": "display_data",
+ "data": {
+ "text/plain": [
+ "Output()"
+ ],
+ "application/vnd.jupyter.widget-view+json": {
+ "version_major": 2,
+ "version_minor": 0,
+ "model_id": "801192cdda5e4736bc53dba478e1dcc1"
+ }
+ },
+ "metadata": {}
+ },
+ {
+ "output_type": "display_data",
+ "data": {
+ "text/plain": [],
+ "text/html": [
+ "\n"
+ ]
+ },
+ "metadata": {}
+ },
+ {
+ "output_type": "display_data",
+ "data": {
+ "text/plain": [
+ "\n"
+ ],
+ "text/html": [
+ "\n",
+ "
\n"
+ ]
+ },
+ "metadata": {}
+ },
+ {
+ "output_type": "display_data",
+ "data": {
+ "text/plain": [
+ "\u001b[2mResponse generated in 2.3s\u001b[0m\n"
+ ],
+ "text/html": [
+ "Response generated in 2.3s\n",
+ "
\n"
+ ]
+ },
+ "metadata": {}
+ },
+ {
+ "output_type": "display_data",
+ "data": {
+ "text/plain": [
+ "\u001b[36mโญโ\u001b[0m\u001b[36mโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ\u001b[0m\u001b[36m Task \u001b[0m\u001b[36mโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ\u001b[0m\u001b[36mโโฎ\u001b[0m\n",
+ "\u001b[36mโ\u001b[0m You are Judge presiding over \"Doe vs. Wilson\". Deliver your final verdict. Consider all evidence: Screenshots \u001b[36mโ\u001b[0m\n",
+ "\u001b[36mโ\u001b[0m of messages, social media posts, therapistโs report Be fair and explain your reasoning. \u001b[36mโ\u001b[0m\n",
+ "\u001b[36mโฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\u001b[0m\n"
+ ],
+ "text/html": [
+ "โญโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ Task โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฎ\n",
+ "โ You are Judge presiding over \"Doe vs. Wilson\". Deliver your final verdict. Consider all evidence: Screenshots โ\n",
+ "โ of messages, social media posts, therapistโs report Be fair and explain your reasoning. โ\n",
+ "โฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\n",
+ "
\n"
+ ]
+ },
+ "metadata": {}
+ },
+ {
+ "output_type": "display_data",
+ "data": {
+ "text/plain": [
+ "\u001b[36mโญโ\u001b[0m\u001b[36mโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ\u001b[0m\u001b[36m Response \u001b[0m\u001b[36mโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ\u001b[0m\u001b[36mโโฎ\u001b[0m\n",
+ "\u001b[36mโ\u001b[0m After careful consideration of the evidence presented, including the screenshots of messages, social media \u001b[36mโ\u001b[0m\n",
+ "\u001b[36mโ\u001b[0m posts, and the therapist's report, I find that the plaintiff, Ms. Doe, has demonstrated a credible case of \u001b[36mโ\u001b[0m\n",
+ "\u001b[36mโ\u001b[0m emotional distress due to the defendant's actions. \u001b[36mโ\u001b[0m\n",
+ "\u001b[36mโ\u001b[0m \u001b[36mโ\u001b[0m\n",
+ "\u001b[36mโ\u001b[0m The evidence shows a pattern of harassment that has significantly impacted Ms. Doe's mental well-being. \u001b[36mโ\u001b[0m\n",
+ "\u001b[36mโ\u001b[0m Therefore, I rule in favor of the plaintiff. \u001b[36mโ\u001b[0m\n",
+ "\u001b[36mโ\u001b[0m \u001b[36mโ\u001b[0m\n",
+ "\u001b[36mโ\u001b[0m Mr. Wilson is ordered to cease all forms of contact with Ms. Doe and to pay damages for the emotional distress \u001b[36mโ\u001b[0m\n",
+ "\u001b[36mโ\u001b[0m caused. Court is adjourned. \u001b[36mโ\u001b[0m\n",
+ "\u001b[36mโฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\u001b[0m\n"
+ ],
+ "text/html": [
+ "โญโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ Response โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฎ\n",
+ "โ After careful consideration of the evidence presented, including the screenshots of messages, social media โ\n",
+ "โ posts, and the therapist's report, I find that the plaintiff, Ms. Doe, has demonstrated a credible case of โ\n",
+ "โ emotional distress due to the defendant's actions. โ\n",
+ "โ โ\n",
+ "โ The evidence shows a pattern of harassment that has significantly impacted Ms. Doe's mental well-being. โ\n",
+ "โ Therefore, I rule in favor of the plaintiff. โ\n",
+ "โ โ\n",
+ "โ Mr. Wilson is ordered to cease all forms of contact with Ms. Doe and to pay damages for the emotional distress โ\n",
+ "โ caused. Court is adjourned. โ\n",
+ "โฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\n",
+ "
\n"
+ ]
+ },
+ "metadata": {}
+ },
+ {
+ "output_type": "display_data",
+ "data": {
+ "text/plain": [
+ ""
+ ],
+ "text/html": [
+ "\n",
+ " \n",
+ "
โ๏ธ Judge โข
Final Verdict\n",
+ "
After careful consideration of the evidence presented, including the screenshots of messages, social media posts, and the therapist's report, I find that the plaintiff, Ms. Doe, has demonstrated a credible case of emotional distress due to the defendant's actions.\n",
+ "\n",
+ "The evidence shows a pattern of harassment that has significantly impacted Ms. Doe's mental well-being. Therefore, I rule in favor of the plaintiff. \n",
+ "\n",
+ "Mr. Wilson is ordered to cease all forms of contact with Ms. Doe and to pay damages for the emotional distress caused. Court is adjourned.
\n",
+ "
๐
Day 1: Opening Statements
\"))\n",
+ "\n",
+ "# Judge opens court\n",
+ "judge_prompt = f\"\"\"\n",
+ "You are Judge presiding over \"{case_title}\".\n",
+ "Open the court proceedings professionally.\n",
+ "Case: {case_description}\n",
+ "Keep it brief and formal.\n",
+ "\"\"\"\n",
+ "judge_response = run_agent(judge, judge_prompt, \"Judge\", \"Presiding Judge\", \"#8B4513\")\n",
+ "\n",
+ "# Prosecutor opening\n",
+ "prosecutor_prompt = f\"\"\"\n",
+ "You are the Prosecutor for \"{case_title}\".\n",
+ "Give your opening statement.\n",
+ "Evidence: {evidence}\n",
+ "Be confident and factual.\n",
+ "\"\"\"\n",
+ "prosecutor_response = run_agent(prosecutor, prosecutor_prompt, \"Prosecutor\", \"State Attorney\", \"#d32f2f\")\n",
+ "\n",
+ "# Defense opening\n",
+ "defense_prompt = f\"\"\"\n",
+ "You are the Defense Attorney for \"{case_title}\".\n",
+ "Give your opening statement.\n",
+ "Challenge the prosecution's case.\n",
+ "Emphasize presumption of innocence.\n",
+ "\"\"\"\n",
+ "defense_response = run_agent(defense, defense_prompt, \"Defense\", \"Defense Attorney\", \"#1976d2\")\n",
+ "\n",
+ "time.sleep(1)\n",
+ "\n",
+ "# Day 2: Testimony\n",
+ "display(HTML(f\"๐
Day 2: Witness Testimony
\"))\n",
+ "\n",
+ "# Witness testimony\n",
+ "witness_prompt = f\"\"\"\n",
+ "You are a witness in \"{case_title}\".\n",
+ "Provide your testimony about what you saw.\n",
+ "Evidence: {evidence}\n",
+ "Be factual and clear.\n",
+ "\"\"\"\n",
+ "witness_response = run_agent(witness, witness_prompt, \"Witness\", \"Court Witness\", \"#ff9800\")\n",
+ "\n",
+ "time.sleep(1)\n",
+ "\n",
+ "# Day 3: Verdict\n",
+ "display(HTML(f\"๐
Day 3: Final Verdict
\"))\n",
+ "\n",
+ "# Judge's verdict\n",
+ "verdict_prompt = f\"\"\"\n",
+ "You are Judge presiding over \"{case_title}\".\n",
+ "Deliver your final verdict.\n",
+ "Consider all evidence: {evidence}\n",
+ "Be fair and explain your reasoning.\n",
+ "\"\"\"\n",
+ "verdict_response = run_agent(judge, verdict_prompt, \"Judge\", \"Final Verdict\", \"#8B4513\")\n",
+ "\n",
+ "print(\"\\nโ
Mini court simulation completed!\")\n"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "source": [
+ "# Summary"
+ ],
+ "metadata": {
+ "id": "0hP6lxvCRI1m"
+ },
+ "id": "0hP6lxvCRI1m"
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 18,
+ "id": "3cc7570e",
+ "metadata": {
+ "colab": {
+ "base_uri": "https://localhost:8080/"
+ },
+ "id": "3cc7570e",
+ "outputId": "2960f533-2a08-45c9-8653-95817f61f6d8"
+ },
+ "outputs": [
+ {
+ "output_type": "stream",
+ "name": "stdout",
+ "text": [
+ "๐ฏ Mini Legalia AI Complete!\n",
+ "\n",
+ "This simplified version demonstrates:\n",
+ "- 4 Essential Agents: Judge, Prosecutor, Defense, Witness\n",
+ "- 3-Day Trial: Opening, Testimony, Verdict\n",
+ "- Real-time Interaction: Each agent responds based on case context\n",
+ "- Easy Setup: Minimal dependencies and configuration\n",
+ "\n",
+ "Perfect for quick demonstrations and learning! โ๏ธ\n"
+ ]
+ }
+ ],
+ "source": [
+ "print(\"๐ฏ Mini Legalia AI Complete!\")\n",
+ "print(\"\\nThis simplified version demonstrates:\")\n",
+ "print(\"- 4 Essential Agents: Judge, Prosecutor, Defense, Witness\")\n",
+ "print(\"- 3-Day Trial: Opening, Testimony, Verdict\")\n",
+ "print(\"- Real-time Interaction: Each agent responds based on case context\")\n",
+ "print(\"- Easy Setup: Minimal dependencies and configuration\")\n",
+ "print(\"\\nPerfect for quick demonstrations and learning! โ๏ธ\")"
+ ]
+ }
+ ],
+ "metadata": {
+ "kernelspec": {
+ "display_name": "Python 3",
+ "language": "python",
+ "name": "python3"
+ },
+ "language_info": {
+ "name": "python",
+ "version": "3.10"
+ },
+ "colab": {
+ "provenance": []
+ },
+ "widgets": {
+ "application/vnd.jupyter.widget-state+json": {
+ "74964427de63470da100903aac0b8c78": {
+ "model_module": "@jupyter-widgets/output",
+ "model_name": "OutputModel",
+ "model_module_version": "1.0.0",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/output",
+ "_model_module_version": "1.0.0",
+ "_model_name": "OutputModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/output",
+ "_view_module_version": "1.0.0",
+ "_view_name": "OutputView",
+ "layout": "IPY_MODEL_45ff535d008a4b30b53a0e72edee016b",
+ "msg_id": "",
+ "outputs": [
+ {
+ "output_type": "display_data",
+ "data": {
+ "text/plain": "\u001b[32mโญโ\u001b[0m\u001b[32mโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ\u001b[0m\u001b[32m Generating... 3.8s \u001b[0m\u001b[32mโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ\u001b[0m\u001b[32mโโฎ\u001b[0m\n\u001b[32mโ\u001b[0m Court is now in session. \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m Case number 2023-001: Doe vs. Wilson. The plaintiff, Ms. Doe, alleges emotional distress resulting from \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m repeated online harassment by the defendant, Mr. Wilson. \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m Both parties will present their evidence and arguments. I remind all present to maintain decorum. \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m Let us proceed with the plaintiff's \u001b[32mโ\u001b[0m\n\u001b[32mโฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\u001b[0m\n",
+ "text/html": "โญโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ Generating... 3.8s โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฎ\nโ Court is now in session. โ\nโ โ\nโ Case number 2023-001: Doe vs. Wilson. The plaintiff, Ms. Doe, alleges emotional distress resulting from โ\nโ repeated online harassment by the defendant, Mr. Wilson. โ\nโ โ\nโ Both parties will present their evidence and arguments. I remind all present to maintain decorum. โ\nโ โ\nโ Let us proceed with the plaintiff's โ\nโฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\n
\n"
+ },
+ "metadata": {}
+ }
+ ]
+ }
+ },
+ "45ff535d008a4b30b53a0e72edee016b": {
+ "model_module": "@jupyter-widgets/base",
+ "model_name": "LayoutModel",
+ "model_module_version": "1.2.0",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "9e740fc3b45c454e96dfe29b4c0edb53": {
+ "model_module": "@jupyter-widgets/output",
+ "model_name": "OutputModel",
+ "model_module_version": "1.0.0",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/output",
+ "_model_module_version": "1.0.0",
+ "_model_name": "OutputModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/output",
+ "_view_module_version": "1.0.0",
+ "_view_name": "OutputView",
+ "layout": "IPY_MODEL_c04a2ce5fcc544d8ac72aa67776c0fd0",
+ "msg_id": "",
+ "outputs": [
+ {
+ "output_type": "display_data",
+ "data": {
+ "text/plain": "\u001b[32mโญโ\u001b[0m\u001b[32mโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ\u001b[0m\u001b[32m Generating... 2.9s \u001b[0m\u001b[32mโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ\u001b[0m\u001b[32mโโฎ\u001b[0m\n\u001b[32mโ\u001b[0m Ladies and gentlemen of the jury, today we present clear evidence against the defendant, Wilson. We will show \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m you screenshots of messages that reveal intent and motive, alongside social media posts that further \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m corroborate this behavior. Additionally, a therapistโs report will demonstrate the psychological impact on the \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m victim, Doe. This evidence will establish a pattern of harassment and intimidation. We ask you to consider the \u001b[32mโ\u001b[0m\n\u001b[32mโฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\u001b[0m\n",
+ "text/html": "โญโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ Generating... 2.9s โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฎ\nโ Ladies and gentlemen of the jury, today we present clear evidence against the defendant, Wilson. We will show โ\nโ you screenshots of messages that reveal intent and motive, alongside social media posts that further โ\nโ corroborate this behavior. Additionally, a therapistโs report will demonstrate the psychological impact on the โ\nโ victim, Doe. This evidence will establish a pattern of harassment and intimidation. We ask you to consider the โ\nโฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\n
\n"
+ },
+ "metadata": {}
+ }
+ ]
+ }
+ },
+ "c04a2ce5fcc544d8ac72aa67776c0fd0": {
+ "model_module": "@jupyter-widgets/base",
+ "model_name": "LayoutModel",
+ "model_module_version": "1.2.0",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "c0c194ce705f4418961c84f0bacb1acf": {
+ "model_module": "@jupyter-widgets/output",
+ "model_name": "OutputModel",
+ "model_module_version": "1.0.0",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/output",
+ "_model_module_version": "1.0.0",
+ "_model_name": "OutputModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/output",
+ "_view_module_version": "1.0.0",
+ "_view_name": "OutputView",
+ "layout": "IPY_MODEL_4ef74a10d03747608c9011027bfa877c",
+ "msg_id": "",
+ "outputs": [
+ {
+ "output_type": "display_data",
+ "data": {
+ "text/plain": "\u001b[32mโญโ\u001b[0m\u001b[32mโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ\u001b[0m\u001b[32m Generating... 3.1s \u001b[0m\u001b[32mโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ\u001b[0m\u001b[32mโโฎ\u001b[0m\n\u001b[32mโ\u001b[0m Ladies and gentlemen of the jury, today you will hear the prosecution's case against my client, Mr. Wilson. \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m However, I urge you to remember that the burden of proof lies with them. They must prove guilt beyond a \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m reasonable doubt. We will show that the evidence is circumstantial, unreliable, and lacks credibility. Mr. \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m Wilson is presumed innocent until proven guilty, and we \u001b[32mโ\u001b[0m\n\u001b[32mโฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\u001b[0m\n",
+ "text/html": "โญโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ Generating... 3.1s โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฎ\nโ Ladies and gentlemen of the jury, today you will hear the prosecution's case against my client, Mr. Wilson. โ\nโ However, I urge you to remember that the burden of proof lies with them. They must prove guilt beyond a โ\nโ reasonable doubt. We will show that the evidence is circumstantial, unreliable, and lacks credibility. Mr. โ\nโ Wilson is presumed innocent until proven guilty, and we โ\nโฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\n
\n"
+ },
+ "metadata": {}
+ }
+ ]
+ }
+ },
+ "4ef74a10d03747608c9011027bfa877c": {
+ "model_module": "@jupyter-widgets/base",
+ "model_name": "LayoutModel",
+ "model_module_version": "1.2.0",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "f4d166f4a67e41679066f05d4f224c31": {
+ "model_module": "@jupyter-widgets/output",
+ "model_name": "OutputModel",
+ "model_module_version": "1.0.0",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/output",
+ "_model_module_version": "1.0.0",
+ "_model_name": "OutputModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/output",
+ "_view_module_version": "1.0.0",
+ "_view_name": "OutputView",
+ "layout": "IPY_MODEL_f47eec09409e440091431311f3048518",
+ "msg_id": "",
+ "outputs": [
+ {
+ "output_type": "display_data",
+ "data": {
+ "text/plain": "\u001b[32mโญโ\u001b[0m\u001b[32mโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ\u001b[0m\u001b[32m Generating... 2.1s \u001b[0m\u001b[32mโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ\u001b[0m\u001b[32mโโฎ\u001b[0m\n\u001b[32mโ\u001b[0m On March 15, 2023, I observed a series of messages exchanged between Doe and Wilson on social media, indicating \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m a heated argument. The screenshots show Wilson making threatening remarks. Additionally, the therapist's report \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m dated March 20, 2023, confirms Doe's emotional distress linked to these \u001b[32mโ\u001b[0m\n\u001b[32mโฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\u001b[0m\n",
+ "text/html": "โญโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ Generating... 2.1s โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฎ\nโ On March 15, 2023, I observed a series of messages exchanged between Doe and Wilson on social media, indicating โ\nโ a heated argument. The screenshots show Wilson making threatening remarks. Additionally, the therapist's report โ\nโ dated March 20, 2023, confirms Doe's emotional distress linked to these โ\nโฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\n
\n"
+ },
+ "metadata": {}
+ }
+ ]
+ }
+ },
+ "f47eec09409e440091431311f3048518": {
+ "model_module": "@jupyter-widgets/base",
+ "model_name": "LayoutModel",
+ "model_module_version": "1.2.0",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "801192cdda5e4736bc53dba478e1dcc1": {
+ "model_module": "@jupyter-widgets/output",
+ "model_name": "OutputModel",
+ "model_module_version": "1.0.0",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/output",
+ "_model_module_version": "1.0.0",
+ "_model_name": "OutputModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/output",
+ "_view_module_version": "1.0.0",
+ "_view_name": "OutputView",
+ "layout": "IPY_MODEL_1928c67194d4409a9528a796bbe1fce1",
+ "msg_id": "",
+ "outputs": [
+ {
+ "output_type": "display_data",
+ "data": {
+ "text/plain": "\u001b[32mโญโ\u001b[0m\u001b[32mโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ\u001b[0m\u001b[32m Generating... 2.1s \u001b[0m\u001b[32mโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ\u001b[0m\u001b[32mโโฎ\u001b[0m\n\u001b[32mโ\u001b[0m After careful consideration of the evidence presented, including the screenshots of messages, social media \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m posts, and the therapist's report, I find that the plaintiff, Ms. Doe, has demonstrated a credible case of \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m emotional distress due to the defendant's actions. \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m The evidence shows a pattern of harassment that has significantly impacted Ms. Doe's mental well-being. \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m Therefore, I rule in favor of the plaintiff. \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m \u001b[32mโ\u001b[0m\n\u001b[32mโ\u001b[0m Mr. Wilson is ordered to cease all forms of contact with Ms. Doe and to pay damages \u001b[32mโ\u001b[0m\n\u001b[32mโฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\u001b[0m\n",
+ "text/html": "โญโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ Generating... 2.1s โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฎ\nโ After careful consideration of the evidence presented, including the screenshots of messages, social media โ\nโ posts, and the therapist's report, I find that the plaintiff, Ms. Doe, has demonstrated a credible case of โ\nโ emotional distress due to the defendant's actions. โ\nโ โ\nโ The evidence shows a pattern of harassment that has significantly impacted Ms. Doe's mental well-being. โ\nโ Therefore, I rule in favor of the plaintiff. โ\nโ โ\nโ Mr. Wilson is ordered to cease all forms of contact with Ms. Doe and to pay damages โ\nโฐโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโฏ\n
\n"
+ },
+ "metadata": {}
+ }
+ ]
+ }
+ },
+ "1928c67194d4409a9528a796bbe1fce1": {
+ "model_module": "@jupyter-widgets/base",
+ "model_name": "LayoutModel",
+ "model_module_version": "1.2.0",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ }
+ }
+ }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}
diff --git a/examples/cookbooks/Cybersecurity-PoC Agent.ipynb b/examples/cookbooks/Cybersecurity-PoC Agent.ipynb
new file mode 100644
index 00000000..81f4e375
--- /dev/null
+++ b/examples/cookbooks/Cybersecurity-PoC Agent.ipynb
@@ -0,0 +1,233 @@
+{
+ "nbformat": 4,
+ "nbformat_minor": 0,
+ "metadata": {
+ "colab": {
+ "provenance": []
+ },
+ "kernelspec": {
+ "name": "python3",
+ "display_name": "Python 3"
+ },
+ "language_info": {
+ "name": "python"
+ }
+ },
+ "cells": [
+ {
+ "cell_type": "markdown",
+ "source": [
+ "# Pocky Query Tool: Automated CVE PoC Search & Validation\n",
+ "\n",
+ "A lightweight, web-scale agent that helps you find, filter, and fetch real-world PoC exploits โ so you don't have to.\n",
+ "\n",
+ "**Features:**\n",
+ "- Automatically searches multiple security-related websites\n",
+ "- Intelligently analyzes and extracts PoC code\n",
+ "- Automatically selects the most reliable PoC samples\n",
+ "- Supports collection of PoCs from multiple sources"
+ ],
+ "metadata": {
+ "id": "BdX56iM1r5aB"
+ }
+ },
+ {
+ "cell_type": "markdown",
+ "source": [
+ "[](https://colab.research.google.com/github/DhivyaBharathy-web/PraisonAI/blob/main/examples/cookbooks/Pocky_Cybersecurity_PoC_Agent.ipynb)\n"
+ ],
+ "metadata": {
+ "id": "VW766102tUUY"
+ }
+ },
+ {
+ "cell_type": "markdown",
+ "source": [
+ "# Install Dependencies"
+ ],
+ "metadata": {
+ "id": "cBu2iXmJsVqE"
+ }
+ },
+ {
+ "cell_type": "code",
+ "source": [
+ "!pip install praisonaiagents exa-py python-dotenv requests beautifulsoup4"
+ ],
+ "metadata": {
+ "id": "VvbA3A7XsTFm"
+ },
+ "execution_count": null,
+ "outputs": []
+ },
+ {
+ "cell_type": "markdown",
+ "source": [
+ "# Set API Keys"
+ ],
+ "metadata": {
+ "id": "Q6B2VEkFsB32"
+ }
+ },
+ {
+ "cell_type": "code",
+ "source": [
+ "import os\n",
+ "\n",
+ "# Set your API keys here (replace with your actual keys)\n",
+ "os.environ[\"EXA_API_KEY\"] = \"your api key\"\n",
+ "os.environ[\"OPENAI_API_KEY\"] = \"your api key\"\n",
+ "os.environ[\"OPENAI_BASE_URL\"] = \"https://api.openai.com/v1\" # Optional, for custom OpenAI endpoints"
+ ],
+ "metadata": {
+ "id": "OlOI3yc_sAkN"
+ },
+ "execution_count": 1,
+ "outputs": []
+ },
+ {
+ "cell_type": "markdown",
+ "source": [
+ "# Tools (Core Functions)"
+ ],
+ "metadata": {
+ "id": "1BvjjjJdsanO"
+ }
+ },
+ {
+ "cell_type": "code",
+ "source": [
+ "import json\n",
+ "from openai import OpenAI\n",
+ "from exa_py import Exa\n",
+ "\n",
+ "# Dummy/Minimal agent classes for notebook demo\n",
+ "class ValidationAgent:\n",
+ " def __init__(self, input_json):\n",
+ " self.input_json = input_json\n",
+ " def run(self):\n",
+ " # Dummy validation logic for notebook demo\n",
+ " data = json.loads(self.input_json)\n",
+ " return True if \"attack_intent\" in data and \"poc_sample\" in data else False\n",
+ "\n",
+ "class AttackIntentAgent:\n",
+ " def __init__(self, description):\n",
+ " self.description = description\n",
+ " def run(self):\n",
+ " # Dummy intent extraction for notebook demo\n",
+ " return f\"Intent for: {self.description[:50]}...\""
+ ],
+ "metadata": {
+ "id": "GYfAJLXOsbga"
+ },
+ "execution_count": 3,
+ "outputs": []
+ },
+ {
+ "cell_type": "markdown",
+ "source": [
+ "## YAML Prompt (Validation Example)\n",
+ "This is the prompt used for PoC validation."
+ ],
+ "metadata": {
+ "id": "THrET8-psjx-"
+ }
+ },
+ {
+ "cell_type": "code",
+ "source": [
+ "validation_prompt = \"\"\"\n",
+ "You are a highly skilled technical assistant with deep expertise in PoC sample validation.\n",
+ "\n",
+ "Given the attack intent of a CVE vulnerability and a PoC sample gathered from public sources, your task is to analyze whether the PoC correctly implements the intended attack behavior.\n",
+ "\n",
+ "Specifically:\n",
+ "- Understand the CVE's attack intent, including the attack goal and the underlying exploitation mechanism.\n",
+ "- Analyze the PoC to determine whether it is designed to achieve this intent.\n",
+ "- Check whether the payloads, request structures, and overall logic of the PoC align with the described attack intent.\n",
+ "- You do not need to execute the PoC. Focus on static validation through reasoning and consistency.\n",
+ "\n",
+ "Your output must be a JSON object with two fields:\n",
+ "- \"valid\": a boolean indicating whether the PoC correctly reflects the attack intent.\n",
+ "- \"reasoning\": a brief explanation of your judgment. If \"valid\" is false, the reasoning must clearly explain what is incorrect or inconsistent in the PoC compared to the attack intent, so that the PoC can be revised accordingly.\n",
+ "\"\"\"\n",
+ "print(validation_prompt)"
+ ],
+ "metadata": {
+ "id": "9q3aKl1xshrb"
+ },
+ "execution_count": null,
+ "outputs": []
+ },
+ {
+ "cell_type": "markdown",
+ "source": [
+ "# Main (Query and Validate a CVE PoC)"
+ ],
+ "metadata": {
+ "id": "531EZgcLsqP6"
+ }
+ },
+ {
+ "cell_type": "code",
+ "source": [
+ "def run_pocky_for_cve(cve_id):\n",
+ " # Example: Simulate fetching a description and PoC (replace with real logic)\n",
+ " description = f\"Description for {cve_id} (replace with real Exa/OpenAI search)\"\n",
+ " poc_sample = f\"PoC code for {cve_id} (replace with real PoC search)\"\n",
+ "\n",
+ " # Stage 2: Attack Intent\n",
+ " intent = AttackIntentAgent(description).run()\n",
+ " print(f\"Attack Intent: {intent}\")\n",
+ "\n",
+ " # Stage 3: Validation\n",
+ " validation_input = json.dumps({\"attack_intent\": intent, \"poc_sample\": poc_sample}, indent=2)\n",
+ " valid = ValidationAgent(validation_input).run()\n",
+ " print(f\"Validation Result: {valid}\")\n",
+ " if valid:\n",
+ " print(f\"PoC for {cve_id} is valid and ready to use.\")\n",
+ " else:\n",
+ " print(f\"PoC for {cve_id} failed validation.\")"
+ ],
+ "metadata": {
+ "id": "PQvtF-RqsrP6"
+ },
+ "execution_count": 5,
+ "outputs": []
+ },
+ {
+ "cell_type": "markdown",
+ "source": [
+ "# Example Usage"
+ ],
+ "metadata": {
+ "id": "XiQOiSz8su3u"
+ }
+ },
+ {
+ "cell_type": "code",
+ "source": [
+ "run_pocky_for_cve(\"CVE-2023-4450\")"
+ ],
+ "metadata": {
+ "colab": {
+ "base_uri": "https://localhost:8080/"
+ },
+ "id": "39w-qpecswjw",
+ "outputId": "cdcb3b29-7338-4e3e-b160-5f9568c194ca"
+ },
+ "execution_count": 6,
+ "outputs": [
+ {
+ "output_type": "stream",
+ "name": "stdout",
+ "text": [
+ "Attack Intent: Intent for: Description for CVE-2023-4450 (replace with real E...\n",
+ "Validation Result: True\n",
+ "PoC for CVE-2023-4450 is valid and ready to use.\n"
+ ]
+ }
+ ]
+ }
+ ]
+}
diff --git a/examples/cookbooks/Government-Services-Assistant.ipynb b/examples/cookbooks/Government-Services-Assistant.ipynb
new file mode 100644
index 00000000..b410962c
--- /dev/null
+++ b/examples/cookbooks/Government-Services-Assistant.ipynb
@@ -0,0 +1,300 @@
+{
+ "cells": [
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "ZH_nR-SvvkDG"
+ },
+ "source": [
+ "# Government Services Assistant - AI Chatbot"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "w8B741JgvpFj"
+ },
+ "source": [
+ "This notebook demonstrates how to use an AI-powered assistant to answer questions about government services and procedures, using the Firecrawl API and a friendly, step-by-step conversational approach."
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "y8jiJYf4FA0m"
+ },
+ "source": [
+ "[](https://colab.research.google.com/github/DhivyaBharathy-web/PraisonAI/blob/main/examples/cookbooks/Chile_Government_Services_Assistant.ipynb)\n"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "RRw8sPG89KNb"
+ },
+ "source": [
+ "# Install dependencies"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "rW8ltqCICV8o"
+ },
+ "outputs": [],
+ "source": [
+ "!pip install flask firecrawl praisonaiagents google-genai python-dotenv deep-translator"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "XGjyt-B_EfbM"
+ },
+ "source": [
+ "# Set API Keys"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "qf8B_YltDiIe"
+ },
+ "outputs": [],
+ "source": [
+ "import os\n",
+ "\n",
+ "os.environ['FIRECRAWL_API_KEY'] = \"your api key here\"\n",
+ "os.environ['OPENAI_API_KEY'] = \"your api key here\""
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "ccO0vwvCEqUJ"
+ },
+ "source": [
+ "# Import Libraries & Translator"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "0prDQ5TpDnFu"
+ },
+ "outputs": [],
+ "source": [
+ "from firecrawl import FirecrawlApp, ScrapeOptions\n",
+ "from deep_translator import GoogleTranslator\n",
+ "import re\n",
+ "\n",
+ "def translate_to_spanish(text):\n",
+ " try:\n",
+ " return GoogleTranslator(source='auto', target='es').translate(text)\n",
+ " except Exception as e:\n",
+ " print(\"Translation to Spanish failed:\", e)\n",
+ " return text\n",
+ "\n",
+ "def translate_to_english(text):\n",
+ " try:\n",
+ " # Remove Markdown images and None values before translation\n",
+ " text = str(text).replace(\"None\", \"\")\n",
+ " text = re.sub(r'!\\[.*?\\]\\(.*?\\)', '', text)\n",
+ " return GoogleTranslator(source='auto', target='en').translate(text)\n",
+ " except Exception as e:\n",
+ " print(\"Translation to English failed:\", e)\n",
+ " return text"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "WxOlCHMmEuK2"
+ },
+ "source": [
+ "# Firecrawl Tool Class"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "G4RyzJ5mDp0t"
+ },
+ "outputs": [],
+ "source": [
+ "class FirecrawlTool:\n",
+ " def __init__(self, api_key, instruction: str, template: str):\n",
+ " if not api_key:\n",
+ " raise ValueError(\"Firecrawl API key not provided.\")\n",
+ " self.app = FirecrawlApp(api_key=api_key)\n",
+ " self.instruction = instruction\n",
+ " self.template = template\n",
+ "\n",
+ " def search(self, search: str) -> str:\n",
+ " if not search or len(search) < 5:\n",
+ " return \"Error: Please provide a valid search query (at least 5 characters).\"\n",
+ " response_md = \"\"\n",
+ " try:\n",
+ " search_result = self.app.search(\n",
+ " query=self.instruction + search,\n",
+ " limit=2,\n",
+ " country=\"cl\",\n",
+ " lang=\"es\", # Always search in Spanish for best results\n",
+ " scrape_options=ScrapeOptions(formats=[\"markdown\", \"links\"])\n",
+ " )\n",
+ " if search_result and hasattr(search_result, 'data') and search_result.data:\n",
+ " filtered_results = [\n",
+ " result for result in search_result.data\n",
+ " if str(result.get(\"url\", \"\")).startswith(\"https://www.chileatiende.gob.cl/fichas\") and not str(result.get(\"url\", \"\")).endswith(\"pdf\")\n",
+ " ]\n",
+ " if filtered_results:\n",
+ " for num, result in enumerate(filtered_results, start=1):\n",
+ " response_md += self.template.format(\n",
+ " result_number=num,\n",
+ " page_title=str(result.get(\"title\", \"\")),\n",
+ " page_url=str(result.get(\"url\", \"\")),\n",
+ " page_content=str(result.get(\"markdown\", \"\"))\n",
+ " )\n",
+ " return response_md\n",
+ " else:\n",
+ " return None\n",
+ " else:\n",
+ " return None\n",
+ " except Exception as e:\n",
+ " return f\"Error during search: {e}\""
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "MjkjTWn_ExS0"
+ },
+ "source": [
+ "# Firecrawl Prompt Template"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "AfivymU8Dufz"
+ },
+ "outputs": [],
+ "source": [
+ "FIRECRAWL_INSTRUCTION = \"ChileAtiende: \"\n",
+ "FIRECRAWL_TEMPLATE = \"\"\"\n",
+ "# Result {result_number}\n",
+ "\n",
+ "## Page Name:\n",
+ "\"{page_title}\"\n",
+ "\n",
+ "## URL:\n",
+ "{page_url}\n",
+ "\n",
+ "## Content:\n",
+ "{page_content}\n",
+ "\n",
+ "\"\"\""
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "zK8AA_DlEz9K"
+ },
+ "source": [
+ "# Initialize Firecrawl Tool"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "c3NKK0ZjDwKT"
+ },
+ "outputs": [],
+ "source": [
+ "firecrawl_tool = FirecrawlTool(\n",
+ " api_key=os.environ['FIRECRAWL_API_KEY'],\n",
+ " instruction=FIRECRAWL_INSTRUCTION,\n",
+ " template=FIRECRAWL_TEMPLATE\n",
+ ")"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "uzXYIF_gE3XV"
+ },
+ "source": [
+ "# Main Chat Loop"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "colab": {
+ "base_uri": "https://localhost:8080/"
+ },
+ "id": "TXMgZQNkDx7n",
+ "outputId": "76303cd1-a576-483f-a22d-9857e5e6d797"
+ },
+ "outputs": [
+ {
+ "name": "stdout",
+ "output_type": "stream",
+ "text": [
+ "Hello! I am your ChileAtiende assistant, Tomรกs. How can I help you today?\n",
+ "You can ask me, for example: How to renew your ID card, How to apply for the Winter Bonus, etc.\n",
+ "\n",
+ "You: exit\n",
+ "Tomรกs: It was a pleasure to help you. Goodbye!\n"
+ ]
+ }
+ ],
+ "source": [
+ "print(\"Hello! I am your ChileAtiende assistant, Tomรกs. How can I help you today?\")\n",
+ "print(\"You can ask me, for example: How to renew your ID card, How to apply for the Winter Bonus, etc.\")\n",
+ "\n",
+ "while True:\n",
+ " user_input = input(\"\\nYou: \")\n",
+ " if user_input.lower() in [\"exit\", \"quit\"]:\n",
+ " print(\"Tomรกs: It was a pleasure to help you. Goodbye!\")\n",
+ " break\n",
+ "\n",
+ " # Translate English input to Spanish for Firecrawl\n",
+ " spanish_query = translate_to_spanish(user_input)\n",
+ " spanish_answer = firecrawl_tool.search(spanish_query)\n",
+ "\n",
+ " # Only translate if we got a real answer\n",
+ " if spanish_answer and isinstance(spanish_answer, str) and spanish_answer.strip() and \"Error\" not in spanish_answer:\n",
+ " try:\n",
+ " english_answer = translate_to_english(spanish_answer)\n",
+ " print(\"\\nTomรกs (in English):\\n\", english_answer)\n",
+ " except Exception as e:\n",
+ " print(f\"\\nTomรกs: I found information, but couldn't translate it. Here it is in Spanish:\\n{spanish_answer}\\n(Translation error: {e})\")\n",
+ " else:\n",
+ " print(\"\\nTomรกs: Sorry, I couldn't find relevant information. Try rephrasing your question or ask about another service.\")"
+ ]
+ }
+ ],
+ "metadata": {
+ "colab": {
+ "provenance": []
+ },
+ "kernelspec": {
+ "display_name": "Python 3",
+ "name": "python3"
+ },
+ "language_info": {
+ "name": "python"
+ }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 0
+}