mirror of
https://github.com/VinciGit00/Scrapegraph-ai.git
synced 2026-06-25 21:11:11 +08:00
21 lines
606 B
Python
21 lines
606 B
Python
"""
|
|
Gemini Module
|
|
"""
|
|
from langchain_google_genai import ChatGoogleGenerativeAI
|
|
|
|
|
|
class Gemini(ChatGoogleGenerativeAI):
|
|
"""
|
|
A wrapper for the Gemini class that provides default configuration
|
|
and could be extended with additional methods if needed.
|
|
|
|
Args:
|
|
llm_config (dict): Configuration parameters for the language model
|
|
(e.g., model="gemini-pro")
|
|
"""
|
|
|
|
def __init__(self, llm_config: dict):
|
|
# replace "api_key" to "google_api_key"
|
|
llm_config["google_api_key"] = llm_config.pop("api_key", None)
|
|
super().__init__(**llm_config)
|