refac: fix merge and format

This commit is contained in:
Ibis Prevedello 2023-04-01 17:42:43 -03:00
commit 53c9812819
24 changed files with 346 additions and 71 deletions

View file

@ -1,11 +1,12 @@
import json
import hashlib
from pathlib import Path
import dill
import json
import tempfile
from langflow.cache.utils import compute_hash, load_cache, save_cache, PREFIX
from langflow.interface.run import load_langchain_object, process_graph
from pathlib import Path
import dill
import pytest
from langflow.cache.utils import PREFIX, compute_hash, load_cache, save_cache
from langflow.interface.run import load_langchain_object, process_graph
def get_graph(_type="basic"):

View file

@ -1,7 +1,8 @@
from typing import Dict, List
from langflow.interface.base import LangChainTypeCreator
from langflow.interface.agents.base import AgentCreator
import pytest
from langflow.interface.agents.base import AgentCreator
from langflow.interface.base import LangChainTypeCreator
@pytest.fixture

View file

@ -1,5 +1,5 @@
from langflow.interface.tools.constants import CUSTOM_TOOLS
from fastapi.testclient import TestClient
from langflow.interface.tools.constants import CUSTOM_TOOLS
def test_get_all(client: TestClient):

View file

@ -1,8 +1,9 @@
import pytest
from typing import Dict, List
from langflow.template.base import TemplateField, FrontendNode, Template
from langflow.interface.base import LangChainTypeCreator
import pytest
from langflow.interface.agents.base import AgentCreator
from langflow.interface.base import LangChainTypeCreator
from langflow.template.base import FrontendNode, Template, TemplateField
@pytest.fixture

View file

@ -1,21 +1,20 @@
import json
from langflow.graph.nodes import (
WrapperNode,
AgentNode,
ToolNode,
ChainNode,
PromptNode,
LLMNode,
ToolkitNode,
FileToolNode,
)
import pytest
from langchain.agents import AgentExecutor
from langflow.graph import Edge, Graph, Node
from langflow.graph.nodes import (
AgentNode,
ChainNode,
FileToolNode,
LLMNode,
PromptNode,
ToolkitNode,
ToolNode,
WrapperNode,
)
from langflow.utils.payload import build_json, get_root_node
# Test cases for the graph module
# now we have three types of graph:

View file

@ -1,17 +1,18 @@
from langflow.utils.constants import CHAT_OPENAI_MODELS, OPENAI_MODELS
from pydantic import BaseModel
import pytest
import re
import importlib
import re
from typing import Dict, List, Optional
import pytest
from langflow.utils.constants import CHAT_OPENAI_MODELS, OPENAI_MODELS
from langflow.utils.util import (
build_template_from_class,
build_template_from_function,
format_dict,
get_base_classes,
get_default_factory,
get_class_doc,
get_default_factory,
)
from pydantic import BaseModel
# Dummy classes for testing purposes