Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-04 06:00:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

add additional model support to chatlitellm (#10134)

Browse Source 
---------

Co-authored-by: Harrison Chase <hw.chase.17@gmail.com>
This commit is contained in:
Krish Dholakia 2023-09-04 00:16:40 -07:00 committed by GitHub
parent de3322609e
commit 31bbe80758
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -188,20 +188,6 @@ def _convert_message_to_dict(message: BaseMessage) -> dict:
class ChatLiteLLM(BaseChatModel): class ChatLiteLLM(BaseChatModel):
"""`LiteLLM` Chat models API.
1. The ``GOOGLE_API_KEY``` environment variable set with your API key, or
2. Pass your API key using the google_api_key kwarg to the ChatGoogle
constructor.
Example:
.. code-block:: python
from langchain.chat_models import ChatGooglePalm
chat = ChatGooglePalm()
"""
client: Any #: :meta private: client: Any #: :meta private:
model: str = "gpt-3.5-turbo" model: str = "gpt-3.5-turbo"
model_name: Optional[str] = None model_name: Optional[str] = None
@ -302,6 +288,15 @@ class ChatLiteLLM(BaseChatModel):
values["openrouter_api_key"] = get_from_dict_or_env( values["openrouter_api_key"] = get_from_dict_or_env(
values, "openrouter_api_key", "OPENROUTER_API_KEY", default="" values, "openrouter_api_key", "OPENROUTER_API_KEY", default=""
) )
values["cohere_api_key"] = get_from_dict_or_env(
values, "cohere_api_key", "COHERE_API_KEY", default=""
)
values["huggingface_api_key"] = get_from_dict_or_env(
values, "huggingface_api_key", "HUGGINGFACE_API_KEY", default=""
)
values["together_ai_api_key"] = get_from_dict_or_env(
values, "together_ai_api_key", "TOGETHERAI_API_KEY", default=""
)
values["client"] = litellm values["client"] = litellm
if values["temperature"] is not None and not 0 <= values["temperature"] <= 1: if values["temperature"] is not None and not 0 <= values["temperature"] <= 1: