mirror of
https://github.com/VinciGit00/Scrapegraph-ai.git
synced 2026-07-01 21:00:48 +08:00
17 lines
430 B
Python
17 lines
430 B
Python
"""
|
|
VertexAI Module
|
|
"""
|
|
from langchain_google_vertexai import ChatVertexAI
|
|
|
|
class VertexAI(ChatVertexAI):
|
|
"""
|
|
A wrapper for the ChatVertexAI 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)
|