Skip to content

Commit d1230f5

Browse files
author
Elliot Huang
committed
openai: support override base_url
1 parent 3583147 commit d1230f5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

fast_graphrag/_llm/_llm_openai.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ class OpenAILLMService(BaseLLMService):
3131

3232
def __post_init__(self):
3333
logger.debug("Initialized OpenAILLMService with patched OpenAI client.")
34-
self.llm_async_client: instructor.AsyncInstructor = instructor.from_openai(AsyncOpenAI(api_key=self.api_key))
34+
self.llm_async_client: instructor.AsyncInstructor = instructor.from_openai(AsyncOpenAI(api_key=self.api_key, base_url=self.base_url))
3535

3636
@retry(
3737
stop=stop_after_attempt(3),
@@ -112,7 +112,7 @@ class OpenAIEmbeddingService(BaseEmbeddingService):
112112
model: Optional[str] = field(default="text-embedding-3-small")
113113

114114
def __post_init__(self):
115-
self.embedding_async_client: AsyncOpenAI = AsyncOpenAI(api_key=self.api_key)
115+
self.embedding_async_client: AsyncOpenAI = AsyncOpenAI(api_key=self.api_key, base_url=self.base_url)
116116
logger.debug("Initialized OpenAIEmbeddingService with OpenAI client.")
117117

118118
@retry(

0 commit comments

Comments
 (0)