yujunjun/graphrag
1
0
Fork 0
mirror of https://github.com/microsoft/graphrag.git synced 2025-12-12 15:31:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix: Add await to LLM execution for async handling (#1206)

Browse Source 
Co-authored-by: Alonso Guevara <alonsog@microsoft.com>
This commit is contained in:
9prodhi 2024-10-09 18:26:28 -05:00 committed by GitHub
parent cd4f1fa9ba
commit ce8749bd19
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
graphrag/llm/openai/openai_completion_llm.py
View File

@ -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