From a8b53c67d89edd40f4e01e9b83304353f25bd3d7 Mon Sep 17 00:00:00 2001 From: Rodrigo Nader Date: Thu, 7 Mar 2024 02:36:12 -0300 Subject: [PATCH] Update display names of model components --- src/backend/langflow/components/{prompts => inputs}/Prompt.py | 1 + src/backend/langflow/components/models/AmazonBedrockModel.py | 2 +- src/backend/langflow/components/models/AnthropicModel.py | 2 +- src/backend/langflow/components/models/AzureOpenAIModel.py | 2 +- src/backend/langflow/components/models/BaiduQianfanChatModel.py | 2 +- src/backend/langflow/components/models/CTransformersModel.py | 2 +- src/backend/langflow/components/models/CohereModel.py | 2 +- .../langflow/components/models/GoogleGenerativeAIModel.py | 2 +- src/backend/langflow/components/models/HuggingFaceModel.py | 2 +- src/backend/langflow/components/models/LlamaCppModel.py | 2 +- src/backend/langflow/components/models/OllamaModel.py | 2 +- src/backend/langflow/components/models/OpenAIModel.py | 2 +- src/backend/langflow/components/models/VertexAiModel.py | 2 +- src/backend/langflow/components/prompts/__init_.py | 0 14 files changed, 13 insertions(+), 12 deletions(-) rename src/backend/langflow/components/{prompts => inputs}/Prompt.py (97%) delete mode 100644 src/backend/langflow/components/prompts/__init_.py diff --git a/src/backend/langflow/components/prompts/Prompt.py b/src/backend/langflow/components/inputs/Prompt.py similarity index 97% rename from src/backend/langflow/components/prompts/Prompt.py rename to src/backend/langflow/components/inputs/Prompt.py index a260c1c47..8ee6f0232 100644 --- a/src/backend/langflow/components/prompts/Prompt.py +++ b/src/backend/langflow/components/inputs/Prompt.py @@ -9,6 +9,7 @@ class PromptComponent(CustomComponent): display_name: str = "Prompt" description: str = "A component for creating prompts using templates" beta = True + icon = "terminal-square" def build_config(self): return { diff --git a/src/backend/langflow/components/models/AmazonBedrockModel.py b/src/backend/langflow/components/models/AmazonBedrockModel.py index ce7347fde..00ece9a23 100644 --- a/src/backend/langflow/components/models/AmazonBedrockModel.py +++ b/src/backend/langflow/components/models/AmazonBedrockModel.py @@ -7,7 +7,7 @@ from langflow.field_typing import Text class AmazonBedrockComponent(LCModelComponent): - display_name: str = "Amazon Bedrock Model" + display_name: str = "Amazon Bedrock" description: str = "Generate text using LLM model from Amazon Bedrock." icon = "Amazon" diff --git a/src/backend/langflow/components/models/AnthropicModel.py b/src/backend/langflow/components/models/AnthropicModel.py index fb891f9a3..83a7fd461 100644 --- a/src/backend/langflow/components/models/AnthropicModel.py +++ b/src/backend/langflow/components/models/AnthropicModel.py @@ -8,7 +8,7 @@ from langflow.field_typing import Text class AnthropicLLM(LCModelComponent): - display_name: str = "AnthropicModel" + display_name: str = "Anthropic" description: str = "Generate text using Anthropic Chat&Completion large language models." icon = "Anthropic" diff --git a/src/backend/langflow/components/models/AzureOpenAIModel.py b/src/backend/langflow/components/models/AzureOpenAIModel.py index 81c22d399..5d16cc0c4 100644 --- a/src/backend/langflow/components/models/AzureOpenAIModel.py +++ b/src/backend/langflow/components/models/AzureOpenAIModel.py @@ -9,7 +9,7 @@ from langflow.field_typing import Text class AzureChatOpenAIComponent(LCModelComponent): - display_name: str = "AzureOpenAI Model" + display_name: str = "AzureOpenAI" description: str = "Generate text using LLM model from Azure OpenAI." documentation: str = "https://python.langchain.com/docs/integrations/llms/azure_openai" beta = False diff --git a/src/backend/langflow/components/models/BaiduQianfanChatModel.py b/src/backend/langflow/components/models/BaiduQianfanChatModel.py index 1ef65ee33..76ae69e42 100644 --- a/src/backend/langflow/components/models/BaiduQianfanChatModel.py +++ b/src/backend/langflow/components/models/BaiduQianfanChatModel.py @@ -8,7 +8,7 @@ from langflow.field_typing import Text class QianfanChatEndpointComponent(LCModelComponent): - display_name: str = "QianfanChat Model" + display_name: str = "QianfanChat" description: str = ( "Generate text using Baidu Qianfan chat models. Get more detail from " "https://python.langchain.com/docs/integrations/chat/baidu_qianfan_endpoint." diff --git a/src/backend/langflow/components/models/CTransformersModel.py b/src/backend/langflow/components/models/CTransformersModel.py index 219d74440..160f698db 100644 --- a/src/backend/langflow/components/models/CTransformersModel.py +++ b/src/backend/langflow/components/models/CTransformersModel.py @@ -7,7 +7,7 @@ from langflow.field_typing import Text class CTransformersComponent(LCModelComponent): - display_name = "CTransformersModel" + display_name = "CTransformers" description = "Generate text using CTransformers LLM models" documentation = "https://python.langchain.com/docs/modules/model_io/models/llms/integrations/ctransformers" diff --git a/src/backend/langflow/components/models/CohereModel.py b/src/backend/langflow/components/models/CohereModel.py index c0b603695..1cab1bec7 100644 --- a/src/backend/langflow/components/models/CohereModel.py +++ b/src/backend/langflow/components/models/CohereModel.py @@ -5,7 +5,7 @@ from langflow.field_typing import Text class CohereComponent(LCModelComponent): - display_name = "CohereModel" + display_name = "Cohere" description = "Generate text using Cohere large language models." documentation = "https://python.langchain.com/docs/modules/model_io/models/llms/integrations/cohere" diff --git a/src/backend/langflow/components/models/GoogleGenerativeAIModel.py b/src/backend/langflow/components/models/GoogleGenerativeAIModel.py index c979e6db4..b8c3d3331 100644 --- a/src/backend/langflow/components/models/GoogleGenerativeAIModel.py +++ b/src/backend/langflow/components/models/GoogleGenerativeAIModel.py @@ -8,7 +8,7 @@ from langflow.field_typing import RangeSpec, Text class GoogleGenerativeAIComponent(LCModelComponent): - display_name: str = "Google Generative AIModel" + display_name: str = "Google Generative AI" description: str = "Generate text using Google Generative AI to generate text." icon = "GoogleGenerativeAI" icon = "Google" diff --git a/src/backend/langflow/components/models/HuggingFaceModel.py b/src/backend/langflow/components/models/HuggingFaceModel.py index 33cc57815..c56015741 100644 --- a/src/backend/langflow/components/models/HuggingFaceModel.py +++ b/src/backend/langflow/components/models/HuggingFaceModel.py @@ -8,7 +8,7 @@ from langflow.field_typing import Text class HuggingFaceEndpointsComponent(LCModelComponent): - display_name: str = "Hugging Face Inference API models" + display_name: str = "Hugging Face Inference API" description: str = "Generate text using LLM model from Hugging Face Inference API." icon = "HuggingFace" diff --git a/src/backend/langflow/components/models/LlamaCppModel.py b/src/backend/langflow/components/models/LlamaCppModel.py index 468a1ac8b..48cea7ad0 100644 --- a/src/backend/langflow/components/models/LlamaCppModel.py +++ b/src/backend/langflow/components/models/LlamaCppModel.py @@ -7,7 +7,7 @@ from langflow.field_typing import Text class LlamaCppComponent(LCModelComponent): - display_name = "LlamaCppModel" + display_name = "LlamaCpp" description = "Generate text using llama.cpp model." documentation = "https://python.langchain.com/docs/modules/model_io/models/llms/integrations/llamacpp" diff --git a/src/backend/langflow/components/models/OllamaModel.py b/src/backend/langflow/components/models/OllamaModel.py index 9de1847f0..cfd1e2a2e 100644 --- a/src/backend/langflow/components/models/OllamaModel.py +++ b/src/backend/langflow/components/models/OllamaModel.py @@ -13,7 +13,7 @@ from langflow.field_typing import Text class ChatOllamaComponent(LCModelComponent): - display_name = "ChatOllamaModel" + display_name = "ChatOllama" description = "Generate text using Local LLM for chat with Ollama." icon = "Ollama" diff --git a/src/backend/langflow/components/models/OpenAIModel.py b/src/backend/langflow/components/models/OpenAIModel.py index f595255ce..2a8314fd2 100644 --- a/src/backend/langflow/components/models/OpenAIModel.py +++ b/src/backend/langflow/components/models/OpenAIModel.py @@ -8,7 +8,7 @@ from langflow.field_typing import NestedDict, Text class OpenAIModelComponent(LCModelComponent): - display_name = "OpenAI Model" + display_name = "OpenAI" description = "Generates text using OpenAI's models." icon = "OpenAI" diff --git a/src/backend/langflow/components/models/VertexAiModel.py b/src/backend/langflow/components/models/VertexAiModel.py index c4354d999..11c49c7b4 100644 --- a/src/backend/langflow/components/models/VertexAiModel.py +++ b/src/backend/langflow/components/models/VertexAiModel.py @@ -7,7 +7,7 @@ from langflow.field_typing import Text class ChatVertexAIComponent(LCModelComponent): - display_name = "ChatVertexAIModel" + display_name = "ChatVertexAI" description = "Generate text using Vertex AI Chat large language models API." icon = "VertexAI" diff --git a/src/backend/langflow/components/prompts/__init_.py b/src/backend/langflow/components/prompts/__init_.py deleted file mode 100644 index e69de29bb..000000000