update CDK airbyte protocol models to fix master build (#12829)

Author: pedroslopez

airbyte-cdk/python/airbyte_cdk/models/airbyte_protocol.py

@@ -36,11 +36,16 @@ class Config:
     namespace: Optional[str] = Field(None, description="the namespace of this record's stream")
 
 
-class AirbyteStateMessage(BaseModel):
-    class Config:
-        extra = Extra.allow
+class AirbyteStateType(Enum):
+    GLOBAL = "GLOBAL"
+    PER_STREAM = "PER_STREAM"
+
 
-    data: Dict[str, Any] = Field(..., description="the state data")
+class AirbyteStateBlob(BaseModel):
+    pass
+
+    class Config:
+        extra = Extra.forbid
 
 
 class Level(Enum):
@@ -157,6 +162,14 @@ class OAuthConfigSpecification(BaseModel):
     )
 
 
+class AirbyteStreamState(BaseModel):
+    class Config:
+        extra = Extra.forbid
+
+    name: Optional[str] = Field(None, description="Stream name")
+    state: Optional[AirbyteStateBlob] = None
+
+
 class AirbyteTraceMessage(BaseModel):
     class Config:
         extra = Extra.allow
@@ -244,6 +257,16 @@ class Config:
     )
 
 
+class AirbyteStateMessage(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    state_type: Optional[AirbyteStateType] = None
+    data: Optional[Dict[str, Any]] = Field(None, description="(Deprecated) the state data")
+    global_: Optional[AirbyteStateBlob] = Field(None, alias="global")
+    streams: Optional[List[AirbyteStreamState]] = None
+
+
 class AirbyteCatalog(BaseModel):
     class Config:
         extra = Extra.allow