Skip to content

Commit

Permalink
chore: remove api_key from OpenAI config
Browse files Browse the repository at this point in the history
  • Loading branch information
maciejmajek committed Dec 4, 2024
1 parent 1d064f0 commit 7039ed4
Showing 1 changed file with 1 addition and 15 deletions.
16 changes: 1 addition & 15 deletions src/rai/rai/utils/model_initialization.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,6 @@
import coloredlogs
import tomli
from langchain_core.callbacks.base import BaseCallbackHandler
from pydantic import SecretStr

logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)
Expand Down Expand Up @@ -54,7 +53,6 @@ class OllamaConfig(ModelConfig):
@dataclass
class OpenAIConfig(ModelConfig):
base_url: str
api_key: str


@dataclass
Expand Down Expand Up @@ -118,20 +116,8 @@ def get_llm_model(
from langchain_openai import ChatOpenAI

model_config = cast(OpenAIConfig, model_config)
api_key = (
model_config.api_key
if model_config.api_key != ""
else os.getenv("OPENAI_API_KEY", None)
)
if api_key is None:
raise ValueError(
"OPENAI_API_KEY is not set. Set it either in config.toml "
"(for openai compatible apis) or as an environment variable."
)

return ChatOpenAI(
model=model, base_url=model_config.base_url, api_key=SecretStr(api_key)
)
return ChatOpenAI(model=model, base_url=model_config.base_url)
elif vendor == "aws":
from langchain_aws import ChatBedrock

Expand Down

0 comments on commit 7039ed4

Please sign in to comment.