Fixed context usage calculation

sia/web_agent.py

@@ -33,20 +33,28 @@ class WebAgent(BaseAgent):
     Broadcasts state changes to registered handlers.
     """
     
-    def __init__(self,
-                system_prompt: str,
-                action_schema: str, 
-                working_memory: WorkingMemory,
-                system_metrics: SystemMetrics,
-                llm: LlmEngine,
-                validator: XMLValidator,
-                parser: ResponseParser):
+    def __init__(
+            self,
+            system_prompt: str,
+            action_schema: str, 
+            working_memory: WorkingMemory,
+            metrics: SystemMetrics,
+            llm: LlmEngine,
+            validator: XMLValidator,
+            parser: ResponseParser
+        ):
         """
         Initialize web agent with required components.
         """
-        super().__init__(action_schema, working_memory, system_metrics, llm, validator, parser)
+        super().__init__(
+            system_prompt,
+            action_schema,
+            working_memory,
+            metrics, llm,
+            validator,
+            parser
+        )
         self._response = ""
-        self._system_prompt = system_prompt
         self._state = WebAgentState.CONTEXT_APPROVAL
         self._validation_error: Optional[str] = None
         self._state_change_handlers: List[Callable[[WebAgentState], None]] = []
@@ -148,7 +156,7 @@ class WebAgent(BaseAgent):
                 
     def _approve_context_thread(self) -> None:
         self._set_response("")
-        response_token_iter = self._llm.infer(f"{self._system_prompt}\n{self._action_schema}", self.context)
+        response_token_iter = self._llm.infer(self.system_prompt, self.context)
         response = ""
         for token in response_token_iter:
             response += token