# Create the cohere rag retriever using the chat model
rag = CohereRagRetriever(llm=llm)
docs = rag.invoke(
    user_query,
    documents=input_docs,
)