fix trait parameter

Author: saviorand

lightbug.🔥

@@ -3,6 +3,7 @@ from lightbug_http.middleware.helpers import Success
 from sys import is_defined
 from lightbug_http import *
 
+@value
 struct HelloWorld(HTTPHandler):
     fn handle(self, context: Context) -> HTTPResponse:
         var name = context.params.get("username", "world")

lightbug_http/middleware/helpers.mojo

@@ -1,3 +1,5 @@
+from lightbug_http.http import HTTPRequest, HTTPResponse, ResponseHeader
+
 ### Helper functions to create HTTP responses
 fn Success(body: String) -> HTTPResponse:
     return Success(body, String("text/plain"))

lightbug_http/middleware/middleware.mojo

@@ -6,13 +6,13 @@ from lightbug_http.middleware import *
 ## It is passed to each middleware in the chain.
 ## It also contains a dictionary of parameters that can be shared between middleware.
 @value
-struct Context:
+struct Context[ParamType: CollectionElement]:
     var request: HTTPRequest
-    var params: Dict[String, AnyType]
+    var params: Dict[String, ParamType]
 
     fn __init__(inout self, request: HTTPRequest):
         self.request = request
-        self.params = Dict[String, AnyType]()
+        self.params = Dict[String, ParamType]()
 
 
 ## Middleware is an interface for processing HTTP requests.

lightbug_http/middleware/router.mojo

@@ -1,18 +1,18 @@
 ## HTTPHandler is an interface for handling HTTP requests in the RouterMiddleware.
 ## It is a leaf node in the middleware chain.
-trait HTTPHandler:
+trait HTTPHandler(CollectionElement):
     fn handle(self, context: Context) -> HTTPResponse:
         ...
 
 
 ## Router middleware routes requests to different middleware based on the path.
 @value
-struct RouterMiddleware(Middleware):
+struct RouterMiddleware[HTTPHandlerType: HTTPHandler](Middleware):
     var next: Middleware
-    var routes: Dict[String, HTTPHandler]
+    var routes: Dict[String, HTTPHandlerType]
 
     fn __init__(inout self):
-        self.routes = Dict[String, HTTPHandler]()
+        self.routes = Dict[String, HTTPHandlerType]()
 
     fn set_next(self, next: Middleware):
         self.next = next