langchain.ChatOpenAIDataset

kedro_datasets_experimental.langchain.ChatOpenAIDataset

ChatOpenAIDataset(credentials, kwargs=None)

Bases: OpenAIDataset[ChatOpenAI]

ChatOpenAIDataset loads a ChatOpenAI langchain model.

Example usage for the YAML API

catalog.yml

gpt_3_5_turbo:
    type: langchain.ChatOpenAIDataset
    kwargs:
        model: "gpt-3.5-turbo"
        temperature: 0.0
    credentials: openai

credentials.yml

openai:
    openai_api_base: <openai-api-base>
    openai_api_key: <openai-api-key>

Example usage for the Python API

from kedro_datasets_experimental.langchain import ChatOpenAIDataset

llm = ChatOpenAIDataset(
    credentials={
        "openai_api_base": "<openai-api-base>",
        "openai_api_key": "<openai-api-key>",
    },
    kwargs={
        "model": "gpt-3.5-turbo",
        "temperature": 0.0,
    },
).load()

# See: https://python.langchain.com/docs/integrations/chat/openai
llm.invoke("Hello world!")

Source code in kedro-datasets/kedro_datasets_experimental/langchain/_openai.py

def __init__(self, credentials: dict[str, str], kwargs: dict[str, Any] = None):
    """Constructor.

    Args:
        credentials: must contain `openai_api_base` and `openai_api_key`.
        kwargs: keyword arguments passed to the underlying constructor.
    """
    self.openai_api_base = credentials["openai_api_base"]
    self.openai_api_key = credentials["openai_api_key"]
    self.kwargs = kwargs or {}

constructor property

constructor