mirror of
https://github.com/hwchase17/langchain.git
synced 2026-03-18 02:53:16 +00:00
26 lines
682 B
Python
26 lines
682 B
Python
import os
|
|
from typing import Any
|
|
|
|
from langchain_core.utils import convert_to_secret_str
|
|
from perplexity import Perplexity
|
|
|
|
|
|
def initialize_client(values: dict[str, Any]) -> dict[str, Any]:
|
|
"""Initialize the Perplexity client."""
|
|
pplx_api_key = (
|
|
values.get("pplx_api_key")
|
|
or os.environ.get("PPLX_API_KEY")
|
|
or os.environ.get("PERPLEXITY_API_KEY")
|
|
or ""
|
|
)
|
|
values["pplx_api_key"] = convert_to_secret_str(pplx_api_key)
|
|
|
|
api_key = (
|
|
values["pplx_api_key"].get_secret_value() if values["pplx_api_key"] else None
|
|
)
|
|
|
|
if not values.get("client"):
|
|
values["client"] = Perplexity(api_key=api_key)
|
|
|
|
return values
|