bug: LLMRails has not attribute kb when trying to use Nemo Guardrails with Langgraph

Did you check docs and existing issues?

• I have read all the NeMo-Guardrails docs
• I have updated the package to the latest version before submitting this issue
• (optional) I have used the develop branch
• I have searched the existing issues of NeMo-Guardrails

Python version (python --version)

Python 3.12.3

Operating system/version

Linux

NeMo-Guardrails version (if you must use a specific version and not the latest

0.16.0 onwards

Describe the bug

AttributeError: 'LLMRails' object has no attribute 'kb'

config = RailsConfig.from_path(guardrail_config_path)
guardrails = RunnableRails(config=config, passthrough=True, verbose=False)
model_with_rails = guardrails | model
response = model_with_rails.invoke([HumanMessage(content=prompt)])

official langchain integration also returning same error

Steps To Reproduce

config = RailsConfig.from_path(guardrail_config_path)
guardrails = RunnableRails(config=config, passthrough=True, verbose=False)
model_with_rails = guardrails | model
response = model_with_rails.invoke([HumanMessage(content=prompt)])

Expected Behavior

It should return a response

Actual Behavior

AttributeError: 'LLMRails' object has no attribute 'kb'