Update base.py.

Author: qiuosier

ads/llm/langchain/plugins/base.py

@@ -6,14 +6,13 @@
 from typing import Any, Dict, List, Optional
 
 from langchain.llms.base import LLM
-from langchain.load.serializable import Serializable
 from langchain.pydantic_v1 import BaseModel, Field, root_validator
 
 from ads.common.auth import default_signer
 from ads.config import COMPARTMENT_OCID
 
 
-class BaseLLM(LLM, Serializable):
+class BaseLLM(LLM):
     """Base OCI LLM class. Contains common attributes."""
 
     auth: dict = Field(default_factory=default_signer, exclude=True)
@@ -58,6 +57,14 @@ def is_lc_serializable(cls) -> bool:
 class GenerativeAiClientModel(BaseModel):
     """Base model for generative AI embedding model and LLM."""
 
+    # This auth is the same as the auth in BaseLLM class.
+    # However, this is needed for the Gen AI embedding model.
+    # Do not remove this attribute
+    auth: dict = Field(default_factory=default_signer, exclude=True)
+    """ADS auth dictionary for OCI authentication.
+    This can be generated by calling `ads.common.auth.api_keys()` or `ads.common.auth.resource_principal()`.
+    If this is not provided then the `ads.common.default_signer()` will be used."""
+
     client: Any  #: :meta private:
     """OCI GenerativeAiClient."""