merge with recent branch

Author: SoldierSacha

tests/server/test_session.py

@@ -109,7 +109,11 @@ async def list_resources():
 
     # Add a complete handler
     @server.completion()
-    async def complete(ref: PromptReference | ResourceReference, argument: CompletionArgument):
+    async def complete(
+        ref: PromptReference | types.ResourceTemplateReference,
+        argument: CompletionArgument,
+        context: types.CompletionContext | None,
+    ):
         return Completion(
             values=["completion1", "completion2"],
         )