diff --git a/graphrag/llm/openai/openai_completion_llm.py b/graphrag/llm/openai/openai_completion_llm.py index bdbac6c1..74511c02 100644 --- a/graphrag/llm/openai/openai_completion_llm.py +++ b/graphrag/llm/openai/openai_completion_llm.py @@ -39,5 +39,5 @@ class OpenAICompletionLLM(BaseLLM[CompletionInput, CompletionOutput]): args = get_completion_llm_args( kwargs.get("model_parameters"), self.configuration ) - completion = self.client.completions.create(prompt=input, **args) + completion = await self.client.completions.create(prompt=input, **args) return completion.choices[0].text