Skip to content

Commit 1e07a4c

Browse files
These are part of the codegen serdes that point at the v1 client, they are unchanged
1 parent b0f4553 commit 1e07a4c

File tree

1 file changed

+1
-39
lines changed

1 file changed

+1
-39
lines changed

src/replit_river/v2/client.py

Lines changed: 1 addition & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -7,12 +7,8 @@
77

88
from opentelemetry import trace
99
from opentelemetry.trace import Span, SpanKind, Status, StatusCode
10-
from pydantic import (
11-
BaseModel,
12-
ValidationInfo,
13-
)
1410

15-
from replit_river.error_schema import ERROR_CODE_UNKNOWN, RiverError, RiverException
11+
from replit_river.error_schema import RiverError, RiverException
1612
from replit_river.transport_options import (
1713
HandshakeMetadataType,
1814
TransportOptions,
@@ -24,40 +20,6 @@
2420
tracer = trace.get_tracer(__name__)
2521

2622

27-
@dataclass(frozen=True)
28-
class RiverUnknownValue(BaseModel):
29-
tag: Literal["RiverUnknownValue"]
30-
value: Any
31-
32-
33-
class RiverUnknownError(RiverError):
34-
pass
35-
36-
37-
def translate_unknown_value(
38-
value: Any, handler: Callable[[Any], Any], info: ValidationInfo
39-
) -> Any | RiverUnknownValue:
40-
try:
41-
return handler(value)
42-
except Exception:
43-
return RiverUnknownValue(tag="RiverUnknownValue", value=value)
44-
45-
46-
def translate_unknown_error(
47-
value: Any, handler: Callable[[Any], Any], info: ValidationInfo
48-
) -> Any | RiverUnknownError:
49-
try:
50-
return handler(value)
51-
except Exception:
52-
if isinstance(value, dict) and "code" in value and "message" in value:
53-
return RiverUnknownError(
54-
code=value["code"],
55-
message=value["message"],
56-
)
57-
else:
58-
return RiverUnknownError(code=ERROR_CODE_UNKNOWN, message="Unknown error")
59-
60-
6123
class Client(Generic[HandshakeMetadataType]):
6224
def __init__(
6325
self,

0 commit comments

Comments
 (0)