Fix #114

Comment From: tzolov

Thanks for the contribution @Omkar-Shetkar . I will review it shortly

Comment From: Omkar-Shetkar

@tzolov @markpollack Thank you. Happy to contribute to Spring AI.

Comment From: tzolov

Rebased, squashed and merged at: 2209bfff9620c19e2fe557a9cd7be5c9bb9a1a86