ci: add type checking to CI

.github/workflows/ci.yml

@@ -21,7 +21,7 @@ jobs:
       - name: Install dependencies
         run: poetry install
       - name: Compile
-        run: poetry run mypy .
+        run: poetry run mypy . --strict
   test:
     runs-on: ubuntu-latest
     steps:

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 AgentMail
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

src/agentmail/__init__.py

@@ -438,7 +438,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/agent/__init__.py

@@ -31,7 +31,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/agent/types/__init__.py

@@ -34,7 +34,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/api_keys/__init__.py

@@ -46,7 +46,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/api_keys/types/__init__.py

@@ -44,7 +44,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/attachments/__init__.py

@@ -46,7 +46,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/attachments/types/__init__.py

@@ -44,7 +44,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/core/__init__.py

@@ -92,7 +92,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/core/events.py

@@ -18,7 +18,7 @@ class EventEmitterMixin:
     """
 
     def __init__(self) -> None:
-        self._callbacks: typing.Dict[EventType, typing.List[typing.Callable]] = {}
+        self._callbacks: typing.Dict[EventType, typing.List[typing.Callable[..., typing.Any]]] = {}
 
     def on(self, event_name: EventType, callback: typing.Callable[[typing.Any], typing.Any]) -> None:
         if event_name not in self._callbacks:

src/agentmail/core/http_sse/__init__.py

@@ -34,7 +34,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/core/query_encoder.py

@@ -1,7 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from typing import Any, Dict, List, Optional, Tuple
 import json
+from typing import Any, Dict, List, Optional, Tuple
 
 import pydantic
 

src/agentmail/domains/__init__.py

@@ -56,7 +56,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/domains/types/__init__.py

@@ -54,7 +54,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/drafts/__init__.py

@@ -70,7 +70,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/drafts/types/__init__.py

@@ -68,7 +68,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/errors/__init__.py

@@ -34,7 +34,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/events/__init__.py

@@ -68,7 +68,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/events/types/__init__.py

@@ -66,7 +66,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/inbox_events/__init__.py

@@ -31,7 +31,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/inbox_events/types/__init__.py

@@ -34,7 +34,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/inboxes/__init__.py

@@ -52,7 +52,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/inboxes/types/__init__.py

@@ -42,7 +42,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/lists/__init__.py

@@ -46,7 +46,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/lists/types/__init__.py

@@ -44,7 +44,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/messages/__init__.py

@@ -106,7 +106,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/messages/errors/__init__.py

@@ -26,7 +26,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/messages/types/__init__.py

@@ -102,7 +102,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/metrics/__init__.py

@@ -46,7 +46,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/metrics/types/__init__.py

@@ -44,7 +44,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/organizations/__init__.py

@@ -26,7 +26,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/organizations/types/__init__.py

@@ -26,7 +26,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/pods/__init__.py

@@ -41,7 +41,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/pods/types/__init__.py

@@ -38,7 +38,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/threads/__init__.py

@@ -68,7 +68,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/threads/types/__init__.py

@@ -66,7 +66,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/types/__init__.py

@@ -56,7 +56,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/webhooks/__init__.py

@@ -48,7 +48,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/webhooks/events/__init__.py

@@ -26,7 +26,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/webhooks/events/types/__init__.py

@@ -32,7 +32,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/webhooks/types/__init__.py

@@ -40,7 +40,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)
 

src/agentmail/websockets/__init__.py

@@ -26,7 +26,7 @@ def __getattr__(attr_name: str) -> typing.Any:
         raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
 
 
-def __dir__():
+def __dir__() -> list[str]:
     lazy_attrs = list(_dynamic_imports.keys())
     return sorted(lazy_attrs)