move classes/protocols for representing HTTP requests and responses data into a separate file

1 files changed, 4 insertions, 3 deletions

diff --git a/src/hydrilla/proxy/policies/rule.py b/src/hydrilla/proxy/policies/rule.py
index eb70147..bcb110e 100644
--- a/src/hydrilla/proxy/policies/rule.py
+++ b/src/hydrilla/proxy/policies/rule.py
@@ -37,6 +37,7 @@ import typing as t
 from ...url_patterns import ParsedPattern
 from .. import csp
 from .. import state
+from ..import http_messages
 from . import base
 
 
@@ -50,7 +51,7 @@ class BlockPolicy(base.Policy):
 
     priority: t.ClassVar[base.PolicyPriority] = base.PolicyPriority._TWO
 
-    def _modify_headers(self, response_info: base.ResponseInfo) \
+    def _modify_headers(self, response_info: http_messages.ResponseInfo) \
         -> t.Iterable[tuple[bytes, bytes]]:
         """...."""
         csp_policies = csp.extract(response_info.headers)
@@ -80,8 +81,8 @@ class BlockPolicy(base.Policy):
         yield b'Content-Security-Policy', extra_csp.encode()
 
 
-    def consume_response(self, response_info: base.ResponseInfo) \
-        -> base.ProducedResponse:
+    def consume_response(self, response_info: http_messages.ResponseInfo) \
+        -> http_messages.ProducedResponse:
         """...."""
         new_response = response_info.make_produced_response()