chain = {

    'search_results': lambda x: get_search_results(x),

    'query': lambda x: x,

} | prompt | model

chain.invoke(query="?")