jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add ChatLiteLLM component to backend

Browse source
This commit is contained in:
Ricardo Henriques 2024-02-18 14:34:37 +00:00
commit 32012df810
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/langflow/components/llms/LiteLLM.py → src/backend/langflow/components/llms/ChatLiteLLM.py
View file

@ -7,7 +7,7 @@ import os
class LiteLLMComponent(CustomComponent): class LiteLLMComponent(CustomComponent):
display_name = "LiteLLM" display_name = "ChatLiteLLM"
description = "`LiteLLM` collection of large language models." description = "`LiteLLM` collection of large language models."
documentation = "https://python.langchain.com/docs/integrations/chat/litellm" documentation = "https://python.langchain.com/docs/integrations/chat/litellm"
@ -37,7 +37,7 @@ class LiteLLMComponent(CustomComponent):
"temperature": { "temperature": {
"display_name": "Temperature", "display_name": "Temperature",
"field_type": "float", "field_type": "float",
"advanced": True, "advanced": False,
"required": False, "required": False,
"default": 0.7, "default": 0.7,
}, },
@ -72,7 +72,7 @@ class LiteLLMComponent(CustomComponent):
"max_tokens": { "max_tokens": {
"display_name": "Max tokens", "display_name": "Max tokens",
"field_type": "int", "field_type": "int",
"advanced": True, "advanced": False,
"required": False, "required": False,
"default": 256, "default": 256,
"info": "The maximum number of tokens to generate for each chat completion.", "info": "The maximum number of tokens to generate for each chat completion.",