Skip to content

Commit 60520dc

Browse files
fix the llm organization parameter is ineffective during queries (#612)
* fix the organization parameter is ineffective during queries * add semver impact document --------- Co-authored-by: Alonso Guevara <alonsog@microsoft.com>
1 parent 2a95e67 commit 60520dc

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed
Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
{
2+
"type": "patch",
3+
"description": "fix the organization parameter is ineffective during queries"
4+
}

graphrag/query/factories.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@ def get_llm(config: GraphRagConfig) -> ChatOpenAI:
5858
else None
5959
),
6060
api_base=config.llm.api_base,
61+
organization=config.llm.organization,
6162
model=config.llm.model,
6263
api_type=OpenaiApiType.AzureOpenAI if is_azure_client else OpenaiApiType.OpenAI,
6364
deployment_name=config.llm.deployment_name,
@@ -89,6 +90,7 @@ def get_text_embedder(config: GraphRagConfig) -> OpenAIEmbedding:
8990
else None
9091
),
9192
api_base=config.embeddings.llm.api_base,
93+
organization=config.llm.organization,
9294
api_type=OpenaiApiType.AzureOpenAI if is_azure_client else OpenaiApiType.OpenAI,
9395
model=config.embeddings.llm.model,
9496
deployment_name=config.embeddings.llm.deployment_name,

0 commit comments

Comments
 (0)