Update app.py

fix agent error

app.py

@@ -28,7 +28,8 @@ try:
     from langchain_community.callbacks import StreamlitCallbackHandler
     from langchain_community.document_loaders import PyPDFLoader
     from langchain_text_splitters import CharacterTextSplitter
-    from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
+    # Added PromptTemplate explicitly here
+    from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder, PromptTemplate
     from langchain_core.runnables import RunnablePassthrough
     from langchain_core.output_parsers import StrOutputParser
     from langchain_core.tools import tool
@@ -146,11 +147,9 @@ def update_metrics(placeholder):
 with st.sidebar:
     st.header("🔐 Authentication")
     
-    # Initialize Session State for Key
     if 'is_key_valid' not in st.session_state:
         st.session_state['is_key_valid'] = False
 
-    # MANUAL ENTRY ONLY (No Secret Check)
     if not st.session_state['is_key_valid']:
         api_key_input = st.text_input("Enter Groq API Key", type="password", key="input_key")
         if st.button("Validate API Key"):
@@ -211,7 +210,6 @@ if st.session_state.get('is_key_valid', False):
     # --- RAG SETUP ---
     @st.cache_resource
     def setup_rag():
-        # Check global variable here
         if pdfs_missing:
             st.error(f"Missing PDFs: {pdfs_missing}")
             st.stop()
@@ -251,7 +249,7 @@ if st.session_state.get('is_key_valid', False):
 
     tools = [get_credit_score, get_account_status, check_pr_status, consult_policy_doc]
 
-    # --- PROMPT TEMPLATE (Steps Enforcement) ---
+    # --- DEFINING THE STRICT STEP-BY-STEP PROMPT ---
     template = """You are a Loan Risk Officer. You must execute the loan assessment process in exactly 4 sequential steps.
 
     TOOLS AVAILABLE:
@@ -293,7 +291,12 @@ if st.session_state.get('is_key_valid', False):
     Question: {input}
     Thought:{agent_scratchpad}"""
 
-    prompt = ChatPromptTemplate.from_template(template)
+    prompt = PromptTemplate.from_template(template)
+    
+    # --- CREATING THE AGENT (THE MISSING STEP) ---
+    agent = create_tool_calling_agent(llm, tools, prompt)
+    
+    # --- CREATING THE EXECUTOR ---
     agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True, return_intermediate_steps=True)
 
     col1, col2 = st.columns([1, 2])
@@ -320,26 +323,14 @@ if st.session_state.get('is_key_valid', False):
                 1. DO NOT query 'get_credit_score' or 'account_status' for Score/Status.
                 2. USE: Score: {sim_score}, Status: {sim_status}
                 3. Query 'get_account_status' ONLY for Name/Nationality.
-                4. Consult Policy Docs for risk/rates.
-                5. Provide a Final Recommendation Report that MUST include:
-                   - Customer Name, ID, Email
-                   - Risk Level, Interest Rate
-                   - Final Decision (Approve/Reject)
-                   - Justification for Decision (Cite specific PDF policies)
-                   - Format in a clear markdown table.
+                4. Follow the 4-step process and Output Final Report in the specified format.
                 """
             else:
                 query = f"""
                 Process Loan for Customer ID: {uid}.
                 1. Query SQL tools for Name, Email, Nationality, Status, Score.
                 2. IF Nationality is 'Singaporean', SKIP 'check_pr_status'.
-                3. Consult Policy Docs for risk/rates.
-                4.  Provide a Final Recommendation Report that MUST include:
-                   - Customer Name, ID, Email
-                   - Risk Level, Interest Rate
-                   - Final Decision (Approve/Reject)
-                   - Justification for Decision (Cite specific PDF policies)
-                   - Format in a clear markdown table.
+                3. Follow the 4-step process and Output Final Report in the specified format.
                 """
             
             with st.status("🤖 Agent is processing...", expanded=True) as status:
@@ -358,6 +349,11 @@ if st.session_state.get('is_key_valid', False):
             st.success("### 📋 Final Recommendation")
             st.markdown(res['output'])
             
+            with st.expander("🔍 Detailed Trace"):
+                steps = res.get("intermediate_steps", [])
+                for i, (action, observation) in enumerate(steps):
+                    st.markdown(f"**Step {i+1}:** Tool `{action.tool}` | Output: `{observation}`")
+
             if not use_simulation:
                 st.divider()
                 with st.expander("✉️ Draft Email"):
@@ -365,11 +361,6 @@ if st.session_state.get('is_key_valid', False):
                     with st.spinner("Drafting..."):
                         email_draft = llm.invoke(email_prompt).content
                     st.text_area("Email Draft", value=email_draft, height=200)
-            
-            with st.expander("🔍 Detailed Trace"):
-                steps = res.get("intermediate_steps", [])
-                for i, (action, observation) in enumerate(steps):
-                    st.markdown(f"**Step {i+1}:** Tool `{action.tool}` | Output: `{observation}`")
 
 elif not st.session_state.get('is_key_valid', False):
     st.info("👈 Please validate your Groq API Key.")