Kendra queries are failing with langchain update

[supreetkt]

After a recent upgrade to langchain-aws (from v0.1.12 to v0.2.0), I'm noticing my kendra retrieval fail due to pydantic errors:

Validation errors for RetrieveResult\nResultItems.0.DocumentAttributes.0.Value.DateValue
Field required [type=missing, input_value={'StringValue': 'http://d...NotificationHowTo.html'}, input_type=dict]
For further information visit https://errors.pydantic.dev/2.9/v/missing
ResultItems.0.DocumentAttributes.0.Value.LongValue\n  Field required [type=missing, input_value={'StringValue': 'http://d...NotificationHowTo.html'}, input_type=dict]
For further information visit https://errors.pydantic.dev/2.9/v/missing\nResultItems.0.DocumentAttributes.0.Value.StringListValue\n  Field required [type=missing, input_value={'StringValue': 'http://d...NotificationHowTo.html'}, input_type=dict]
For further information visit https://errors.pydantic.dev/2.9/v/missing\nResultItems.1.DocumentAttributes.0.Value.DateValue\n  Field required [type=missing, input_value={'StringValue': 'http://d...NotificationHowTo.html'}, input_type=dict]\n    For further information visit https://errors.pydantic.dev/2.9/v/missing

Essentially, the retrieved results from Kendra are not fitting the pydantic schema set out for DocumentAttributeValue:

class DocumentAttributeValue(BaseModel, extra="allow"):  # type: ignore[call-arg]
    """Value of a document attribute."""

    DateValue: Optional[str]
    """The date expressed as an ISO 8601 string."""
    LongValue: Optional[int]
    """The long value."""
    StringListValue: Optional[List[str]]
    """The string list value."""
    StringValue: Optional[str]
    """The string value."""

Setting DateValue, LongValue, StringListValue and StringValue "optional" values are no longer working optionally. A default value of None needs to be set up on these above like below which would fix these issues:

class DocumentAttributeValue(BaseModel, extra="allow"):  # type: ignore[call-arg]
    """Value of a document attribute."""

    DateValue: Optional[str] = None
    """The date expressed as an ISO 8601 string."""
    LongValue: Optional[int] = None
    """The long value."""
    StringListValue: Optional[List[str]] = None
    """The string list value."""
    StringValue: Optional[str] = None
    """The string value."""

I'm assuming this has to do with a pydantic upgrade.

[ccurme]

Thanks @supreetkt for reporting this. Would you mind taking a look at #203 and confirming this will fix your workflow?

[vaibhavqrcg]

I'm still getting this issue on latest release today 0.2.7, does this still needs fixing?

[Spencer10798]

Still an error on latest

[supreetkt]

This was fixed when it was last reported. Maybe open another issue with the new version and error details.