feat: complex example test passing

This commit is contained in:
Gabriel Almeida 2023-03-26 01:15:01 -03:00
commit 7dbda097f5
5 changed files with 39 additions and 27 deletions

View file

@ -149,7 +149,7 @@
"show": true,
"password": true,
"multiline": false,
"value": null
"value": "sk-"
},
"batch_size": {
"type": "int",

View file

@ -281,3 +281,11 @@ def test_build():
agent = graph.build()
# The agent should be a AgentExecutor
assert isinstance(agent, AgentExecutor)
# Now we test the complex example
graph = get_graph(basic=False)
assert isinstance(graph, Graph)
# Now we test the build method
agent = graph.build()
# The agent should be a AgentExecutor
assert isinstance(agent, AgentExecutor)

View file

@ -2,7 +2,7 @@ import json
from langchain import LLMChain, OpenAI
from langflow.utils.graph import Graph
import pytest
from pathlib import Path
from langflow import load_flow_from_json
from langflow.interface.loading import extract_json
from langflow.utils.payload import get_root_node, build_json