Fix connectors base

Author: tuliren

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

@@ -20,6 +20,7 @@ class Type(Enum):
     SPEC = "SPEC"
     CONNECTION_STATUS = "CONNECTION_STATUS"
     CATALOG = "CATALOG"
+    TRACE = "TRACE"
 
 
 class AirbyteRecordMessage(BaseModel):
@@ -59,6 +60,25 @@ class Config:
     message: str = Field(..., description="the log message")
 
 
+class Type1(Enum):
+    ERROR = "ERROR"
+
+
+class FailureType(Enum):
+    system_error = "system_error"
+    config_error = "config_error"
+
+
+class AirbyteErrorTraceMessage(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    message: str = Field(..., description="A user-friendly message that indicates the cause of the error")
+    internal_message: Optional[str] = Field(None, description="The internal error that caused the failure")
+    stack_trace: Optional[str] = Field(None, description="The full stack trace of the error")
+    failure_type: Optional[FailureType] = Field(None, description="The type of error")
+
+
 class Status(Enum):
     SUCCEEDED = "SUCCEEDED"
     FAILED = "FAILED"
@@ -137,6 +157,15 @@ class OAuthConfigSpecification(BaseModel):
     )
 
 
+class AirbyteTraceMessage(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    type: Type1 = Field(..., description="the type of trace message")
+    emitted_at: float = Field(..., description="the time in ms that the message was emitted")
+    error: Optional[AirbyteErrorTraceMessage] = Field(None, description="error trace message: the error object")
+
+
 class AirbyteStream(BaseModel):
     class Config:
         extra = Extra.allow
@@ -246,6 +275,10 @@ class Config:
         None,
         description="schema message: the state. Must be the last message produced. The platform uses this information",
     )
+    trace: Optional[AirbyteTraceMessage] = Field(
+        None,
+        description="trace message: a message to communicate information about the status and performance of a connector",
+    )
 
 
 class AirbyteProtocol(BaseModel):

airbyte-integrations/bases/airbyte-protocol/airbyte_protocol/models/airbyte_protocol.py

@@ -20,6 +20,7 @@ class Type(Enum):
     SPEC = "SPEC"
     CONNECTION_STATUS = "CONNECTION_STATUS"
     CATALOG = "CATALOG"
+    TRACE = "TRACE"
 
 
 class AirbyteRecordMessage(BaseModel):
@@ -56,6 +57,25 @@ class Config:
     message: str = Field(..., description="the log message")
 
 
+class Type1(Enum):
+    ERROR = "ERROR"
+
+
+class FailureType(Enum):
+    system_error = "system_error"
+    config_error = "config_error"
+
+
+class AirbyteErrorTraceMessage(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    message: str = Field(..., description="A user-friendly message that indicates the cause of the error")
+    internal_message: Optional[str] = Field(None, description="The internal error that caused the failure")
+    stack_trace: Optional[str] = Field(None, description="The full stack trace of the error")
+    failure_type: Optional[FailureType] = Field(None, description="The type of error")
+
+
 class Status(Enum):
     SUCCEEDED = "SUCCEEDED"
     FAILED = "FAILED"
@@ -133,6 +153,15 @@ class OAuthConfigSpecification(BaseModel):
     )
 
 
+class AirbyteTraceMessage(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    type: Type1 = Field(..., description="the type of trace message")
+    emitted_at: float = Field(..., description="the time in ms that the message was emitted")
+    error: Optional[AirbyteErrorTraceMessage] = Field(None, description="error trace message: the error object")
+
+
 class AirbyteStream(BaseModel):
     class Config:
         extra = Extra.allow
@@ -231,6 +260,9 @@ class Config:
     state: Optional[AirbyteStateMessage] = Field(
         None, description="schema message: the state. Must be the last message produced. The platform uses this information"
     )
+    trace: Optional[AirbyteTraceMessage] = Field(
+        None, description="trace message: a message to communicate information about the status and performance of a connector"
+    )
 
 
 class AirbyteProtocol(BaseModel):