Skip to content

Type the state parameter in memory_node.py #806

@mjnovice

Description

@mjnovice

Context

In src/uipath_langchain/agent/react/memory_node.py, the memory_recall_node function accepts state: Any. The reviewer asked if we could type this properly.

From PR #793 review (r3129672333): "have we tried to type the state?"

Details

The state is a dynamically created Pydantic model (CompleteAgentGraphState = AgentGraphState + input_schema) which makes static typing difficult. Initial attempt was tedious due to the dynamic nature of the combined state class.

Action

Revisit typing the state parameter in memory_recall_node and other graph nodes that currently use Any.

Ref: #793 (comment)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions