Fix code formatting and import statements

src/backend/langflow/api/v1/chat.py

@@ -1,19 +1,15 @@
 import time
 
-from fastapi import (APIRouter, Depends, HTTPException, Query, WebSocket,
+from fastapi import APIRouter, Depends, HTTPException, Query, WebSocket, WebSocketException, status
-                     WebSocketException, status)
 from fastapi.responses import StreamingResponse
 from langflow.api.utils import build_input_keys_response, format_elapsed_time
-from langflow.api.v1.schemas import (BuildStatus, BuiltResponse, InitResponse,
+from langflow.api.v1.schemas import BuildStatus, BuiltResponse, InitResponse, StreamData
-                                     StreamData)
 from langflow.graph.graph.base import Graph
-from langflow.services.auth.utils import (get_current_active_user,
+from langflow.services.auth.utils import get_current_active_user, get_current_user_by_jwt
-                                          get_current_user_by_jwt)
 from langflow.services.cache.service import BaseCacheService
 from langflow.services.cache.utils import update_build_status
 from langflow.services.chat.service import ChatService
-from langflow.services.deps import (get_cache_service, get_chat_service,
+from langflow.services.deps import get_cache_service, get_chat_service, get_session
-                                    get_session)
 from loguru import logger
 from sqlmodel import Session
 

src/backend/langflow/components/llms/AzureChatOpenAI.py

@@ -3,6 +3,7 @@ from langflow import CustomComponent
 from langchain.llms.base import BaseLanguageModel
 from langchain.chat_models.azure_openai import AzureChatOpenAI
 
+
 class AzureChatOpenAIComponent(CustomComponent):
     display_name: str = "AzureChatOpenAI"
     description: str = "LLM model from Azure OpenAI."
@@ -28,7 +29,7 @@ class AzureChatOpenAIComponent(CustomComponent):
             "azure_endpoint": {
                 "display_name": "Azure Endpoint",
                 "required": True,
-                "info": "Your Azure endpoint, including the resource.. Example: `https://example-resource.azure.openai.com/`"
+                "info": "Your Azure endpoint, including the resource.. Example: `https://example-resource.azure.openai.com/`",
             },
             "azure_deployment": {
                 "display_name": "Deployment Name",
@@ -40,11 +41,7 @@ class AzureChatOpenAIComponent(CustomComponent):
                 "required": True,
                 "advanced": True,
             },
-            "api_key": {
+            "api_key": {"display_name": "API Key", "required": True, "password": True},
-                "display_name": "API Key",
-                "required": True,
-                "password": True
-            },
             "temperature": {
                 "display_name": "Temperature",
                 "value": 0.7,
@@ -71,8 +68,6 @@ class AzureChatOpenAIComponent(CustomComponent):
         temperature: float = 0.7,
         max_tokens: Optional[int] = 1000,
     ) -> BaseLanguageModel:
-
-
         return AzureChatOpenAI(
             model=model,
             azure_endpoint=azure_endpoint,
@@ -80,5 +75,5 @@ class AzureChatOpenAIComponent(CustomComponent):
             api_version=api_version,
             api_key=api_key,
             temperature=temperature,
-            max_tokens=max_tokens
+            max_tokens=max_tokens,
         )

src/backend/langflow/components/vectorstores/Weaviate.py

@@ -12,18 +12,29 @@ from langchain.embeddings.base import Embeddings
 class WeaviateVectorStore(CustomComponent):
     display_name: str = "Weaviate"
     description: str = "Implementation of Vector Store using Weaviate"
