github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-14 15:16:21 +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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats 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, Union,
) )
from pydantic import BaseModel, Field, root_validator from pydantic_v1 import Field, root_validator
from langchain.callbacks.manager import ( from langchain.callbacks.manager import (
AsyncCallbackManagerForLLMRun, AsyncCallbackManagerForLLMRun,
@ -186,7 +186,7 @@ def _convert_message_to_dict(message: BaseMessage) -> dict:
return message_dict return message_dict
class ChatLiteLLM(BaseChatModel, BaseModel): class ChatLiteLLM(BaseChatModel):
"""Wrapper around the LiteLLM Model I/O library. """Wrapper around the LiteLLM Model I/O library.
To use you must have the google.generativeai Python package installed and To use you must have the google.generativeai Python package installed and