Scrapegraph-ai/scrapegraphai/models/openai.py
2024-05-01 23:33:42 +02:00

18 lines
412 B
Python

"""
OpenAI Module
"""
from langchain_openai import ChatOpenAI
class OpenAI(ChatOpenAI):
"""
A wrapper for the ChatOpenAI class that provides default configuration
and could be extended with additional methods if needed.
Args:
llm_config (dict): Configuration parameters for the language model.
"""
def __init__(self, llm_config: dict):
super().__init__(**llm_config)