Scrapegraph-ai/scrapegraphai/models/oneapi.py
Marco Vinciguerra b6f1766bd1 add OneAPI integration
Co-Authored-By: Federico Aguzzi <62149513+f-aguzzi@users.noreply.github.com>
Co-Authored-By: wangdongpeng1 <74647183+wangdongpeng1@users.noreply.github.com>
2024-05-24 13:46:40 +02:00

18 lines
408 B
Python

"""
OpenAI Module
"""
from langchain_openai import ChatOpenAI
class OneApi(ChatOpenAI):
"""
A wrapper for the OneApi 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)