types

PostHog · Mar 26, 2024 · 3307e5f · 3307e5f
1 parent 08f27b2
commit 3307e5f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/posthog/hogql/ast.py b/posthog/hogql/ast.py
@@ -309,7 +309,7 @@ class AsteriskType(Type):
 
 @dataclass(kw_only=True)
 class FieldTraverserType(Type):
-    chain: List[str | int] | List[str]
+    chain: List[str | int]
     table_type: TableOrSelectType
 
 

diff --git a/posthog/hogql/database/models.py b/posthog/hogql/database/models.py
@@ -65,15 +65,15 @@ class ExpressionField(DatabaseField):
 class FieldTraverser(FieldOrTable):
     model_config = ConfigDict(extra="forbid")
 
-    chain: List[str]
+    chain: List[str | int]
 
 
 class Table(FieldOrTable):
     fields: Dict[str, FieldOrTable]
     model_config = ConfigDict(extra="forbid")
 
-    def has_field(self, name: str) -> bool:
-        return name in self.fields
+    def has_field(self, name: str | int) -> bool:
+        return str(name) in self.fields
 
     def get_field(self, name: str) -> FieldOrTable:
         if self.has_field(name):