Update escalation_server.py

escalation_server.py

@@ -104,12 +104,13 @@ 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 = (await 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:
-        await ctx.disable_components(tools=["explain_from_resource", "generate_quick_revision"])
+        # await ctx.disable_components(tools=["explain_from_resource", "generate_quick_revision"])
+        await ctx.disable_components(names=["explain_from_resource", "generate_quick_revision"])
         return (
             "FATAL_ERROR: TUTORING_LIMIT_REACHED. "
             "Internal tutoring tools have been deactivated for this session. "
@@ -119,9 +120,8 @@ async def explain_from_resource(concept: str, ctx: Context = CurrentContext()):
             "3. Execute get_teacher_info and escalate_to_teacher."
         )
 
-    # FIX 2: Remove 'await' from ctx.set_state
-    # ctx.set_state("math_attempts", attempts + 1)
-    await ctx.set_state("math_attempts", attempts + 1)
+    new_attempts = attempts + 1
+    ctx.set_state("math_attempts", new_attempts)
     
     # 2. FETCH RESOURCE
     material = read_resource_file() 
@@ -129,7 +129,7 @@ async def explain_from_resource(concept: str, ctx: Context = CurrentContext()):
     # 3. STRUCTURED RESPONSE
     # Note: We return the prompt back to the LLM so it can generate the explanation
     return f"""
-    <attempt>{attempts + 1}/3</attempt>
+    <attempt>{new_attempts}/3</attempt>
     <content>{material}</content>
     <instruction>Explain {concept} concisely. Use LaTeX.</instruction>
     """
@@ -141,19 +141,20 @@ 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 = (await 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"])
+        # await ctx.disable_components(tools=["explain_from_resource", "generate_quick_revision"])
+        await ctx.disable_components(names=["explain_from_resource", "generate_quick_revision"])
         return "FATAL_ERROR: TUTORING_LIMIT_REACHED. Transition to escalation immediately."
 
-    # ctx.set_state("math_attempts", attempts + 1)
-    await ctx.set_state("math_attempts", attempts + 1)
+    new_attempts = attempts + 1
+    ctx.set_state("math_attempts", new_attempts)
     
     material = read_resource_file()
     return f"""
-    <attempt>{attempts + 1}/3</attempt>
+    <attempt>{new_attempts}/3</attempt>
     <material>{material}</material>
     <request>Create a {type} for quick revision.</request>
     <format_requirements>