# Create the Cohere rag retriever using the chat model with the web search connector
rag = CohereRagRetriever(llm=llm, connectors=[{"id": "web-search"}])
docs = rag.invoke("Who founded Cohere?")