update

Author: sansyrox

integration_tests/test_sse.py

@@ -6,8 +6,6 @@
 import time
 import concurrent.futures
 from typing import List, Dict, Any
-from unittest.mock import patch
-import re
 
 import pytest
 import requests
@@ -317,7 +315,6 @@ def test_SSE_Message_formatter(self):
 
     def test_sse_concurrent_connections(self):
         """Test multiple concurrent SSE connections"""
-        import concurrent.futures
 
         def get_events():
             return get_sse_stream("/sse/basic")
@@ -759,8 +756,6 @@ def test_gen():
 
     def test_sse_concurrent_generator_creation(self):
         """Test creating multiple SSE responses concurrently"""
-        import threading
-        import time
 
         responses = []
         errors = []
@@ -1149,7 +1144,6 @@ def test_sse_custom_status_code(self):
 
     def test_sse_concurrent_requests(self):
         """Test multiple concurrent SSE requests"""
-        import concurrent.futures
 
         def get_basic_sse():
             content = get_SSE_Response_quickly("/sse/basic")

robyn/mcp.py

@@ -37,13 +37,13 @@ def _generate_schema_from_function(func: Callable) -> Dict[str, Any]:
 
         # Try to infer type from annotation
         if param.annotation != inspect.Parameter.empty:
-            if param.annotation == str:
+            if param.annotation is str:
                 param_schema["type"] = "string"
-            elif param.annotation == int:
+            elif param.annotation is int:
                 param_schema["type"] = "integer"
-            elif param.annotation == float:
+            elif param.annotation is float:
                 param_schema["type"] = "number"
-            elif param.annotation == bool:
+            elif param.annotation is bool:
                 param_schema["type"] = "boolean"
             elif hasattr(param.annotation, "__origin__"):
                 # Handle generic types like List, Dict, etc.