Docs() generates exception due to langchain AIMessage() not supplying __len__() when I specify a langchain llm

[maspotts]

Docs() generates exception due to langchain AIMessage() not supplying __len__() when I specify a langchain llm

[maspotts]

I guess the root issue is that LLMModel.count_tokens() expects text to be a string, whereas it is (now, lately?) a langchain_core.messages.AIMessage, which contains a string (or union of dicts of strings). So the best approach might be to rewrite LLMModel to expect a string OR AIMessage and process it accordingly? I tried monkey-patching LLMModel to try this out but I got sucked into a maze of imports and methods that receive AIMessage instances (another is in Docs.aadd() where len(citation) similarly throws an exception).