-    documentation = (
+    documentation = "https://python.langchain.com/docs/integrations/vectorstores/weaviate"
-        "https://python.langchain.com/docs/integrations/vectorstores/weaviate"
-    )
     beta = True
     field_config = {
         "url": {"display_name": "Weaviate URL", "value": "http://localhost:8080"},
-        "api_key": { "display_name": "API Key", "password": True,"required": False, },
+        "api_key": {
-        "index_name": {"display_name": "Index name","required": False,},
+            "display_name": "API Key",
+            "password": True,
+            "required": False,
+        },
+        "index_name": {
+            "display_name": "Index name",
+            "required": False,
+        },
         "text_key": {"display_name": "Text Key", "required": False, "advanced": True, "value": "text"},
         "documents": {"display_name": "Documents", "is_list": True},
         "embedding": {"display_name": "Embedding"},
-        "attributes": {"display_name": "Attributes", "required": False, "is_list": True, "field_type": "str", "advanced": True},
+        "attributes": {
+            "display_name": "Attributes",
+            "required": False,
+            "is_list": True,
+            "field_type": "str",
+            "advanced": True,
+        },
         "search_by_text": {"display_name": "Search By Text", "field_type": "bool", "advanced": True},
         "code": {"show": False},
     }

src/backend/langflow/interface/agents/custom.py

@@ -1,6 +1,6 @@
-from typing import Any, List, Optional
+from typing import Any, Optional
 
-from langchain.agents import AgentExecutor, AgentType, Tool, ZeroShotAgent, initialize_agent
+from langchain.agents import AgentExecutor, ZeroShotAgent
 from langchain.agents.agent_toolkits import (
     SQLDatabaseToolkit,
     VectorStoreInfo,
@@ -15,7 +15,6 @@ from langchain.agents.agent_toolkits.vectorstore.prompt import ROUTER_PREFIX as
 from langchain.agents.mrkl.prompt import FORMAT_INSTRUCTIONS
 from langchain.base_language import BaseLanguageModel
 from langchain.chains.llm import LLMChain
-from langchain.memory.chat_memory import BaseChatMemory
 from langchain.sql_database import SQLDatabase
 from langchain.tools.sql_database.prompt import QUERY_CHECKER
 from langchain_experimental.agents.agent_toolkits.pandas.prompt import PREFIX as PANDAS_PREFIX

src/backend/langflow/interface/custom/code_parser/code_parser.py

@@ -6,8 +6,7 @@ from typing import Any, Dict, List, Type, Union
 
 from cachetools import TTLCache, cachedmethod, keys
 from fastapi import HTTPException
-from langflow.interface.custom.schema import (CallableCodeDetails,
+from langflow.interface.custom.schema import CallableCodeDetails, ClassCodeDetails
-                                              ClassCodeDetails)
 
 
 class CodeSyntaxError(HTTPException):

src/backend/langflow/interface/custom/directory_reader/directory_reader.py

@@ -65,6 +65,7 @@ class DirectoryReader:
 
     def filter_loaded_components(self, data: dict, with_errors: bool) -> dict:
         from langflow.interface.custom.utils import build_component
+
         items = [
             {
                 "name": menu["name"],

src/backend/langflow/interface/custom/directory_reader/utils.py

@@ -1,6 +1,5 @@
 from langflow.interface.custom.directory_reader import DirectoryReader
-from langflow.template.frontend_node.custom_components import \
+from langflow.template.frontend_node.custom_components import CustomComponentFrontendNode
-    CustomComponentFrontendNode
 from loguru import logger
 
 
@@ -79,7 +78,6 @@ def create_invalid_component_template(component, component_name):
         display_name=f"ERROR - {component_name}",
     )
 
-
     component_frontend_node.error = component.get("error", None)
     field = component_frontend_node.template.get_field("code")
     field.value = component_code

src/backend/langflow/interface/types.py

@@ -1,9 +1,7 @@
-
 from cachetools import LRUCache, cached
 from langflow.interface.agents.base import agent_creator
 from langflow.interface.chains.base import chain_creator
-from langflow.interface.custom.directory_reader.utils import \
+from langflow.interface.custom.directory_reader.utils import merge_nested_dicts_with_renaming
-    merge_nested_dicts_with_renaming
 from langflow.interface.custom.utils import build_custom_components
 from langflow.interface.document_loaders.base import documentloader_creator
 from langflow.interface.embeddings.base import embedding_creator
@@ -70,5 +68,3 @@ def get_all_types_dict(settings_service):
     native_components = build_langchain_types_dict()
     custom_components_from_file = build_custom_components(settings_service)
     return merge_nested_dicts_with_renaming(native_components, custom_components_from_file)
-
-