neo4j-graphrag-python icon indicating copy to clipboard operation
neo4j-graphrag-python copied to clipboard

No async_client for VertexAILLM

Open xpilasneo4j opened this issue 4 months ago • 1 comments

Writing some code for various LLMs, it works with AzureOpenAI and OpenAI but not with VertexAI Here is my code

    res_pipeline = await define_and_run_pipeline(llm, file_path)
    await llm.async_client.close()
    return res_pipeline

and here is the error

 File "C:\Users\XavierPilas\Documents\GitHub\rag-graphrag\GenericGraphRag\loadDataFromPDF.py", line 167, in main
    await llm.async_client.close()
          ^^^^^^^^^^^^^^^^
AttributeError: 'VertexAILLM' object has no attribute 'async_client'

xpilasneo4j avatar Aug 19 '25 06:08 xpilasneo4j

Hi Xavier,

Yes, this line is a hack because we're not managing client closure properly internally for now. If your LLM doesn't have a client, you can ignore this line.

stellasia avatar Aug 25 '25 09:08 stellasia