feat: Add ruff rules to sort imports (I) (#3869)

Add ruff rules to sort imports (I)
This commit is contained in:
Christophe Bornet 2024-09-25 17:46:35 +02:00 committed by GitHub
commit 381348456d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
163 changed files with 384 additions and 332 deletions

View file

@ -8,8 +8,6 @@ from typing import Optional
import click
import httpx
from langflow.utils.version import get_version_info, fetch_latest_version
from langflow.utils.version import is_pre_release as langflow_is_pre_release
import typer
from dotenv import load_dotenv
from multiprocess import cpu_count # type: ignore
@ -32,6 +30,8 @@ from langflow.services.deps import get_db_service, get_settings_service, session
from langflow.services.settings.constants import DEFAULT_SUPERUSER
from langflow.services.utils import initialize_services
from langflow.utils.util import update_settings
from langflow.utils.version import fetch_latest_version, get_version_info
from langflow.utils.version import is_pre_release as langflow_is_pre_release
console = Console()

View file

@ -1,5 +1,5 @@
from langflow.api.router import router
from langflow.api.health_check_router import health_check_router
from langflow.api.log_router import log_router
from langflow.api.router import router
__all__ = ["router", "health_check_router", "log_router"]

View file

@ -1,10 +1,11 @@
import asyncio
import json
from http import HTTPStatus
from typing import Any
from fastapi import APIRouter, Query, HTTPException, Request
from fastapi import APIRouter, HTTPException, Query, Request
from fastapi.responses import JSONResponse, StreamingResponse
from http import HTTPStatus
from langflow.logging.logger import log_buffer
log_router = APIRouter(tags=["Log"])

View file

@ -7,14 +7,14 @@ from langflow.api.v1 import (
endpoints_router,
files_router,
flows_router,
folders_router,
login_router,
monitor_router,
starter_projects_router,
store_router,
users_router,
validate_router,
variables_router,
folders_router,
starter_projects_router,
)
router = APIRouter(

View file

@ -3,14 +3,14 @@ import warnings
from typing import TYPE_CHECKING, Any
from fastapi import HTTPException
from langflow.services.database.models.transactions.model import TransactionTable
from langflow.services.database.models.vertex_builds.model import VertexBuildTable
from sqlalchemy import delete
from sqlmodel import Session
from langflow.graph.graph.base import Graph
from langflow.services.chat.service import ChatService
from langflow.services.database.models.flow import Flow
from langflow.services.database.models.transactions.model import TransactionTable
from langflow.services.database.models.vertex_builds.model import VertexBuildTable
from langflow.services.store.schema import StoreComponentCreate
from langflow.services.store.utils import get_lf_version_from_pypi

View file

@ -3,14 +3,14 @@ from langflow.api.v1.chat import router as chat_router
from langflow.api.v1.endpoints import router as endpoints_router
from langflow.api.v1.files import router as files_router
from langflow.api.v1.flows import router as flows_router
from langflow.api.v1.folders import router as folders_router
from langflow.api.v1.login import router as login_router
from langflow.api.v1.monitor import router as monitor_router
from langflow.api.v1.starter_projects import router as starter_projects_router
from langflow.api.v1.store import router as store_router
from langflow.api.v1.users import router as users_router
from langflow.api.v1.validate import router as validate_router
from langflow.api.v1.variable import router as variables_router
from langflow.api.v1.folders import router as folders_router
from langflow.api.v1.starter_projects import router as starter_projects_router
__all__ = [
"chat_router",

View file

@ -1,12 +1,13 @@
from typing import TYPE_CHECKING, Any
from uuid import UUID
from langchain_core.agents import AgentAction, AgentFinish
from langchain_core.callbacks.base import AsyncCallbackHandler
from loguru import logger
from langflow.api.v1.schemas import ChatResponse, PromptResponse
from langflow.services.deps import get_chat_service, get_socket_service
from langflow.utils.util import remove_ansi_escape_codes
from langchain_core.agents import AgentAction, AgentFinish
if TYPE_CHECKING:
from langflow.services.socket.service import SocketIOService

View file

@ -1,14 +1,13 @@
from datetime import datetime
import hashlib
from datetime import datetime
from http import HTTPStatus
from io import BytesIO
from uuid import UUID
from pathlib import Path
from uuid import UUID
from fastapi import APIRouter, Depends, HTTPException, UploadFile
from fastapi.responses import StreamingResponse
from langflow.api.v1.schemas import UploadFileResponse
from langflow.services.auth.utils import get_current_active_user
from langflow.services.database.models.flow import Flow

View file

@ -1,9 +1,9 @@
from langflow.api.utils import cascade_delete_flow
import orjson
from fastapi import APIRouter, Depends, File, HTTPException, Response, UploadFile, status
from sqlalchemy import or_, update
from sqlmodel import Session, select
from langflow.api.utils import cascade_delete_flow
from langflow.api.v1.flows import create_flows
from langflow.api.v1.schemas import FlowListCreate, FlowListReadWithFolderName
from langflow.helpers.flow import generate_unique_flow_name

View file

@ -1,6 +1,5 @@
from fastapi import APIRouter, Depends, HTTPException, Request, Response, status
from fastapi.security import OAuth2PasswordRequestForm
from langflow.services.database.models.user.crud import get_user_by_id
from sqlmodel import Session
from langflow.api.v1.schemas import Token
@ -11,6 +10,7 @@ from langflow.services.auth.utils import (
create_user_tokens,
)
from langflow.services.database.models.folder.utils import create_default_folder_if_it_doesnt_exist
from langflow.services.database.models.user.crud import get_user_by_id
from langflow.services.deps import get_session, get_settings_service, get_variable_service
from langflow.services.settings.service import SettingsService
from langflow.services.variable.service import VariableService

View file

@ -10,8 +10,8 @@ from langflow.services.database.models.transactions.crud import get_transactions
from langflow.services.database.models.transactions.model import TransactionReadResponse
from langflow.services.database.models.user.model import User
from langflow.services.database.models.vertex_builds.crud import (
get_vertex_builds_by_flow_id,
delete_vertex_builds_by_flow_id,
get_vertex_builds_by_flow_id,
)
from langflow.services.database.models.vertex_builds.model import VertexBuildMapModel
from langflow.services.deps import get_session

View file

@ -1,10 +1,11 @@
from abc import abstractmethod
from langflow.custom import Component
from langflow.field_typing import BaseChatMessageHistory, BaseChatMemory
from langflow.template import Output
from langchain.memory import ConversationBufferMemory
from langflow.custom import Component
from langflow.field_typing import BaseChatMemory, BaseChatMessageHistory
from langflow.template import Output
class LCChatMemoryComponent(Component):
trace_type = "chat_memory"

View file

@ -5,9 +5,8 @@ from fastapi import HTTPException
from langchain_core.prompts import PromptTemplate
from loguru import logger
from langflow.interface.utils import extract_input_variables_from_prompt
from langflow.inputs.inputs import DefaultPromptField
from langflow.interface.utils import extract_input_variables_from_prompt
_INVALID_CHARACTERS = {
" ",

View file

@ -1,15 +1,16 @@
import json
from typing import Dict, Any, Union
from markdown import markdown
from bs4 import BeautifulSoup
from typing import Any, Dict, Union
import requests
from bs4 import BeautifulSoup
from langchain.tools import StructuredTool
from markdown import markdown
from pydantic import BaseModel, Field
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.inputs import SecretStrInput, StrInput, MultilineInput
from langflow.schema import Data
from langflow.field_typing import Tool
from langchain.tools import StructuredTool
from pydantic import BaseModel, Field
from langflow.inputs import MultilineInput, SecretStrInput, StrInput
from langflow.schema import Data
class AddContentToPage(LCToolComponent):

View file

@ -1,12 +1,14 @@
import json
from typing import Dict, Any, Union
from typing import Any, Dict, Union
import requests
from pydantic import BaseModel, Field
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.inputs import SecretStrInput, StrInput, MultilineInput
from langflow.schema import Data
from langflow.field_typing import Tool
from langchain.tools import StructuredTool
from pydantic import BaseModel, Field
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.field_typing import Tool
from langflow.inputs import MultilineInput, SecretStrInput, StrInput
from langflow.schema import Data
class NotionPageCreator(LCToolComponent):

View file

@ -1,11 +1,13 @@
import requests
from typing import Dict, Union
import requests
from langchain.tools import StructuredTool
from pydantic import BaseModel, Field
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.field_typing import Tool
from langflow.inputs import SecretStrInput, StrInput
from langflow.schema import Data
from langflow.field_typing import Tool
from langchain.tools import StructuredTool
class NotionDatabaseProperties(LCToolComponent):

View file

@ -1,13 +1,14 @@
import requests
import json
from typing import Dict, Any, List, Optional
from typing import Any, Dict, List, Optional
import requests
from langchain.tools import StructuredTool
from pydantic import BaseModel, Field
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.inputs import SecretStrInput, StrInput, MultilineInput
from langflow.schema import Data
from langflow.field_typing import Tool
from langchain.tools import StructuredTool
from langflow.inputs import MultilineInput, SecretStrInput, StrInput
from langflow.schema import Data
class NotionListPages(LCToolComponent):

View file

@ -1,12 +1,13 @@
from typing import Dict, List
import requests
from typing import List, Dict
from langchain.tools import StructuredTool
from pydantic import BaseModel
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.field_typing import Tool
from langflow.inputs import SecretStrInput
from langflow.schema import Data
from langflow.field_typing import Tool
from langchain.tools import StructuredTool
class NotionUserList(LCToolComponent):

View file

@ -1,10 +1,11 @@
import requests
from langchain.tools import StructuredTool
from pydantic import BaseModel, Field
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.field_typing import Tool
from langflow.inputs import SecretStrInput, StrInput
from langflow.schema import Data
from langflow.field_typing import Tool
from langchain.tools import StructuredTool
class NotionPageContent(LCToolComponent):

View file

@ -1,12 +1,13 @@
from typing import Any, Dict, List
import requests
from typing import Dict, Any, List
from langchain.tools import StructuredTool
from pydantic import BaseModel, Field
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.inputs import SecretStrInput, StrInput, DropdownInput
from langflow.schema import Data
from langflow.field_typing import Tool
from langchain.tools import StructuredTool
from langflow.inputs import DropdownInput, SecretStrInput, StrInput
from langflow.schema import Data
class NotionSearch(LCToolComponent):

View file

@ -1,13 +1,15 @@
import json
from typing import Any, Dict, Union
import requests
from typing import Dict, Any, Union
from pydantic import BaseModel, Field
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.inputs import SecretStrInput, StrInput, MultilineInput
from langflow.schema import Data
from langflow.field_typing import Tool
from langchain.tools import StructuredTool
from loguru import logger
from pydantic import BaseModel, Field
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.field_typing import Tool
from langflow.inputs import MultilineInput, SecretStrInput, StrInput
from langflow.schema import Data
class NotionPageUpdate(LCToolComponent):

View file

@ -1,10 +1,10 @@
from langchain_experimental.agents.agent_toolkits.csv.base import create_csv_agent
from langflow.base.agents.agent import LCAgentComponent
from langflow.field_typing import AgentExecutor
from langflow.inputs import HandleInput, FileInput, DropdownInput
from langflow.inputs import DropdownInput, FileInput, HandleInput
from langflow.inputs.inputs import MessageTextInput
from langflow.schema.message import Message
from langflow.template.field.base import Output

View file

@ -7,7 +7,7 @@ from langchain_community.agent_toolkits.json.toolkit import JsonToolkit
from langchain_community.tools.json.tool import JsonSpec
from langflow.base.agents.agent import LCAgentComponent
from langflow.inputs import HandleInput, FileInput
from langflow.inputs import FileInput, HandleInput
class JsonAgentComponent(LCAgentComponent):

View file

@ -1,11 +1,11 @@
from typing import Optional, List
from typing import List, Optional
from langchain.agents import create_openai_tools_agent
from langchain_core.prompts import ChatPromptTemplate, PromptTemplate, HumanMessagePromptTemplate
from langchain_core.prompts import ChatPromptTemplate, HumanMessagePromptTemplate, PromptTemplate
from langflow.base.agents.agent import LCToolsAgentComponent
from langflow.inputs import MultilineInput
from langflow.inputs.inputs import HandleInput, DataInput
from langflow.inputs.inputs import DataInput, HandleInput
from langflow.schema import Data

View file

@ -3,12 +3,12 @@ from pathlib import Path
import yaml
from langchain.agents import AgentExecutor
from langchain_community.agent_toolkits import create_openapi_agent
from langchain_community.tools.json.tool import JsonSpec
from langchain_community.agent_toolkits.openapi.toolkit import OpenAPIToolkit
from langchain_community.tools.json.tool import JsonSpec
from langchain_community.utilities.requests import TextRequestsWrapper
from langflow.base.agents.agent import LCAgentComponent
from langflow.inputs import BoolInput, HandleInput, FileInput
from langchain_community.utilities.requests import TextRequestsWrapper
from langflow.inputs import BoolInput, FileInput, HandleInput
class OpenAPIAgentComponent(LCAgentComponent):

View file

@ -4,7 +4,7 @@ from langchain_community.agent_toolkits.sql.base import create_sql_agent
from langchain_community.utilities import SQLDatabase
from langflow.base.agents.agent import LCAgentComponent
from langflow.inputs import MessageTextInput, HandleInput
from langflow.inputs import HandleInput, MessageTextInput
class SQLAgentComponent(LCAgentComponent):

View file

@ -1,10 +1,11 @@
from typing import Optional, List
from typing import List, Optional
from langchain.agents import create_tool_calling_agent
from langchain_core.prompts import ChatPromptTemplate, PromptTemplate, HumanMessagePromptTemplate
from langchain_core.prompts import ChatPromptTemplate, HumanMessagePromptTemplate, PromptTemplate
from langflow.base.agents.agent import LCToolsAgentComponent
from langflow.inputs import MultilineInput
from langflow.inputs.inputs import HandleInput, DataInput
from langflow.inputs.inputs import DataInput, HandleInput
from langflow.schema import Data

View file

@ -1,5 +1,6 @@
from langchain.agents import AgentExecutor, create_vectorstore_agent
from langchain.agents.agent_toolkits.vectorstore.toolkit import VectorStoreToolkit
from langflow.base.agents.agent import LCAgentComponent
from langflow.inputs import HandleInput

View file

@ -1,8 +1,7 @@
from langchain.agents import create_vectorstore_router_agent
from langchain.agents import AgentExecutor, create_vectorstore_router_agent
from langchain.agents.agent_toolkits.vectorstore.toolkit import VectorStoreRouterToolkit
from langflow.base.agents.agent import LCAgentComponent
from langchain.agents import AgentExecutor
from langflow.inputs import HandleInput

View file

@ -1,6 +1,8 @@
from typing import List, Optional
from langchain.agents import create_xml_agent
from langchain_core.prompts import ChatPromptTemplate, PromptTemplate, HumanMessagePromptTemplate
from langchain_core.prompts import ChatPromptTemplate, HumanMessagePromptTemplate, PromptTemplate
from langflow.base.agents.agent import LCToolsAgentComponent
from langflow.inputs import MultilineInput
from langflow.inputs.inputs import DataInput, HandleInput

View file

@ -2,9 +2,9 @@ from .create_assistant import AssistantsCreateAssistant
from .create_thread import AssistantsCreateThread
from .dotenv import Dotenv
from .get_assistant import AssistantsGetAssistantName
from .getenvvar import GetEnvVar
from .list_assistants import AssistantsListAssistants
from .run import AssistantsRun
from .getenvvar import GetEnvVar
__all__ = [
"AssistantsCreateAssistant",

View file

@ -1,9 +1,10 @@
from astra_assistants import patch # type: ignore
from openai import OpenAI
from langflow.custom import Component
from langflow.inputs import StrInput, MultilineInput
from langflow.template import Output
from langflow.inputs import MultilineInput, StrInput
from langflow.schema.message import Message
from langflow.template import Output
class AssistantsCreateAssistant(Component):

View file

@ -1,6 +1,7 @@
from astra_assistants import patch # type: ignore
from langflow.custom import Component
from openai import OpenAI
from langflow.custom import Component
from langflow.inputs import MultilineInput
from langflow.schema.message import Message
from langflow.template import Output

View file

@ -1,5 +1,7 @@
import io
from dotenv import load_dotenv
from langflow.custom import Component
from langflow.inputs import MultilineSecretInput
from langflow.schema.message import Message

View file

@ -1,8 +1,8 @@
from astra_assistants import patch # type: ignore
from langflow.custom import Component
from openai import OpenAI
from langflow.inputs import StrInput, MultilineInput
from langflow.custom import Component
from langflow.inputs import MultilineInput, StrInput
from langflow.schema.message import Message
from langflow.template import Output

View file

@ -1,4 +1,5 @@
import os
from langflow.custom import Component
from langflow.inputs import StrInput
from langflow.schema.message import Message

View file

@ -1,9 +1,10 @@
from astra_assistants import patch # type: ignore
from typing import Any, Optional
from langflow.custom import Component
from astra_assistants import patch # type: ignore
from openai import OpenAI
from openai.lib.streaming import AssistantEventHandler
from langflow.custom import Component
from langflow.inputs import MultilineInput
from langflow.schema import dotdict
from langflow.schema.message import Message

View file

@ -2,7 +2,7 @@ from langchain.chains import ConversationChain
from langflow.base.chains.model import LCChainComponent
from langflow.field_typing import Message
from langflow.inputs import MultilineInput, HandleInput
from langflow.inputs import HandleInput, MultilineInput
class ConversationChainComponent(LCChainComponent):

View file

@ -2,7 +2,7 @@ from langchain.chains import LLMCheckerChain
from langflow.base.chains.model import LCChainComponent
from langflow.field_typing import Message
from langflow.inputs import MultilineInput, HandleInput
from langflow.inputs import HandleInput, MultilineInput
class LLMCheckerChainComponent(LCChainComponent):

View file

@ -2,7 +2,7 @@ from langchain.chains import LLMMathChain
from langflow.base.chains.model import LCChainComponent
from langflow.field_typing import Message
from langflow.inputs import MultilineInput, HandleInput
from langflow.inputs import HandleInput, MultilineInput
from langflow.template import Output

View file

@ -2,7 +2,7 @@ from langchain.chains import RetrievalQA
from langflow.base.chains.model import LCChainComponent
from langflow.field_typing import Message
from langflow.inputs import HandleInput, MultilineInput, BoolInput, DropdownInput
from langflow.inputs import BoolInput, DropdownInput, HandleInput, MultilineInput
class RetrievalQAComponent(LCChainComponent):

View file

@ -1,9 +1,10 @@
from langchain.chains import create_sql_query_chain
from langchain_core.prompts import PromptTemplate
from langchain_core.runnables import Runnable
from langflow.base.chains.model import LCChainComponent
from langflow.field_typing import Message
from langflow.inputs import MultilineInput, HandleInput, IntInput
from langflow.inputs import HandleInput, IntInput, MultilineInput
from langflow.template import Output

View file

@ -1,19 +1,21 @@
import base64
import re
import json
import re
from json.decoder import JSONDecodeError
from typing import Any, Iterator, List, Optional
from google.auth.exceptions import RefreshError
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from langchain_core.chat_sessions import ChatSession
from langchain_core.messages import HumanMessage
from langchain_google_community.gmail.loader import GMailLoader
from langflow.custom import Component
from langflow.inputs import MessageTextInput
from langflow.io import SecretStrInput
from langflow.template import Output
from langflow.schema import Data
from langchain_google_community.gmail.loader import GMailLoader
from langchain_core.chat_sessions import ChatSession
from langchain_core.messages import HumanMessage
from json.decoder import JSONDecodeError
from google.auth.exceptions import RefreshError
from langflow.template import Output
class GmailLoaderComponent(Component):

View file

@ -1,16 +1,17 @@
import json
from json.decoder import JSONDecodeError
from typing import Optional
from google.oauth2.credentials import Credentials
from google.auth.exceptions import RefreshError
from google.oauth2.credentials import Credentials
from langchain_google_community import GoogleDriveLoader
from langflow.custom import Component
from langflow.helpers.data import docs_to_data
from langflow.inputs import MessageTextInput
from langflow.io import SecretStrInput
from langflow.template import Output
from langflow.schema import Data
from langchain_google_community import GoogleDriveLoader
from langflow.helpers.data import docs_to_data
from json.decoder import JSONDecodeError
from langflow.template import Output
class GoogleDriveComponent(Component):

View file

@ -1,12 +1,14 @@
import json
from typing import List
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from langflow.custom import Component
from langflow.inputs import MessageTextInput, DropdownInput
from langflow.inputs import DropdownInput, MessageTextInput
from langflow.io import SecretStrInput
from langflow.template import Output
from langflow.schema import Data
from langflow.template import Output
class GoogleDriveSearchComponent(Component):

View file

@ -2,8 +2,8 @@ import re
from langchain_community.document_loaders import AsyncHtmlLoader, WebBaseLoader
from langflow.helpers.data import data_to_text
from langflow.custom import Component
from langflow.helpers.data import data_to_text
from langflow.io import DropdownInput, MessageTextInput, Output
from langflow.schema import Data
from langflow.schema.message import Message

View file

@ -1,5 +1,7 @@
from typing import Any, List, Optional
from loguru import logger
from langflow.base.flow_processing.utils import build_data_from_result_data
from langflow.custom import CustomComponent
from langflow.graph.graph.base import Graph
@ -9,7 +11,6 @@ from langflow.helpers.flow import get_flow_inputs
from langflow.schema import Data
from langflow.schema.dotdict import dotdict
from langflow.template.field.base import Input
from loguru import logger
class SubFlowComponent(CustomComponent):

View file

@ -1,10 +1,12 @@
from typing import List
from langflow.custom import Component
from langflow.io import StrInput, SecretStrInput, BoolInput, DropdownInput, Output, IntInput
from langflow.schema import Data
from langchain_community.document_loaders import ConfluenceLoader
from langchain_community.document_loaders.confluence import ContentFormat
from langflow.custom import Component
from langflow.io import BoolInput, DropdownInput, IntInput, Output, SecretStrInput, StrInput
from langflow.schema import Data
class ConfluenceComponent(Component):
display_name = "Confluence"

View file

@ -1,8 +1,9 @@
import re
from pathlib import Path
from typing import List
import re
from langchain_community.document_loaders.git import GitLoader
from langflow.custom import Component
from langflow.io import MessageTextInput, Output
from langflow.schema import Data

View file

@ -1,11 +1,11 @@
from typing import List
from langchain_unstructured import UnstructuredLoader
from langflow.custom import Component
from langflow.inputs import FileInput, SecretStrInput
from langflow.template import Output
from langflow.schema import Data
from langchain_unstructured import UnstructuredLoader
from langflow.template import Output
class UnstructuredComponent(Component):

View file

@ -1,5 +1,7 @@
from typing import List
import numpy as np
from langflow.custom import Component
from langflow.io import DataInput, DropdownInput, Output
from langflow.schema import Data

View file

@ -1,21 +1,20 @@
# from langflow.field_typing import Data
from langflow.custom import Component
from langflow.io import MessageTextInput, Output, SecretStrInput
from langchain_google_genai import GoogleGenerativeAIEmbeddings
from typing import List, Optional
import numpy as np
# TODO: remove ignore once the google package is published with types
from google.ai.generativelanguage_v1beta.types import (
BatchEmbedContentsRequest,
)
from langchain_core.embeddings import Embeddings
from langchain_google_genai import GoogleGenerativeAIEmbeddings
from langchain_google_genai._common import (
GoogleGenerativeAIError,
)
import numpy as np
from langflow.custom import Component
from langflow.io import MessageTextInput, Output, SecretStrInput
class GoogleGenerativeAIEmbeddingsComponent(Component):

View file

@ -1,9 +1,9 @@
from urllib.parse import urlparse
from tenacity import retry, stop_after_attempt, wait_fixed
import requests
from langchain_community.embeddings.huggingface import HuggingFaceInferenceAPIEmbeddings
from pydantic.v1.types import SecretStr
from tenacity import retry, stop_after_attempt, wait_fixed
from langflow.base.embeddings.model import LCEmbeddingsModel
from langflow.field_typing import Embeddings

View file

@ -1,8 +1,8 @@
from langflow.custom import Component
from langflow.io import HandleInput, MessageInput, Output
from langflow.field_typing import Embeddings
from langflow.schema.message import Message
from langflow.io import HandleInput, MessageInput, Output
from langflow.schema import Data
from langflow.schema.message import Message
class TextEmbedderComponent(Component):

View file

@ -3,11 +3,11 @@ from .AmazonBedrockEmbeddings import AmazonBedrockEmbeddingsComponent
from .AstraVectorize import AstraVectorizeComponent
from .AzureOpenAIEmbeddings import AzureOpenAIEmbeddingsComponent
from .CohereEmbeddings import CohereEmbeddingsComponent
from .GoogleGenerativeAIEmbeddings import GoogleGenerativeAIEmbeddingsComponent
from .HuggingFaceInferenceAPIEmbeddings import HuggingFaceInferenceAPIEmbeddingsComponent
from .OllamaEmbeddings import OllamaEmbeddingsComponent
from .OpenAIEmbeddings import OpenAIEmbeddingsComponent
from .VertexAIEmbeddings import VertexAIEmbeddingsComponent
from .GoogleGenerativeAIEmbeddings import GoogleGenerativeAIEmbeddingsComponent
__all__ = [
"AIMLEmbeddingsComponent",

View file

@ -1,8 +1,8 @@
from langflow.custom import Component
from langflow.inputs import MessageInput, StrInput, HandleInput
from langflow.inputs import HandleInput, MessageInput, StrInput
from langflow.memory import get_messages, store_message
from langflow.schema.message import Message
from langflow.template import Output
from langflow.memory import get_messages, store_message
from langflow.utils.constants import MESSAGE_SENDER_AI, MESSAGE_SENDER_NAME_AI

View file

@ -1,4 +1,5 @@
from .CombineText import CombineTextComponent
from .CreateList import CreateListComponent
from .CustomComponent import CustomComponent
from .FilterData import FilterDataComponent
from .IDGenerator import IDGeneratorComponent
@ -7,8 +8,6 @@ from .MergeData import MergeDataComponent
from .ParseData import ParseDataComponent
from .SplitText import SplitTextComponent
from .StoreMessage import StoreMessageComponent
from .CreateList import CreateListComponent
__all__ = [
"CreateListComponent",

View file

@ -4,7 +4,7 @@ from langflow.inputs import BoolInput
from langflow.io import DropdownInput, FileInput, MessageTextInput, MultilineInput, Output
from langflow.memory import store_message
from langflow.schema.message import Message
from langflow.utils.constants import MESSAGE_SENDER_AI, MESSAGE_SENDER_USER, MESSAGE_SENDER_NAME_USER
from langflow.utils.constants import MESSAGE_SENDER_AI, MESSAGE_SENDER_NAME_USER, MESSAGE_SENDER_USER
class ChatInput(ChatComponent):

View file

@ -1,8 +1,9 @@
from langchain_community.utilities.sql_database import SQLDatabase
from langflow.custom import CustomComponent
from sqlalchemy import create_engine
from sqlalchemy.pool import StaticPool
from langflow.custom import CustomComponent
class SQLDatabaseComponent(CustomComponent):
display_name = "SQLDatabase"

View file

@ -1,10 +1,10 @@
from typing import Any
from langchain_community.graph_vectorstores.extractors import LinkExtractorTransformer, HtmlLinkExtractor
from langchain_community.graph_vectorstores.extractors import HtmlLinkExtractor, LinkExtractorTransformer
from langchain_core.documents import BaseDocumentTransformer
from langflow.base.document_transformers.model import LCDocumentTransformerComponent
from langflow.inputs import DataInput, StrInput, BoolInput
from langflow.inputs import BoolInput, DataInput, StrInput
class HtmlLinkExtractorComponent(LCDocumentTransformerComponent):

View file

@ -1,6 +1,6 @@
from langflow.base.memory.model import LCChatMemoryComponent
from langflow.inputs import MessageTextInput, StrInput, SecretStrInput
from langflow.field_typing import BaseChatMessageHistory
from langflow.inputs import MessageTextInput, SecretStrInput, StrInput
class AstraDBChatMemory(LCChatMemoryComponent):

View file

@ -1,6 +1,6 @@
from langflow.base.memory.model import LCChatMemoryComponent
from langflow.inputs import MessageTextInput, SecretStrInput, DictInput
from langflow.field_typing import BaseChatMessageHistory
from langflow.inputs import DictInput, MessageTextInput, SecretStrInput
class CassandraChatMemory(LCChatMemoryComponent):

View file

@ -1,6 +1,6 @@
from langflow.base.memory.model import LCChatMemoryComponent
from langflow.inputs import MessageTextInput, SecretStrInput, DropdownInput
from langflow.field_typing import BaseChatMessageHistory
from langflow.inputs import DropdownInput, MessageTextInput, SecretStrInput
class ZepChatMemory(LCChatMemoryComponent):

View file

@ -1,10 +1,10 @@
from langflow.field_typing.range_spec import RangeSpec
from langchain_openai import ChatOpenAI
from pydantic.v1 import SecretStr
from langflow.base.models.aiml_constants import AIML_CHAT_MODELS
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import LanguageModel
from langflow.field_typing.range_spec import RangeSpec
from langflow.inputs import (
DictInput,
DropdownInput,

View file

@ -1,4 +1,5 @@
from langchain_openai import AzureChatOpenAI
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import LanguageModel
from langflow.inputs import MessageTextInput

View file

@ -1,5 +1,6 @@
import requests
from typing import List
import requests
from langchain_groq import ChatGroq
from pydantic.v1 import SecretStr

View file

@ -1,11 +1,12 @@
from tenacity import retry, stop_after_attempt, wait_fixed
from typing import Any, Dict, Optional
from langchain_community.llms.huggingface_endpoint import HuggingFaceEndpoint
from tenacity import retry, stop_after_attempt, wait_fixed
# TODO: langchain_community.llms.huggingface_endpoint is depreciated. Need to update to langchain_huggingface, but have dependency with langchain_core 0.3.0
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import LanguageModel
from langflow.io import DictInput, DropdownInput, SecretStrInput, StrInput, IntInput, FloatInput
from typing import Any, Dict, Optional
from langflow.io import DictInput, DropdownInput, FloatInput, IntInput, SecretStrInput, StrInput
class HuggingFaceEndpointsComponent(LCModelComponent):

View file

@ -1,13 +1,13 @@
import operator
from functools import reduce
from langflow.field_typing.range_spec import RangeSpec
from langchain_openai import ChatOpenAI
from pydantic.v1 import SecretStr
from langflow.base.models.model import LCModelComponent
from langflow.base.models.openai_constants import OPENAI_MODEL_NAMES
from langflow.field_typing import LanguageModel
from langflow.field_typing.range_spec import RangeSpec
from langflow.inputs import (
BoolInput,
DictInput,

View file

@ -3,7 +3,7 @@ from pydantic.v1 import SecretStr
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import LanguageModel
from langflow.io import FloatInput, SecretStrInput, DropdownInput, IntInput
from langflow.io import DropdownInput, FloatInput, IntInput, SecretStrInput
class PerplexityComponent(LCModelComponent):

View file

@ -8,8 +8,8 @@ from .GoogleGenerativeAIModel import GoogleGenerativeAIComponent
from .HuggingFaceModel import HuggingFaceEndpointsComponent
from .OllamaModel import ChatOllamaComponent
from .OpenAIModel import OpenAIModelComponent
from .VertexAiModel import ChatVertexAIComponent
from .PerplexityModel import PerplexityComponent
from .VertexAiModel import ChatVertexAIComponent
__all__ = [
"AIMLModelComponent",

View file

@ -1,15 +1,13 @@
import re
from typing import List
from langflow.custom import Component
from langflow.inputs import StrInput, SecretStrInput, DefaultPromptField
from langflow.inputs import DefaultPromptField, SecretStrInput, StrInput
from langflow.io import Output
from langflow.schema.message import Message
from langchain_core.prompts import HumanMessagePromptTemplate
import re
class LangChainHubPromptComponent(Component):
display_name: str = "LangChain Hub"
description: str = "Prompt Component that uses LangChain Hub prompts"

View file

@ -1,4 +1,4 @@
from .Prompt import PromptComponent
from .LangChainHubPrompt import LangChainHubPromptComponent
from .Prompt import PromptComponent
__all__ = ["PromptComponent", "LangChainHubPromptComponent"]

View file

@ -1,10 +1,9 @@
from typing import Any
from langflow.custom import Component
from langflow.inputs.inputs import IntInput, MessageTextInput, DictInput, BoolInput
from langflow.io import Output
from langflow.field_typing.range_spec import RangeSpec
from langflow.inputs.inputs import BoolInput, DictInput, IntInput, MessageTextInput
from langflow.io import Output
from langflow.schema import Data
from langflow.schema.dotdict import dotdict

View file

@ -1,10 +1,11 @@
import json
import re
import unicodedata
from langflow.custom import Component
from langflow.inputs import MessageTextInput, BoolInput
from langflow.template import Output
from langflow.inputs import BoolInput, MessageTextInput
from langflow.schema.message import Message
from langflow.template import Output
class JSONCleaner(Component):

View file

@ -1,8 +1,9 @@
from langchain.agents import AgentExecutor
from langflow.custom import Component
from langflow.inputs import HandleInput, MessageTextInput, BoolInput
from langflow.inputs import BoolInput, HandleInput, MessageTextInput
from langflow.schema.message import Message
from langflow.template import Output
from langchain.agents import AgentExecutor
class RunnableExecComponent(Component):

View file

@ -1,8 +1,7 @@
from langflow.custom import Component
from langflow.field_typing.range_spec import RangeSpec
from langflow.inputs.inputs import DataInput, IntInput
from langflow.io import Output
from langflow.field_typing.range_spec import RangeSpec
from langflow.schema import Data

View file

@ -1,10 +1,9 @@
from typing import Any
from langflow.custom import Component
from langflow.inputs.inputs import IntInput, MessageTextInput, DictInput, DataInput, BoolInput
from langflow.io import Output
from langflow.field_typing.range_spec import RangeSpec
from langflow.inputs.inputs import BoolInput, DataInput, DictInput, IntInput, MessageTextInput
from langflow.io import Output
from langflow.schema import Data
from langflow.schema.dotdict import dotdict

View file

@ -1,4 +1,5 @@
from .ConditionalRouter import ConditionalRouterComponent
from .CreateData import CreateDataComponent
from .FlowTool import FlowToolComponent
from .Listen import ListenComponent
from .Notify import NotifyComponent
@ -8,7 +9,6 @@ from .RunFlow import RunFlowComponent
from .RunnableExecutor import RunnableExecComponent
from .SQLExecutor import SQLExecutorComponent
from .SubFlow import SubFlowComponent
from .CreateData import CreateDataComponent
from .UpdateData import UpdateDataComponent
__all__ = [

View file

@ -3,7 +3,7 @@ from typing import Any
from langchain_text_splitters import Language, RecursiveCharacterTextSplitter, TextSplitter
from langflow.base.textsplitters.model import LCTextSplitterComponent
from langflow.inputs import IntInput, DataInput, DropdownInput
from langflow.inputs import DataInput, DropdownInput, IntInput
class LanguageRecursiveTextSplitterComponent(LCTextSplitterComponent):

View file

@ -1,5 +1,7 @@
from typing import Any
from langchain_text_splitters import RecursiveCharacterTextSplitter, TextSplitter
from langflow.base.textsplitters.model import LCTextSplitterComponent
from langflow.inputs.inputs import DataInput, IntInput, MessageTextInput
from langflow.utils.util import unescape_string

View file

@ -1,7 +1,7 @@
from .CharacterTextSplitter import CharacterTextSplitterComponent
from .LanguageRecursiveTextSplitter import LanguageRecursiveTextSplitterComponent
from .RecursiveCharacterTextSplitter import RecursiveCharacterTextSplitterComponent
from .NaturalLanguageTextSplitter import NaturalLanguageTextSplitterComponent
from .RecursiveCharacterTextSplitter import RecursiveCharacterTextSplitterComponent
__all__ = [
"CharacterTextSplitterComponent",

View file

@ -1,6 +1,7 @@
from langchain.agents.agent_toolkits.vectorstore.toolkit import VectorStoreInfo
from langflow.custom import Component
from langflow.inputs import HandleInput, MultilineInput, MessageTextInput
from langflow.inputs import HandleInput, MessageTextInput, MultilineInput
from langflow.template import Output

View file

@ -1,6 +1,6 @@
from .ComposioAPI import ComposioAPIComponent
from .Metaphor import MetaphorToolkit
from .VectorStoreInfo import VectorStoreInfoComponent
from .ComposioAPI import ComposioAPIComponent
__all__ = [
"MetaphorToolkit",

View file

@ -1,12 +1,14 @@
import ast
import operator
from typing import List
from langchain.tools import StructuredTool
from pydantic import BaseModel, Field
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.field_typing import Tool
from langflow.inputs import MessageTextInput
from langflow.schema import Data
from langflow.field_typing import Tool
from langchain.tools import StructuredTool
class CalculatorToolComponent(LCToolComponent):

View file

@ -1,11 +1,13 @@
from typing import Dict, Any, List
from pydantic import BaseModel, Field
from langchain_community.tools import DuckDuckGoSearchRun
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.inputs import MessageTextInput, IntInput
from langflow.schema import Data
from langflow.field_typing import Tool
from typing import Any, Dict, List
from langchain.tools import StructuredTool
from langchain_community.tools import DuckDuckGoSearchRun
from pydantic import BaseModel, Field
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.field_typing import Tool
from langflow.inputs import IntInput, MessageTextInput
from langflow.schema import Data
class DuckDuckGoSearchComponent(LCToolComponent):

View file

@ -1,14 +1,13 @@
import httpx
import json
from typing import Any, Dict, Optional, Union
from urllib.parse import urljoin
import httpx
from langchain_core.pydantic_v1 import BaseModel
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.inputs import SecretStrInput, StrInput, NestedDictInput, IntInput
from langflow.field_typing import Tool
from langflow.inputs import IntInput, NestedDictInput, SecretStrInput, StrInput
from langflow.schema import Data

View file

@ -3,7 +3,7 @@ from typing import Union
from langchain_core.tools import Tool
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.inputs import SecretStrInput, MultilineInput, IntInput
from langflow.inputs import IntInput, MultilineInput, SecretStrInput
from langflow.schema import Data

View file

@ -3,9 +3,9 @@ from typing import Union
from langchain_community.utilities.google_serper import GoogleSerperAPIWrapper
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.inputs import SecretStrInput, MultilineInput, IntInput
from langflow.schema import Data
from langflow.field_typing import Tool
from langflow.inputs import IntInput, MultilineInput, SecretStrInput
from langflow.schema import Data
class GoogleSerperAPIComponent(LCToolComponent):

View file

@ -3,12 +3,12 @@ import json
from typing import Any
from langchain.agents import Tool
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.inputs.inputs import MultilineInput, MessageTextInput, BoolInput, DropdownInput, HandleInput, FieldTypes
from langchain_core.tools import StructuredTool
from pydantic.v1 import Field, create_model
from pydantic.v1.fields import Undefined
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.inputs.inputs import BoolInput, DropdownInput, FieldTypes, HandleInput, MessageTextInput, MultilineInput
from langflow.io import Output
from langflow.schema import Data
from langflow.schema.dotdict import dotdict

View file

@ -1,12 +1,14 @@
import importlib
from typing import List, Union
from pydantic import BaseModel, Field
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.inputs import StrInput
from langflow.schema import Data
from langflow.field_typing import Tool
from langchain.tools import StructuredTool
from langchain_experimental.utilities import PythonREPL
from pydantic import BaseModel, Field
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.field_typing import Tool
from langflow.inputs import StrInput
from langflow.schema import Data
class PythonREPLToolComponent(LCToolComponent):

View file

@ -1,15 +1,15 @@
from typing import Any
import requests
import json
from typing import Any
from pydantic.v1 import Field, create_model
import requests
from langchain.agents import Tool
from langchain_core.tools import StructuredTool
from pydantic.v1 import Field, create_model
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.inputs import MessageTextInput, MultiselectInput, DropdownInput, IntInput
from langflow.schema.dotdict import dotdict
from langflow.inputs import DropdownInput, IntInput, MessageTextInput, MultiselectInput
from langflow.io import Output
from langflow.schema.dotdict import dotdict
class SearXNGToolComponent(LCToolComponent):

View file

@ -1,11 +1,13 @@
from typing import Dict, Any, Optional, List
from pydantic import BaseModel, Field
from langchain_community.utilities.searchapi import SearchApiAPIWrapper
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.inputs import SecretStrInput, MultilineInput, DictInput, MessageTextInput, IntInput
from langflow.schema import Data
from langflow.field_typing import Tool
from typing import Any, Dict, List, Optional
from langchain.tools import StructuredTool
from langchain_community.utilities.searchapi import SearchApiAPIWrapper
from pydantic import BaseModel, Field
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.field_typing import Tool
from langflow.inputs import DictInput, IntInput, MessageTextInput, MultilineInput, SecretStrInput
from langflow.schema import Data
class SearchAPIComponent(LCToolComponent):

View file

@ -1,11 +1,13 @@
from typing import Dict, Any, Optional, List
from pydantic import BaseModel, Field
from langchain_community.utilities.serpapi import SerpAPIWrapper
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.inputs import SecretStrInput, DictInput, MultilineInput, IntInput
from langflow.schema import Data
from langflow.field_typing import Tool
from typing import Any, Dict, List, Optional
from langchain.tools import StructuredTool
from langchain_community.utilities.serpapi import SerpAPIWrapper
from pydantic import BaseModel, Field
from langflow.base.langchain_utilities.model import LCToolComponent
from langflow.field_typing import Tool
from langflow.inputs import DictInput, IntInput, MultilineInput, SecretStrInput
from langflow.schema import Data
class SerpAPIComponent(LCToolComponent):

View file

@ -1,4 +1,5 @@
from typing import cast
from langchain_community.tools import WikipediaQueryRun
from langchain_community.utilities.wikipedia import WikipediaAPIWrapper

View file

@ -1,17 +1,16 @@
from .PythonREPLTool import PythonREPLToolComponent
from .RetrieverTool import RetrieverToolComponent
from .BingSearchAPI import BingSearchAPIComponent
from .Calculator import CalculatorToolComponent
from .GleanSearchAPI import GleanSearchAPIComponent
from .GoogleSearchAPI import GoogleSearchAPIComponent
from .GoogleSerperAPI import GoogleSerperAPIComponent
from .PythonCodeStructuredTool import PythonCodeStructuredTool
from .PythonREPLTool import PythonREPLToolComponent
from .RetrieverTool import RetrieverToolComponent
from .SearchAPI import SearchAPIComponent
from .SearXNGTool import SearXNGToolComponent
from .SerpAPI import SerpAPIComponent
from .WikipediaAPI import WikipediaAPIComponent
from .WolframAlphaAPI import WolframAlphaAPIComponent
from .Calculator import CalculatorToolComponent
__all__ = [
"RetrieverToolComponent",

View file

@ -1,8 +1,8 @@
from typing import List
from uuid import UUID
from langchain_community.graph_vectorstores import CassandraGraphVectorStore
from loguru import logger
from uuid import UUID
from langflow.base.vectorstores.model import LCVectorStoreComponent, check_cached_vector_store
from langflow.helpers.data import docs_to_data

View file

@ -7,7 +7,7 @@ from loguru import logger
from langflow.base.vectorstores.model import LCVectorStoreComponent, check_cached_vector_store
from langflow.base.vectorstores.utils import chroma_collection_to_data
from langflow.io import BoolInput, DataInput, DropdownInput, HandleInput, IntInput, StrInput, MultilineInput
from langflow.io import BoolInput, DataInput, DropdownInput, HandleInput, IntInput, MultilineInput, StrInput
from langflow.schema import Data
if TYPE_CHECKING:

View file

@ -6,14 +6,14 @@ from langflow.base.vectorstores.model import LCVectorStoreComponent, check_cache
from langflow.helpers.data import docs_to_data
from langflow.inputs import BoolInput, FloatInput
from langflow.io import (
DataInput,
DictInput,
DropdownInput,
HandleInput,
IntInput,
StrInput,
SecretStrInput,
DataInput,
DropdownInput,
MultilineInput,
DictInput,
SecretStrInput,
StrInput,
)
from langflow.schema import Data

View file

@ -5,7 +5,7 @@ from langchain_community.vectorstores import CouchbaseVectorStore
from langflow.base.vectorstores.model import LCVectorStoreComponent, check_cached_vector_store
from langflow.helpers.data import docs_to_data
from langflow.io import HandleInput, IntInput, StrInput, SecretStrInput, DataInput, MultilineInput
from langflow.io import DataInput, HandleInput, IntInput, MultilineInput, SecretStrInput, StrInput
from langflow.schema import Data

View file

@ -185,8 +185,8 @@ class HCDVectorStoreComponent(LCVectorStoreComponent):
)
try:
from astrapy.constants import Environment
from astrapy.authentication import UsernamePasswordTokenProvider
from astrapy.constants import Environment
except ImportError:
raise ImportError(
"Could not import astrapy integration package. " "Please install it with `pip install astrapy`."

Some files were not shown because too many files have changed in this diff Show more