more graph setup, comments on how the flow should work

src/chatbot/graph_manager.py

@@ -0,0 +1,59 @@
+# graph_manager.py
+# graph definition and compilation
+
+from langgraph.graph import StateGraph, START, END
+from langgraph.checkpoint.memory import MemorySaver
+from nodes import (
+    State,
+    speaker_chatbot
+)
+from form_management.form_management import Form
+import sys
+
+class GraphManager:
+    def __init__(self):
+        self.graph = None
+        self.init_graph()
+
+    # user input -> 
+    #   if user intent is clear -> speaker form management -> audit review and feedback -> speaker generate response based on feedback
+    #   if user intent is unclear -> speaker generate response for clarification
+    def init_graph(self):
+        graph_memory = MemorySaver()
+        graph_builder = StateGraph(State)
+
+        graph_builder.add_node("speaker_chatbot", 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 process_user_input(self, user_input, session_id):
+        """
+        Process the user input and update the graph state.
+        """
+        config = {"configurable": {"thread_id": session_id}}
+
+        input_dict = {
+            "messages": [("user", user_input)],
+            "session_id": session_id
+        }
+
+        for event in self.graph.stream(
+            input_dict,
+            config,
+            stream_mode="values"
+        ):
+            event["messages"][-1].pretty_print()
+
+        snapshot = self.graph.get_state(config)
+        return snapshot[0]["messages"][-1].content
+    
+graph_manager = GraphManager()
+
+# Example usage
+response = graph_manager.process_user_input(
+    "hi",
+    1
+)

src/chatbot/nodes.py

@@ -0,0 +1,72 @@
+# nodes.py
+
+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.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
+
+# placeholder form
+form = Form()
+
+class State(TypedDict):
+    messages: Annotated[list, add_messages]
+    session_id: str
+
+@tool
+def placeholder_tool(input: str) -> str:
+    """
+    Placeholder tool that does nothing.
+    """
+    return "This is a placeholder tool."
+
+tools = [placeholder_tool]
+
+speaker_memory = MemorySaver()
+speaker_agent = create_react_agent(
+    llm_overall_agent,
+    tools,
+    state_modifier=speaker_system_message,
+    checkpointer=speaker_memory
+)
+
+def speaker_chatbot(state: State):
+    config = {"configurable": {"thread_id": state["session_id"]}}
+
+    response = speaker_agent.invoke(
+        {
+            "messages": [
+                (
+                    "user", "{}".format(state["messages"][-1].content)
+                )
+            ]
+        },
+        config,        
+    )["messages"][-1]
+
+    return {
+        "messages": response.content,
+        "session_id": state["session_id"]
+    }
+
+auditor_memory = MemorySaver()
+auditor_agent = create_react_agent(
+    llm_overall_agent,
+    tools,
+    state_modifier=auditor_system_message,
+    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()
+
+    # auditor looks at form info and acceptance criteria and gives speaker feedback
+
+    
+    pass

src/chatbot/prompts/agent_prompts.py

@@ -0,0 +1,7 @@
+speaker_system_message = """
+You are an assistant that helps users submit receipts and fill in a form.
+Your job is to communicate with the user and determine which tools to use given the user input.
+"""
+
+auditor_system_message = """
+"""

src/form_management/acceptance_criteria.json

@@ -1,32 +0,0 @@
-{
-    "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/form_management/form_management.py

@@ -1,72 +0,0 @@
-class Form:
-    class PaymentDetailsTable:
-        def __init__(self, column_headers, row_entries):
-            self.column_headers = column_headers
-            self.row_entries = row_entries
-
-    def __init__(self, receipt_title, project_name, date, receipt_number, payer_name, payment_details_table, total_payment_amount, incompleteness_description):
-        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(**payment_details_table)
-        self.total_payment_amount = total_payment_amount
-        self.incompleteness_description = 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
-

src/main.py

@@ -1,3 +1,5 @@
+# from chatbot.graph_manager
+
 def main():
     print("Hello, World!")
 

src/pyproject.toml

@@ -6,11 +6,14 @@ authors = [
     {name = "suratan.boon",email = "suratan.boon@cjexpress.co.th"}
 ]
 readme = "README.md"
-requires-python = ">=3.11"
+requires-python = ">=3.12.4,<4.0.0"
 dependencies = [
     "gradio (>=5.28.0,<6.0.0)",
     "openai (>=1.76.2,<2.0.0)",
-    "dotenv (>=0.9.9,<0.10.0)"
+    "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)"
 ]