From dfcf4913e9c5c582ea4d2cb97a114c6dd0168873 Mon Sep 17 00:00:00 2001
From: Stanislav Novosad <stas@skyvern.com>
Date: Fri, 19 Dec 2025 13:37:03 -0700
Subject: [PATCH] Stricter mypy rules for skyvern.library.* (#4345)

---
 mypy.ini                              | 10 ++++++++++
 skyvern/forge/sdk/workflow/service.py |  4 ++--
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/mypy.ini b/mypy.ini
index 741177e7..5a621cd4 100644
--- a/mypy.ini
+++ b/mypy.ini
@@ -6,5 +6,15 @@ disallow_untyped_defs = True
 exclude = ^(venv|venv-dev)
 ignore_missing_imports = True
 
+[mypy-skyvern.library.*]
+ignore_missing_imports = False
+follow_imports = normal
+disable_error_code = arg-type, override, return-value, valid-type, no-any-return
+
+; Disabling for generated code
+[mypy-skyvern.client.*]
+follow_imports = normal
+ignore_errors = True
+
 [mypy-numpy.*]
 ignore_errors = True
diff --git a/skyvern/forge/sdk/workflow/service.py b/skyvern/forge/sdk/workflow/service.py
index 2d60ec68..e2ec9ef3 100644
--- a/skyvern/forge/sdk/workflow/service.py
+++ b/skyvern/forge/sdk/workflow/service.py
@@ -1693,9 +1693,9 @@ class WorkflowService:
                 continue
 
             block_output_parameter["created_at"] = output_parameter.created_at
-            labels_to_outputs[label] = block_output_parameter
+            labels_to_outputs[label] = block_output_parameter  # type: ignore[assignment]
 
-        return labels_to_outputs
+        return labels_to_outputs  # type: ignore[return-value]
 
     async def get_workflows_by_permanent_ids(
         self,