test: Update compiler tests for no-wrapper behavior

- create_agent_wrapper now returns code as-is
- Tests verify code is not modified and future imports are preserved

Author: echobt

src/compiler.rs

@@ -1146,20 +1146,22 @@ mod tests {
     use super::*;
 
     #[test]
-    fn test_create_agent_wrapper() {
+    fn test_create_agent_wrapper_no_modification() {
+        // Wrapper no longer modifies code to preserve `from __future__` imports
         let code = r#"
 class MyAgent(Agent):
     def solve(self, req):
         return Response.cmd("ls")
 "#;
         let wrapped = create_agent_wrapper(code);
-        assert!(wrapped.contains("if __name__"));
-        assert!(wrapped.contains("agent_class"));
+        // Code should be returned as-is
+        assert_eq!(wrapped, code);
     }
 
     #[test]
-    fn test_wrapper_preserves_existing_entry() {
-        let code = r#"
+    fn test_wrapper_preserves_future_imports() {
+        let code = r#"from __future__ import annotations
+
 class MyAgent(Agent):
     def solve(self, req):
         return Response.cmd("ls")
@@ -1168,11 +1170,8 @@ if __name__ == "__main__":
     run(MyAgent())
 "#;
         let wrapped = create_agent_wrapper(code);
-        // Should add HTTP mode setup but preserve existing entry point
-        assert!(wrapped.contains("FORCE_HTTP_SERVER"));
-        assert!(wrapped.contains("if __name__"));
-        assert!(wrapped.contains("run(MyAgent())"));
-        // Should not add auto-generated entry point since one exists
-        assert!(!wrapped.contains("Auto-generated entry point"));
+        // Code should be returned as-is, preserving the future import at the start
+        assert_eq!(wrapped, code);
+        assert!(wrapped.starts_with("from __future__"));
     }
 }