fixing typing issues (#316)

Author: alexander-belikov

.cursorignore

@@ -38,6 +38,6 @@
 !*.md
 !.pre-commit-config.yaml
 !run-tests.sh
-
+!precommit.log
 
 

graflo/architecture/graph_types/context.py

@@ -103,7 +103,7 @@ class ExtractionContext(ConfigBaseModel):
         edge_intents: Explicit edge intents for assembly phase
     """
 
-    model_config = ConfigDict(kw_only=True)  # type: ignore[assignment]
+    model_config = ConfigDict(kw_only=True)  # ty: ignore[invalid-key]
 
     # Pydantic cannot schema nested defaultdict with custom key types (e.g. LocationIndex),
     # so we use Any; runtime type is as documented in Attributes
@@ -185,7 +185,7 @@ def record_transform_failure(
 class AssemblyContext(ConfigBaseModel):
     """Assembly-phase context built from extraction outputs."""
 
-    model_config = ConfigDict(kw_only=True)  # type: ignore[assignment]
+    model_config = ConfigDict(kw_only=True)  # ty: ignore[invalid-key]
 
     extraction: ExtractionContext
     acc_global: Any = Field(default_factory=dd_factory)

graflo/architecture/graph_types/transform.py

@@ -16,7 +16,7 @@ class VertexRep(ConfigBaseModel):
         vertex: doc representing a vertex
     """
 
-    model_config = ConfigDict(kw_only=True)  # type: ignore[assignment]
+    model_config = ConfigDict(kw_only=True)  # ty: ignore[invalid-key]
 
     vertex: dict[str, Any]
 

graflo/architecture/pipeline/runtime/actor/base.py

@@ -79,7 +79,7 @@ def _filter_items(self, items: dict[str, object]) -> dict[str, object]:
     def _stringify_items(self, items: dict[str, object]) -> dict[str, str]:
         """Convert items to string representation."""
         return {
-            k: ", ".join(list(v)) if isinstance(v, (tuple, list)) else str(v)
+            k: ", ".join(str(x) for x in v) if isinstance(v, (tuple, list)) else str(v)
             for k, v in items.items()
         }
 

graflo/architecture/pipeline/runtime/actor/vertex_router.py

@@ -92,11 +92,14 @@ def _get_or_create_wrapper(self, vertex_type: str) -> "ActorWrapper | None":
             per_type_from = self.vertex_from_map[vertex_type]
         else:
             per_type_from = self.from_doc
-        config = VertexActorConfig(
-            vertex=vertex_type,
-            from_doc=per_type_from,
-            keep_fields=list(self.keep_fields) if self.keep_fields else None,
-            extraction_scope=self.extraction_scope,
+        config = VertexActorConfig.model_validate(
+            {
+                "type": "vertex",
+                "vertex": vertex_type,
+                "from": per_type_from,
+                "keep_fields": list(self.keep_fields) if self.keep_fields else None,
+                "extraction_scope": self.extraction_scope,
+            }
         )
         wrapper = ActorWrapper.from_config(config)
         wrapper.finish_init(self._init_ctx)

graflo/data_source/api.py

@@ -44,7 +44,7 @@ class APIConfig(ConfigBaseModel):
     method: str = "GET"
     headers: dict[str, str] = Field(default_factory=dict)
     auth: ApiAuth | None = None
-    params: dict[str, object] = Field(default_factory=dict)
+    params: dict[str, Any] = Field(default_factory=dict)
     timeout: float | None = None
     retries: int = 0
     retry_backoff_factor: float = 0.1

graflo/db/falkordb/conn.py

@@ -832,15 +832,15 @@ def fetch_edges(
             target_match = "(target)"
 
         # Build WHERE clauses
-        where_clauses = []
+        where_clauses: list[str] = []
         if to_id:
             where_clauses.append(f"target.id = '{to_id}'")
 
         # Add additional filters if provided
         if filters is not None:
             ff = FilterExpression.from_dict(filters)
             filter_clause = ff(doc_name="r", kind=self.expression_flavor())
-            where_clauses.append(filter_clause)
+            where_clauses.append(str(filter_clause))
 
         where_clause = f"WHERE {' AND '.join(where_clauses)}" if where_clauses else ""
 

graflo/db/memgraph/conn.py

@@ -81,7 +81,7 @@
 from typing import Any
 from urllib.parse import urlparse
 
-import mgclient  # type: ignore[import-untyped]
+import mgclient  # ty: ignore[unresolved-import]
 
 from graflo.architecture.schema.edge import Edge
 from graflo.architecture.graph_types import Index
@@ -1026,15 +1026,15 @@ def fetch_edges(
         q += f" {rel_pattern} {target_match}"
 
         # Build WHERE clauses
-        where_clauses = []
+        where_clauses: list[str] = []
         if to_id:
             where_clauses.append("t.id = $to_id")
 
         # Add relationship property filters
         if filters is not None:
             ff = FilterExpression.from_dict(filters)
             filter_str = ff(doc_name="r", kind=self.expression_flavor())
-            where_clauses.append(filter_str)
+            where_clauses.append(str(filter_str))
 
         if where_clauses:
             q += f" WHERE {' AND '.join(where_clauses)}"

graflo/db/neo4j/conn.py

@@ -721,15 +721,15 @@ def fetch_edges(
             target_match = "(target)"
 
         # Add target ID filter if provided
-        where_clauses = []
+        where_clauses: list[str] = []
         if to_id:
             where_clauses.append(f"target.id = '{to_id}'")
 
         # Add additional filters if provided
         if filters is not None:
             ff = FilterExpression.from_dict(filters)
             filter_clause = ff(doc_name="r", kind=self.expression_flavor())
-            where_clauses.append(filter_clause)
+            where_clauses.append(str(filter_clause))
 
         where_clause = f"WHERE {' AND '.join(where_clauses)}" if where_clauses else ""
 

graflo/db/tigergraph/auth.py

@@ -227,21 +227,22 @@ def _get_token_from_secret(
                     raise ValueError(f"No token in response: {result}")
 
             except requests.exceptions.HTTPError as e:
+                response = e.response
+                status_code = response.status_code if response is not None else None
                 # Track if this was a 404 error
-                if e.response.status_code != 404:
+                if status_code != 404:
                     all_404_errors = False
 
                 # If 404 and we have more endpoints to try, continue
-                if e.response.status_code == 404 and len(endpoints_to_try) > 1:
+                if status_code == 404 and len(endpoints_to_try) > 1:
                     logger.debug(
                         f"Endpoint {url} returned 404, trying next endpoint..."
                     )
                     last_error = e
                     continue
                 # For other HTTP errors, log and try next endpoint if available
-                logger.debug(
-                    f"HTTP error {e.response.status_code} on {url}: {e.response.text}"
-                )
+                response_text = response.text if response is not None else ""
+                logger.debug(f"HTTP error {status_code} on {url}: {response_text}")
                 last_error = e
                 continue
             except Exception as e: