Update factories.py to allow the usage of the request timeout ChatOpe… (

#1115) Update factories.py to allow the usage of the request timeout ChatOpenAI parameter allow the usage of the request timeout ChatOpenAI parameter Co-authored-by: Alonso Guevara <[email protected]>
microsoft · Sep 12, 2024 · fcfa7b1 · fcfa7b1
1 parent 7b8f5ba
commit fcfa7b1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/graphrag/query/factories.py b/graphrag/query/factories.py
@@ -65,6 +65,7 @@ def get_llm(config: GraphRagConfig) -> ChatOpenAI:
         deployment_name=config.llm.deployment_name,
         api_version=config.llm.api_version,
         max_retries=config.llm.max_retries,
+        request_timeout=config.llm.request_timeout,
     )