From b5a57acf6c7d40cf823442a2656f573d1a8ec802 Mon Sep 17 00:00:00 2001
From: Bagatur <22008038+baskaryan@users.noreply.github.com>
Date: Mon, 14 Aug 2023 08:03:06 -0700
Subject: [PATCH] lite llm lint (#9208)

---
 libs/langchain/langchain/chat_models/litellm.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/libs/langchain/langchain/chat_models/litellm.py b/libs/langchain/langchain/chat_models/litellm.py
index 165e1d1a6e..5dcaa8653f 100644
--- a/libs/langchain/langchain/chat_models/litellm.py
+++ b/libs/langchain/langchain/chat_models/litellm.py
@@ -15,7 +15,7 @@ from typing import (
     Union,
 )
 
-from pydantic import BaseModel, Field, root_validator
+from pydantic_v1 import Field, root_validator
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForLLMRun,
@@ -186,7 +186,7 @@ def _convert_message_to_dict(message: BaseMessage) -> dict:
     return message_dict
 
 
-class ChatLiteLLM(BaseChatModel, BaseModel):
+class ChatLiteLLM(BaseChatModel):
     """Wrapper around the LiteLLM Model I/O library.
 
     To use you must have the google.generativeai Python package installed and