Skip to content

Commit 23c7fd9

Browse files
Adding UnknownTypeExpr
1 parent 7b6a90b commit 23c7fd9

File tree

1 file changed

+17
-1
lines changed

1 file changed

+17
-1
lines changed

src/replit_river/codegen/typing.py

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,18 @@ class UnionTypeExpr:
3030
nested: list["TypeExpression"]
3131

3232

33+
@dataclass
34+
class UnknownTypeExpr:
35+
name: TypeName
36+
37+
3338
TypeExpression = (
34-
TypeName | DictTypeExpr | ListTypeExpr | LiteralTypeExpr | UnionTypeExpr
39+
TypeName
40+
| DictTypeExpr
41+
| ListTypeExpr
42+
| LiteralTypeExpr
43+
| UnionTypeExpr
44+
| UnknownTypeExpr
3545
)
3646

3747

@@ -47,6 +57,8 @@ def render_type_expr(value: TypeExpression) -> str:
4757
return " | ".join(render_type_expr(x) for x in inner)
4858
case str(name):
4959
return TypeName(name)
60+
case UnknownTypeExpr(name):
61+
return TypeName(name)
5062
case other:
5163
assert_never(other)
5264

@@ -65,6 +77,8 @@ def extract_inner_type(value: TypeExpression) -> TypeName:
6577
)
6678
case str(name):
6779
return TypeName(name)
80+
case UnknownTypeExpr(name):
81+
return name
6882
case other:
6983
assert_never(other)
7084

@@ -89,5 +103,7 @@ def ensure_literal_type(value: TypeExpression) -> TypeName:
89103
)
90104
case str(name):
91105
return TypeName(name)
106+
case UnknownTypeExpr(name):
107+
return name
92108
case other:
93109
assert_never(other)

0 commit comments

Comments
 (0)