refactor and setup langgraph flow

src/chatbot/form_management/acceptance_criteria.json

@@ -0,0 +1,32 @@
+{
+    "fields": [
+      {
+        "fieldName": "Project Name",
+        "priority": 1,
+        "description": "The name of the project or purpose for the payment.",
+        "validationRules": [
+          "Must be associated with environmental or sustainability projects."
+        ],
+        "exampleInput": "โครงการ \"การจัดการคาร์บอนเครดิตในป่าเพื่อการพัฒนาที่ยั่งยืน\""
+      },
+      {
+        "fieldName": "Date",
+        "priority": 2,
+        "description": "The date of the transaction.",
+        "validationRules": [
+          "Must be a date that is in the past or present."
+        ],
+        "exampleInput": "3 เมษายน 2567"
+      },
+      {
+        "fieldName": "Total Payment Amount",
+        "priority": 3,
+        "description": "The total amount paid.",
+        "validationRules": [
+          "Must be a positive amount."
+        ],
+        "exampleInput": "3 เมษายน 2567"
+      }
+    ]
+  }
+  

src/chatbot/form_management/form_management.py

@@ -0,0 +1,89 @@
+class Form:
+    class PaymentDetailsTable:
+        def __init__(self, column_headers = None, row_entries = None):
+            self.column_headers = column_headers
+            self.row_entries = row_entries
+
+    def __init__(self, receipt_title = None, project_name = None, date = None, receipt_number = None, payer_name = None, column_headers = None, row_entries = None, total_payment_amount = None, incompleteness_description = None):
+        self.receipt_title = receipt_title
+        self.project_name = project_name
+        self.date = date
+        self.receipt_number = receipt_number
+        self.payer_name = payer_name
+        self.payment_details_table = self.PaymentDetailsTable(column_headers, row_entries)
+        self.total_payment_amount = total_payment_amount
+        self.incompleteness_description = incompleteness_description
+
+    def get_info_as_dict(self):
+        return {
+            "receipt_title": self.receipt_title,
+            "project_name": self.project_name,
+            "date": self.date,
+            "receipt_number": self.receipt_number,
+            "payer_name": self.payer_name,
+            "payment_details_table": {
+                "column_headers": self.payment_details_table.column_headers,
+                "row_entries": self.payment_details_table.row_entries
+            },
+            "total_payment_amount": self.total_payment_amount,
+            "incompleteness_description": self.incompleteness_description
+        }
+
+    # Getter and Setter for receipt_title
+    def get_receipt_title(self):
+        return self.receipt_title
+
+    def set_receipt_title(self, receipt_title):
+        self.receipt_title = receipt_title
+
+    # Getter and Setter for project_name
+    def get_project_name(self):
+        return self.project_name
+
+    def set_project_name(self, project_name):
+        self.project_name = project_name
+
+    # Getter and Setter for date
+    def get_date(self):
+        return self.date
+
+    def set_date(self, date):
+        self.date = date
+
+    # Getter and Setter for receipt_number
+    def get_receipt_number(self):
+        return self.receipt_number
+
+    def set_receipt_number(self, receipt_number):
+        self.receipt_number = receipt_number
+
+    # Getter and Setter for payer_name
+    def get_payer_name(self):
+        return self.payer_name
+
+    def set_payer_name(self, payer_name):
+        self.payer_name = payer_name
+
+    # Getter and Setter for payment_details_table
+    def get_payment_details_table(self):
+        return self.payment_details_table
+
+    def set_payment_details_table(self, payment_details_table):
+        self.payment_details_table = self.PaymentDetailsTable(**payment_details_table)
+
+    # Getter and Setter for total_payment_amount
+    def get_total_payment_amount(self):
+        return self.total_payment_amount
+
+    def set_total_payment_amount(self, total_payment_amount):
+        self.total_payment_amount = total_payment_amount
+
+    # Getter and Setter for incompleteness_description
+    def get_incompleteness_description(self):
+        return self.incompleteness_description
+
+    def set_incompleteness_description(self, incompleteness_description):
+        self.incompleteness_description = incompleteness_description
+
+# # Example Usage
+# form = Form()

src/chatbot/graph_manager.py

