github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-06 21:43:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

lite llm lint (#9208)

Browse Source
This commit is contained in:
Bagatur
2023-08-14 08:03:06 -07:00
committed by GitHub
parent 49f1d8477c
commit b5a57acf6c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libs/langchain/langchain/chat_models/litellm.py
View File

@@ -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