fix: use SecretStr get_secret_value for LMStudio API key handling (#5200)

* 🐛 (lmstudiomodel.py): fix issue with accessing secret value in api_key variable to prevent potential errors

* ♻️ (lmstudiomodel.py): refactor api_key assignment to simplify code and improve readability

* [autofix.ci] apply automated fixes

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
Cristhian Zanforlin Lousa 2024-12-12 09:08:04 -03:00 committed by GitHub
commit 262e3bd3cc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -3,7 +3,6 @@ from urllib.parse import urljoin
import httpx
from langchain_openai import ChatOpenAI
from pydantic.v1 import SecretStr
from typing_extensions import override
from langflow.base.models.model import LCModelComponent
@ -103,14 +102,12 @@ class LMStudioModelComponent(LCModelComponent):
base_url = self.base_url or "http://localhost:1234/v1"
seed = self.seed
api_key = SecretStr(lmstudio_api_key) if lmstudio_api_key else None
return ChatOpenAI(
max_tokens=max_tokens or None,
model_kwargs=model_kwargs,
model=model_name,
base_url=base_url,
api_key=api_key,
api_key=lmstudio_api_key,
temperature=temperature if temperature is not None else 0.1,
seed=seed,
)