@@ -5,10 +5,13 @@ from langgraph.graph import StateGraph, START, END
 from langgraph.checkpoint.memory import MemorySaver
 from nodes import (
     State,
-    speaker_chatbot
+    speaker_chatbot_node,
+    tool_node
 )
+from langgraph.prebuilt import tools_condition
 from form_management.form_management import Form
 import sys
+from IPython.display import Image, display
 
 class GraphManager:
     def __init__(self):
@@ -22,13 +25,29 @@ class GraphManager:
         graph_memory = MemorySaver()
         graph_builder = StateGraph(State)
 
-        graph_builder.add_node("speaker_chatbot", speaker_chatbot)
+        graph_builder.add_node("speaker_chatbot", speaker_chatbot_node)
+        graph_builder.add_node("tools", tool_node)
+
+        graph_builder.add_conditional_edges(
+            "speaker_chatbot",
+            tools_condition,
+        )
+        # Any time a tool is called, we return to the chatbot to decide the next step
+        graph_builder.add_edge("tools", "speaker_chatbot")
 
         graph_builder.add_edge(START, "speaker_chatbot")
         graph_builder.add_edge("speaker_chatbot", END)
 
         self.graph = graph_builder.compile(checkpointer=graph_memory)
 
+    def save_graph_image(self):
+        """
+        Save the graph as a PNG image.
+        """
+        image = Image(self.graph.get_graph().draw_mermaid_png())
+        with open("graph.png", "wb") as f:
+            f.write(image.data)
+
     def process_user_input(self, user_input, session_id):
         """
         Process the user input and update the graph state.
@@ -51,9 +70,10 @@ class GraphManager:
         return snapshot[0]["messages"][-1].content
     
 graph_manager = GraphManager()
+graph_manager.save_graph_image()
 
-# Example usage
-response = graph_manager.process_user_input(
-    "hi",
-    1
-)
+# # Example usage
+# response = graph_manager.process_user_input(
+#     "hi",
+#     1
+# )

src/chatbot/nodes.py

@@ -4,28 +4,77 @@ from typing import Annotated
 from typing_extensions import TypedDict
 from langgraph.graph.message import add_messages
 from langchain.tools import tool
-from langgraph.prebuilt import create_react_agent
+from langgraph.prebuilt import create_react_agent, ToolNode
 from langgraph.checkpoint.memory import MemorySaver
 
 from agent_llm_engine import llm_overall_agent
 from prompts.agent_prompts import speaker_system_message, auditor_system_message
 from form_management.form_management import Form
+from vlm.kie import receipt_kie
+import ast
 
-# placeholder form
-form = Form()
+# placeholder, init empty form info
+form_info = {
+        "receipt_title": "",
+        "project_name": "",
+        "date": "",
+        "receipt_number": "",
+        "payer_name": "",
+        "payment_details_table": {
+            "column_headers": "",
+            "row_entries": "",
+        },
+        "total_payment_amount": "",
+        "incompleteness_description": ""
+    }
 
 class State(TypedDict):
     messages: Annotated[list, add_messages]
     session_id: str
+    form_info: dict
 
+# @tool
+# def placeholder_tool(input: str) -> str:
+#     """
+#     Placeholder tool that does nothing.
+#     """
+#     return "This is a placeholder tool."
+
+# session_id: str might need to be added later when there are concurrent users
 @tool
-def placeholder_tool(input: str) -> str:
+def apply_ocr(input: str) -> str:
     """
-    Placeholder tool that does nothing.
+    Apply optical-character-recognition (OCR) to the base64 string (input image) and initalize the form.
+    Feedback will then be provided on whether the form is complete or not.
+    Use this when the user provides a base64 string.
     """
-    return "This is a placeholder tool."
+    kie_info = receipt_kie(input)
+    # overwrite form with kie info
+    try:
+        form_info = ast.literal_eval(kie_info)
+    except:
+        print("KIE ERROR")
+ 
+    feedback = auditor_feedback(form_info)
+
+    return feedback
 
