Provide openai-key support from config (#1052)

This commit is contained in:
Sidharth Mohanty
2023-12-23 14:42:18 +05:30
committed by GitHub
parent e90673ae5b
commit 11f0d719f5
9 changed files with 31 additions and 43 deletions

View File

@@ -1,4 +1,5 @@
import json
import os
from typing import Any, Dict, Optional
from langchain.chat_models import ChatOpenAI
@@ -30,6 +31,7 @@ class OpenAILlm(BaseLlm):
"max_tokens": config.max_tokens,
"model_kwargs": {},
}
api_key = config.api_key or os.environ["OPENAI_API_KEY"]
if config.top_p:
kwargs["model_kwargs"]["top_p"] = config.top_p
if config.stream:
@@ -37,9 +39,9 @@ class OpenAILlm(BaseLlm):
StreamingStdOutCallbackHandler
callbacks = config.callbacks if config.callbacks else [StreamingStdOutCallbackHandler()]
chat = ChatOpenAI(**kwargs, streaming=config.stream, callbacks=callbacks)
chat = ChatOpenAI(**kwargs, streaming=config.stream, callbacks=callbacks, api_key=api_key)
else:
chat = ChatOpenAI(**kwargs)
chat = ChatOpenAI(**kwargs, api_key=api_key)
if self.functions is not None:
from langchain.chains.openai_functions import \
create_openai_fn_runnable