Update model imports in component files

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-25 14:45:48 -03:00
commit db4a4e52aa
14 changed files with 12 additions and 12 deletions

View file

@ -0,0 +1,3 @@
from .model import LCModelComponent
__all__ = ["LCModelComponent"]

View file

@ -0,0 +1,28 @@
from langchain_core.runnables import Runnable
from langflow import CustomComponent
class LCModelComponent(CustomComponent):
display_name: str = "Model Name"
description: str = "Model Description"
def get_result(self, output: Runnable, stream: bool, input_value: str):
"""
Retrieves the result from the output of a Runnable object.
Args:
output (Runnable): The output object to retrieve the result from.
stream (bool): Indicates whether to use streaming or invocation mode.
input_value (str): The input value to pass to the output object.
Returns:
The result obtained from the output object.
"""
if stream:
result = output.stream(input_value)
else:
message = output.invoke(input_value)
result = message.content if hasattr(message, "content") else message
self.status = result
return result

View file

@ -2,7 +2,7 @@ from typing import Optional
from langchain_community.chat_models.bedrock import BedrockChat
from langflow.components.models.base.model import LCModelComponent
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import Text

View file

@ -3,7 +3,7 @@ from typing import Optional
from langchain_anthropic.chat_models import ChatAnthropic
from pydantic.v1 import SecretStr
from langflow.components.models.base.model import LCModelComponent
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import Text

View file

@ -4,7 +4,7 @@ from langchain.llms.base import BaseLanguageModel
from langchain_openai import AzureChatOpenAI
from pydantic.v1 import SecretStr
from langflow.components.models.base.model import LCModelComponent
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import Text

View file

@ -3,7 +3,7 @@ from typing import Optional
from langchain_community.chat_models.baidu_qianfan_endpoint import QianfanChatEndpoint
from pydantic.v1 import SecretStr
from langflow.components.models.base.model import LCModelComponent
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import Text

View file

@ -2,7 +2,7 @@ from typing import Dict, Optional
from langchain_community.llms.ctransformers import CTransformers
from langflow.components.models.base.model import LCModelComponent
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import Text

View file

@ -1,7 +1,7 @@
from langchain_community.chat_models.cohere import ChatCohere
from pydantic.v1 import SecretStr
from langflow.components.models.base.model import LCModelComponent
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import Text

View file

@ -3,7 +3,7 @@ from typing import Optional
from langchain_google_genai import ChatGoogleGenerativeAI
from pydantic.v1 import SecretStr
from langflow.components.models.base.model import LCModelComponent
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import RangeSpec, Text

View file

@ -3,7 +3,7 @@ from typing import Optional
from langchain_community.chat_models.huggingface import ChatHuggingFace
from langchain_community.llms.huggingface_endpoint import HuggingFaceEndpoint
from langflow.components.models.base.model import LCModelComponent
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import Text

View file

@ -2,7 +2,7 @@ from typing import Any, Dict, List, Optional
from langchain_community.llms.llamacpp import LlamaCpp
from langflow.components.models.base.model import LCModelComponent
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import Text

View file

@ -3,7 +3,7 @@ from typing import Any, Dict, List, Optional
# from langchain_community.chat_models import ChatOllama
from langchain_community.chat_models import ChatOllama
from langflow.components.models.base.model import LCModelComponent
from langflow.base.models.model import LCModelComponent
# from langchain.chat_models import ChatOllama
from langflow.field_typing import Text

View file

@ -2,7 +2,7 @@ from typing import Optional
from langchain_openai import ChatOpenAI
from langflow.components.models.base.model import LCModelComponent
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import NestedDict, Text

View file

@ -2,7 +2,7 @@ from typing import List, Optional
from langchain_core.messages.base import BaseMessage
from langflow.components.models.base.model import LCModelComponent
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import Text