-tools = [placeholder_tool]
+@tool
+def edit_form(key: str, value: str) -> str:
+    """
+    This is used to edit the form data.
+    The target field will be assigned with the provided value.
+    Feedback will then be provided on whether the form is complete or not.
+    Use this when the user wants to edit the form data.
+    """
+
+    form_info[key] = value
+    feedback = auditor_feedback(form_info)
+
+    return feedback
+
+
+tools = [apply_ocr, edit_form]
 
 speaker_memory = MemorySaver()
 speaker_agent = create_react_agent(
@@ -35,7 +84,7 @@ speaker_agent = create_react_agent(
     checkpointer=speaker_memory
 )
 
-def speaker_chatbot(state: State):
+def speaker_chatbot_node(state: State):
     config = {"configurable": {"thread_id": state["session_id"]}}
 
     response = speaker_agent.invoke(
@@ -54,6 +103,8 @@ def speaker_chatbot(state: State):
         "session_id": state["session_id"]
     }
 
+tool_node = ToolNode(tools=[apply_ocr, edit_form])
+
 auditor_memory = MemorySaver()
 auditor_agent = create_react_agent(
     llm_overall_agent,
@@ -62,11 +113,8 @@ auditor_agent = create_react_agent(
     checkpointer=speaker_memory
 )
 
-def auditor_feedback(state: State):
-    # get form based on session id, but for now just use placeholder
-    form_info = form.get_info_as_dict()
-
+# provides feedback based on form info
+def auditor_feedback(form_info):
     # auditor looks at form info and acceptance criteria and gives speaker feedback
 
-    
-    pass
+    return "Form is complete."

src/{vlm → chatbot/vlm}/kie.py

@@ -6,8 +6,8 @@ import openai
 import base64
 from dotenv import load_dotenv
 import os
-from prompts import kie_prompt
-
+from vlm.prompts import kie_prompt
+import ast
 # Load environment variables from .env file
 load_dotenv()
 client = openai.OpenAI()
@@ -47,6 +47,7 @@ def receipt_kie(image_path: str) -> str:
 
     return response.output_text
 
-# test case
-res = receipt_kie(r"C:\Users\Gavin\Desktop\hello-earth\data\jpeg_raw_data\bill\กองทุนพัฒนาบ้านห้วยทราย อ.แม่ออน จ.เชียงใหม่_page-0004.jpg")
-print(res)
+# # test case
+# res = receipt_kie(r"C:\Users\Gavin\Desktop\hello-earth\data\jpeg_raw_data\bill\กองทุนพัฒนาบ้านห้วยทราย อ.แม่ออน จ.เชียงใหม่_page-0004.jpg")
+# res = ast.literal_eval(res)
+# print(res)

src/poetry.lock

@@ -46,6 +46,22 @@ doc = ["Sphinx (>=8.2,<9.0)", "packaging", "sphinx-autodoc-typehints (>=1.2.0)",
 test = ["anyio[trio]", "blockbuster (>=1.5.23)", "coverage[toml] (>=7)", "exceptiongroup (>=1.2.0)", "hypothesis (>=4.0)", "psutil (>=5.9)", "pytest (>=7.0)", "trustme", "truststore (>=0.9.1) ; python_version >= \"3.10\"", "uvloop (>=0.21) ; platform_python_implementation == \"CPython\" and platform_system != \"Windows\" and python_version < \"3.14\""]
 trio = ["trio (>=0.26.1)"]
 
+[[package]]
+name = "asttokens"
+version = "3.0.0"
+description = "Annotate AST trees with source code positions"
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+files = [
+    {file = "asttokens-3.0.0-py3-none-any.whl", hash = "sha256:e3078351a059199dd5138cb1c706e6430c05eff2ff136af5eb4790f9d28932e2"},
+    {file = "asttokens-3.0.0.tar.gz", hash = "sha256:0dcd8baa8d62b0c1d118b399b2ddba3c4aff271d0d7a9e0d4c1681c79035bbc7"},
+]
+
+[package.extras]
+astroid = ["astroid (>=2,<4)"]
+test = ["astroid (>=2,<4)", "pytest", "pytest-cov", "pytest-xdist"]
+
 [[package]]
 name = "audioop-lts"
 version = "0.2.1"
@@ -308,12 +324,24 @@ description = "Cross-platform colored terminal text."
 optional = false
 python-versions = "!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*,!=3.5.*,!=3.6.*,>=2.7"
 groups = ["main"]
-markers = "platform_system == \"Windows\""
+markers = "platform_system == \"Windows\" or sys_platform == \"win32\""
 files = [
     {file = "colorama-0.4.6-py2.py3-none-any.whl", hash = "sha256:4f1d9991f5acc0ca119f9d443620b77f9d6b33703e51011c16baf57afb285fc6"},
     {file = "colorama-0.4.6.tar.gz", hash = "sha256:08695f5cb7ed6e0531a20572697297273c47b8cae5a63ffc6d6ed5c201be6e44"},
 ]
 
+[[package]]
+name = "decorator"
+version = "5.2.1"
+description = "Decorators for Humans"
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+files = [
+    {file = "decorator-5.2.1-py3-none-any.whl", hash = "sha256:d316bb415a2d9e2d2b3abcc4084c6502fc09240e292cd76a76afc106a1c8e04a"},
+    {file = "decorator-5.2.1.tar.gz", hash = "sha256:65f266143752f734b0a7cc83c46f4618af75b8c5911b00ccb61d0ac9b6da0360"},
+]
+
 [[package]]
 name = "distro"
 version = "1.9.0"
@@ -340,6 +368,21 @@ files = [
 [package.dependencies]
 python-dotenv = "*"
 
+[[package]]
+name = "executing"
+version = "2.2.0"
+description = "Get the currently executing AST node of a frame, and other information"
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+files = [
+    {file = "executing-2.2.0-py2.py3-none-any.whl", hash = "sha256:11387150cad388d62750327a53d3339fad4888b39a6fe233c3afbb54ecffd3aa"},
+    {file = "executing-2.2.0.tar.gz", hash = "sha256:5d108c028108fe2551d1a7b2e8b713341e2cb4fc0aa7dcf966fa4327a5226755"},
+]
+
+[package.extras]
+tests = ["asttokens (>=2.1.0)", "coverage", "coverage-enable-subprocess", "ipython", "littleutils", "pytest", "rich ; python_version >= \"3.11\""]
+
 [[package]]
 name = "fastapi"
 version = "0.115.12"
@@ -692,6 +735,73 @@ files = [
 [package.extras]
 all = ["flake8 (>=7.1.1)", "mypy (>=1.11.2)", "pytest (>=8.3.2)", "ruff (>=0.6.2)"]
 
+[[package]]
+name = "ipython"
+version = "9.2.0"
+description = "IPython: Productive Interactive Computing"
+optional = false
+python-versions = ">=3.11"
+groups = ["main"]
+files = [
+    {file = "ipython-9.2.0-py3-none-any.whl", hash = "sha256:fef5e33c4a1ae0759e0bba5917c9db4eb8c53fee917b6a526bd973e1ca5159f6"},
+    {file = "ipython-9.2.0.tar.gz", hash = "sha256:62a9373dbc12f28f9feaf4700d052195bf89806279fc8ca11f3f54017d04751b"},
+]
+
+[package.dependencies]
+colorama = {version = "*", markers = "sys_platform == \"win32\""}
+decorator = "*"
+ipython-pygments-lexers = "*"
+jedi = ">=0.16"
+matplotlib-inline = "*"
+pexpect = {version = ">4.3", markers = "sys_platform != \"win32\" and sys_platform != \"emscripten\""}
+prompt_toolkit = ">=3.0.41,<3.1.0"
+pygments = ">=2.4.0"
+stack_data = "*"
+traitlets = ">=5.13.0"
+
+[package.extras]
+all = ["ipython[doc,matplotlib,test,test-extra]"]
+black = ["black"]
+doc = ["docrepr", "exceptiongroup", "intersphinx_registry", "ipykernel", "ipython[test]", "matplotlib", "setuptools (>=18.5)", "sphinx (>=1.3)", "sphinx-rtd-theme", "sphinx_toml (==0.0.4)", "typing_extensions"]
+matplotlib = ["matplotlib"]
+test = ["packaging", "pytest", "pytest-asyncio (<0.22)", "testpath"]
+test-extra = ["curio", "ipykernel", "ipython[test]", "jupyter_ai", "matplotlib (!=3.2.0)", "nbclient", "nbformat", "numpy (>=1.23)", "pandas", "trio"]
+
+[[package]]
+name = "ipython-pygments-lexers"
+version = "1.1.1"
+description = "Defines a variety of Pygments lexers for highlighting IPython code."
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+files = [
+    {file = "ipython_pygments_lexers-1.1.1-py3-none-any.whl", hash = "sha256:a9462224a505ade19a605f71f8fa63c2048833ce50abc86768a0d81d876dc81c"},
+    {file = "ipython_pygments_lexers-1.1.1.tar.gz", hash = "sha256:09c0138009e56b6854f9535736f4171d855c8c08a563a0dcd8022f78355c7e81"},
+]
+
+[package.dependencies]
+pygments = "*"
+
+[[package]]
+name = "jedi"
+version = "0.19.2"
+description = "An autocompletion tool for Python that can be used for text editors."
+optional = false
+python-versions = ">=3.6"
+groups = ["main"]
+files = [
+    {file = "jedi-0.19.2-py2.py3-none-any.whl", hash = "sha256:a8ef22bde8490f57fe5c7681a3c83cb58874daf72b4784de3cce5b6ef6edb5b9"},
+    {file = "jedi-0.19.2.tar.gz", hash = "sha256:4770dc3de41bde3966b02eb84fbcf557fb33cce26ad23da12c742fb50ecb11f0"},
+]
+
+[package.dependencies]
+parso = ">=0.8.4,<0.9.0"
+
+[package.extras]
+docs = ["Jinja2 (==2.11.3)", "MarkupSafe (==1.1.1)", "Pygments (==2.8.1)", "alabaster (==0.7.12)", "babel (==2.9.1)", "chardet (==4.0.0)", "commonmark (==0.8.1)", "docutils (==0.17.1)", "future (==0.18.2)", "idna (==2.10)", "imagesize (==1.2.0)", "mock (==1.0.1)", "packaging (==20.9)", "pyparsing (==2.4.7)", "pytz (==2021.1)", "readthedocs-sphinx-ext (==2.1.4)", "recommonmark (==0.5.0)", "requests (==2.25.1)", "six (==1.15.0)", "snowballstemmer (==2.1.0)", "sphinx (==1.8.5)", "sphinx-rtd-theme (==0.4.3)", "sphinxcontrib-serializinghtml (==1.1.4)", "sphinxcontrib-websupport (==1.2.4)", "urllib3 (==1.26.4)"]
+qa = ["flake8 (==5.0.4)", "mypy (==0.971)", "types-setuptools (==67.2.0.1)"]
+testing = ["Django", "attrs", "colorama", "docopt", "pytest (<9.0.0)"]
+
 [[package]]
 name = "jinja2"
 version = "3.1.6"
@@ -1108,6 +1218,21 @@ files = [
     {file = "markupsafe-3.0.2.tar.gz", hash = "sha256:ee55d3edf80167e48ea11a923c7386f4669df67d7994554387f84e7d8b0a2bf0"},
 ]
 
+[[package]]
+name = "matplotlib-inline"
+version = "0.1.7"
+description = "Inline Matplotlib backend for Jupyter"
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+files = [
+    {file = "matplotlib_inline-0.1.7-py3-none-any.whl", hash = "sha256:df192d39a4ff8f21b1895d72e6a13f5fcc5099f00fa84384e0ea28c2cc0653ca"},
+    {file = "matplotlib_inline-0.1.7.tar.gz", hash = "sha256:8423b23ec666be3d16e16b60bdd8ac4e86e840ebd1dd11a30b9f117f2fa0ab90"},
+]
+
+[package.dependencies]
+traitlets = "*"
+
 [[package]]
 name = "mdurl"
 version = "0.1.2"
@@ -1441,6 +1566,38 @@ sql-other = ["SQLAlchemy (>=2.0.0)", "adbc-driver-postgresql (>=0.8.0)", "adbc-d
 test = ["hypothesis (>=6.46.1)", "pytest (>=7.3.2)", "pytest-xdist (>=2.2.0)"]
 xml = ["lxml (>=4.9.2)"]
 
+[[package]]
+name = "parso"
+version = "0.8.4"
+description = "A Python Parser"
+optional = false
+python-versions = ">=3.6"
+groups = ["main"]
+files = [
+    {file = "parso-0.8.4-py2.py3-none-any.whl", hash = "sha256:a418670a20291dacd2dddc80c377c5c3791378ee1e8d12bffc35420643d43f18"},
+    {file = "parso-0.8.4.tar.gz", hash = "sha256:eb3a7b58240fb99099a345571deecc0f9540ea5f4dd2fe14c2a99d6b281ab92d"},
+]
+
+[package.extras]
+qa = ["flake8 (==5.0.4)", "mypy (==0.971)", "types-setuptools (==67.2.0.1)"]
+testing = ["docopt", "pytest"]
+
+[[package]]
+name = "pexpect"
+version = "4.9.0"
+description = "Pexpect allows easy control of interactive console applications."
+optional = false
+python-versions = "*"
+groups = ["main"]
+markers = "sys_platform != \"win32\" and sys_platform != \"emscripten\""
+files = [
+    {file = "pexpect-4.9.0-py2.py3-none-any.whl", hash = "sha256:7236d1e080e4936be2dc3e326cec0af72acf9212a7e1d060210e70a47e253523"},
+    {file = "pexpect-4.9.0.tar.gz", hash = "sha256:ee7d41123f3c9911050ea2c2dac107568dc43b2d3b0c7557a33212c398ead30f"},
+]
+
+[package.dependencies]
+ptyprocess = ">=0.5"
+
 [[package]]
 name = "pillow"
 version = "11.2.1"
@@ -1541,6 +1698,49 @@ tests = ["check-manifest", "coverage (>=7.4.2)", "defusedxml", "markdown2", "ole
 typing = ["typing-extensions ; python_version < \"3.10\""]
 xmp = ["defusedxml"]
 
+[[package]]
+name = "prompt-toolkit"
+version = "3.0.51"
+description = "Library for building powerful interactive command lines in Python"
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+files = [
+    {file = "prompt_toolkit-3.0.51-py3-none-any.whl", hash = "sha256:52742911fde84e2d423e2f9a4cf1de7d7ac4e51958f648d9540e0fb8db077b07"},
+    {file = "prompt_toolkit-3.0.51.tar.gz", hash = "sha256:931a162e3b27fc90c86f1b48bb1fb2c528c2761475e57c9c06de13311c7b54ed"},
+]
+
+[package.dependencies]
+wcwidth = "*"
+
+[[package]]
+name = "ptyprocess"
+version = "0.7.0"
+description = "Run a subprocess in a pseudo terminal"
+optional = false
+python-versions = "*"
+groups = ["main"]
+markers = "sys_platform != \"win32\" and sys_platform != \"emscripten\""
+files = [
+    {file = "ptyprocess-0.7.0-py2.py3-none-any.whl", hash = "sha256:4b41f3967fce3af57cc7e94b888626c18bf37a083e3651ca8feeb66d492fef35"},
+    {file = "ptyprocess-0.7.0.tar.gz", hash = "sha256:5c5d0a3b48ceee0b48485e0c26037c0acd7d29765ca3fbb5cb3831d347423220"},
+]
+
+[[package]]
+name = "pure-eval"
+version = "0.2.3"
+description = "Safely evaluate AST nodes without side effects"
+optional = false
+python-versions = "*"
+groups = ["main"]
+files = [
+    {file = "pure_eval-0.2.3-py3-none-any.whl", hash = "sha256:1db8e35b67b3d218d818ae653e27f06c3aa420901fa7b081ca98cbedc874e0d0"},
+    {file = "pure_eval-0.2.3.tar.gz", hash = "sha256:5f4e983f40564c576c7c8635ae88db5956bb2229d7e9237d03b3c0b0190eaf42"},
+]
+
+[package.extras]
+tests = ["pytest"]
+
 [[package]]
 name = "pycparser"
 version = "2.22"
@@ -1707,7 +1907,6 @@ description = "Pygments is a syntax highlighting package written in Python."
 optional = false
 python-versions = ">=3.8"
 groups = ["main"]
-markers = "sys_platform != \"emscripten\""
 files = [
     {file = "pygments-2.19.1-py3-none-any.whl", hash = "sha256:9ea1544ad55cecf4b8242fab6dd35a93bbce657034b0611ee383099054ab6d8c"},
     {file = "pygments-2.19.1.tar.gz", hash = "sha256:61c16d2a8576dc0649d9f39e089b5f02bcd27fba10d8fb4dcc28173f7a45151f"},
@@ -2190,6 +2389,26 @@ postgresql-psycopgbinary = ["psycopg[binary] (>=3.0.7)"]
 pymysql = ["pymysql"]
 sqlcipher = ["sqlcipher3_binary"]
 
+[[package]]
+name = "stack-data"
+version = "0.6.3"
+description = "Extract data from python stack frames and tracebacks for informative displays"
+optional = false
+python-versions = "*"
+groups = ["main"]
+files = [
+    {file = "stack_data-0.6.3-py3-none-any.whl", hash = "sha256:d5558e0c25a4cb0853cddad3d77da9891a08cb85dd9f9f91b9f8cd66e511e695"},
+    {file = "stack_data-0.6.3.tar.gz", hash = "sha256:836a778de4fec4dcd1dcd89ed8abff8a221f58308462e1c4aa2a3cf30148f0b9"},
+]
+
+[package.dependencies]
+asttokens = ">=2.1.0"
+executing = ">=1.2.0"
+pure-eval = "*"
+
+[package.extras]
+tests = ["cython", "littleutils", "pygments", "pytest", "typeguard"]
+
 [[package]]
 name = "starlette"
 version = "0.46.2"
@@ -2306,6 +2525,22 @@ notebook = ["ipywidgets (>=6)"]
 slack = ["slack-sdk"]
 telegram = ["requests"]
 
+[[package]]
+name = "traitlets"
+version = "5.14.3"
+description = "Traitlets Python configuration system"
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+files = [
+    {file = "traitlets-5.14.3-py3-none-any.whl", hash = "sha256:b74e89e397b1ed28cc831db7aea759ba6640cb3de13090ca145426688ff1ac4f"},
+    {file = "traitlets-5.14.3.tar.gz", hash = "sha256:9ed0579d3502c94b4b3732ac120375cda96f923114522847de4b3bb98b96b6b7"},
+]
+
+[package.extras]
+docs = ["myst-parser", "pydata-sphinx-theme", "sphinx"]
+test = ["argcomplete (>=3.0.3)", "mypy (>=1.7.0)", "pre-commit", "pytest (>=7.0,<8.2)", "pytest-mock", "pytest-mypy-testing"]
+
 [[package]]
 name = "typer"
 version = "0.15.3"
@@ -2402,6 +2637,18 @@ h11 = ">=0.8"
 [package.extras]
 standard = ["colorama (>=0.4) ; sys_platform == \"win32\"", "httptools (>=0.6.3)", "python-dotenv (>=0.13)", "pyyaml (>=5.1)", "uvloop (>=0.14.0,!=0.15.0,!=0.15.1) ; sys_platform != \"win32\" and sys_platform != \"cygwin\" and platform_python_implementation != \"PyPy\"", "watchfiles (>=0.13)", "websockets (>=10.4)"]
 
+[[package]]
+name = "wcwidth"
+version = "0.2.13"
+description = "Measures the displayed width of unicode strings in a terminal"
+optional = false
+python-versions = "*"
+groups = ["main"]
+files = [
+    {file = "wcwidth-0.2.13-py2.py3-none-any.whl", hash = "sha256:3da69048e4540d84af32131829ff948f1e022c1c6bdb8d6102117aac784f6859"},
+    {file = "wcwidth-0.2.13.tar.gz", hash = "sha256:72ea0c06399eb286d978fdedb6923a9eb47e1c486ce63e9b4e64fc18303972b5"},
+]
+
 [[package]]
 name = "websockets"
 version = "15.0.1"
@@ -2730,4 +2977,4 @@ cffi = ["cffi (>=1.11)"]
 [metadata]
 lock-version = "2.1"
 python-versions = ">=3.12.4,<4.0.0"
-content-hash = "3293db5a94662f3c1087e04a9f805f71f32ec3efbab1302d4d9eaa48d1b6a4b0"
+content-hash = "efbddde7effbf9adb8d0fd2294ffca2d5f4460bf12f75cfab57a0a47ca868c0e"

src/pyproject.toml

@@ -13,7 +13,8 @@ dependencies = [
     "dotenv (>=0.9.9,<0.10.0)",
     "langgraph (>=0.4.1,<0.5.0)",
     "langchain (>=0.3.24,<0.4.0)",
-    "langchain-openai (>=0.3.14,<0.4.0)"
+    "langchain-openai (>=0.3.14,<0.4.0)",
+    "ipython (>=9.2.0,<10.0.0)"
 ]