Update escalation_server.py

escalation_server.py

@@ -99,7 +99,8 @@ async def explain_from_resource(concept: str, ctx: Context = CurrentContext()):
     and triggers teacher escalation after 3 failed attempts.
     """
     # FIX 1: Remove 'await' from ctx.get_state (it is synchronous)
-    attempts = ctx.get_state("math_attempts") or 0
+    # attempts = ctx.get_state("math_attempts") or 0
+    attempts = (await ctx.get_state("math_attempts")) or 0
 
     # 1. CIRCUIT BREAKER: Enforce the 3-attempt limit
     if attempts >= 3:
@@ -114,7 +115,8 @@ async def explain_from_resource(concept: str, ctx: Context = CurrentContext()):
         )
 
     # FIX 2: Remove 'await' from ctx.set_state
-    ctx.set_state("math_attempts", attempts + 1)
+    # ctx.set_state("math_attempts", attempts + 1)
+    await ctx.set_state("math_attempts", attempts + 1)
     
     # 2. FETCH RESOURCE
     material = read_resource_file() 
@@ -134,14 +136,16 @@ async def generate_quick_revision(type: str = "summary", ctx: Context = CurrentC
     Types: 'summary' (3 key points) or 'short_notes' (condensed formulas and definitions).
     """
     # ENFORCE ATTEMPT TRACKING
-    attempts = ctx.get_state("math_attempts") or 0
+    # attempts = ctx.get_state("math_attempts") or 0
+    attempts = (await ctx.get_state("math_attempts")) or 0
     
     if attempts >= 3:
         await ctx.disable_components(tools=["explain_from_resource", "generate_quick_revision"])
         return "FATAL_ERROR: TUTORING_LIMIT_REACHED. Transition to escalation immediately."
 
-    ctx.set_state("math_attempts", attempts + 1)
-
+    # ctx.set_state("math_attempts", attempts + 1)
+    await ctx.set_state("math_attempts", attempts + 1)
+    
     material = read_resource_file()
     return f"""
     <attempt>{attempts + 1}/3</attempt>
@@ -235,7 +239,9 @@ async def escalate_to_teacher(
         await aiosmtplib.send(msg, hostname=SMTP_SERVER, port=SMTP_PORT, 
                               username=SENDER_EMAIL, password=SENDER_PASSWORD, start_tls=True)
         # --- THE FIX: RESET LOGIC ---
-        ctx.set_state("math_attempts", 0) # Reset counter to 0
+        # ctx.set_state("math_attempts", 0) # Reset counter to 0
+        await ctx.set_state("math_attempts", 0)
+        
         # await ctx.enable_components(tools=["explain_from_resource", "generate_quick_revision"]) # Re-enable tools
         
         return f"Success: {teacher_name} ({subject} Expert) has been notified."