",
"Rodrigo Nader ",
]
-repository = "https://github.com/logspace-ai/langflow"
+repository = "https://github.com/langflow-ai/langflow"
license = "MIT"
readme = "README.md"
keywords = ["nlp", "langchain", "openai", "gpt", "gui"]
@@ -43,7 +43,7 @@ typer = "^0.12.0"
cachetools = "^5.3.1"
platformdirs = "^4.2.0"
python-multipart = "^0.0.7"
-orjson = "3.9.15"
+orjson = "3.10.0"
alembic = "^1.13.0"
passlib = "^1.7.4"
bcrypt = "4.0.1"
diff --git a/src/frontend/src/components/CrashErrorComponent/index.tsx b/src/frontend/src/components/CrashErrorComponent/index.tsx
index 57dc83d83..978c9b340 100644
--- a/src/frontend/src/components/CrashErrorComponent/index.tsx
+++ b/src/frontend/src/components/CrashErrorComponent/index.tsx
@@ -28,7 +28,7 @@ export default function CrashErrorComponent({
Please report errors with detailed tracebacks on the{" "}
Restart Langflow
diff --git a/src/frontend/src/components/headerComponent/index.tsx b/src/frontend/src/components/headerComponent/index.tsx
index 68e95cf2c..c302800c0 100644
--- a/src/frontend/src/components/headerComponent/index.tsx
+++ b/src/frontend/src/components/headerComponent/index.tsx
@@ -114,7 +114,7 @@ export default function Header(): JSX.Element {
{
const authorizedDomains = [
- "https://raw.githubusercontent.com/logspace-ai/langflow_examples/main/examples",
- "https://api.github.com/repos/logspace-ai/langflow_examples/contents/examples",
- "https://api.github.com/repos/logspace-ai/langflow",
+ "https://raw.githubusercontent.com/langflow-ai/langflow_examples/main/examples",
+ "https://api.github.com/repos/langflow-ai/langflow_examples/contents/examples",
+ "https://api.github.com/repos/langflow-ai/langflow",
"auto_login",
];
diff --git a/src/frontend/src/controllers/API/index.ts b/src/frontend/src/controllers/API/index.ts
index 52082abd9..920335a97 100644
--- a/src/frontend/src/controllers/API/index.ts
+++ b/src/frontend/src/controllers/API/index.ts
@@ -90,7 +90,7 @@ export async function postValidatePrompt(
*/
export async function getExamples(): Promise {
const url =
- "https://api.github.com/repos/logspace-ai/langflow_examples/contents/examples?ref=main";
+ "https://api.github.com/repos/langflow-ai/langflow_examples/contents/examples?ref=main";
const response = await api.get(url);
const jsonFiles = response.data.filter((file: any) => {
diff --git a/src/frontend/src/stores/typesStore.ts b/src/frontend/src/stores/typesStore.ts
index 808a5d646..142118f44 100644
--- a/src/frontend/src/stores/typesStore.ts
+++ b/src/frontend/src/stores/typesStore.ts
@@ -29,6 +29,7 @@ export const useTypesStore = create((set, get) => ({
.catch((error) => {
console.error("An error has occurred while fetching types.");
console.log(error);
+ setLoading(false);
reject();
});
});
diff --git a/src/frontend/src/utils/utils.ts b/src/frontend/src/utils/utils.ts
index 91d7a72f9..afcc264f6 100644
--- a/src/frontend/src/utils/utils.ts
+++ b/src/frontend/src/utils/utils.ts
@@ -456,7 +456,7 @@ export function getWidgetCode(
const inputs = buildInputs();
let chat_input_field = getChatInputField(flowState);
- return `
+ return `
Document:
+ def build(self, url: str,file:str,integer:int,nested:NestedDict,flt:float,boolean:bool,lisst:list[str],dictionary:dict, llm: BaseLanguageModel, prompt: PromptTemplate) -> Document:
return "test"`;
diff --git a/src/frontend/tests/end-to-end/dropdownComponent.spec.ts b/src/frontend/tests/end-to-end/dropdownComponent.spec.ts
index a31ceae44..0623e7460 100644
--- a/src/frontend/tests/end-to-end/dropdownComponent.spec.ts
+++ b/src/frontend/tests/end-to-end/dropdownComponent.spec.ts
@@ -177,7 +177,7 @@ test("dropDownComponent", async ({ page }) => {
.click();
await page.locator("textarea").press("Control+a");
const emptyOptionsCode = `from typing import Optional
-from langchain.llms.base import BaseLLM
+from langflow.field_typing import BaseLanguageModel
from langchain_community.llms.bedrock import Bedrock
from langflow.interface.custom.custom_component import CustomComponent
@@ -212,7 +212,7 @@ class AmazonBedrockComponent(CustomComponent):
endpoint_url: Optional[str] = None,
streaming: bool = False,
cache: Optional[bool] = None,
- ) -> BaseLLM:
+ ) -> BaseLanguageModel:
try:
output = Bedrock(
credentials_profile_name=credentials_profile_name,
diff --git a/tests/conftest.py b/tests/conftest.py
index 9b6d7c0d6..cfec4f7ef 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -22,9 +22,6 @@ from langflow.services.database.models.flow.model import Flow, FlowCreate
from langflow.services.database.models.user.model import User, UserCreate
from langflow.services.database.utils import session_getter
from langflow.services.deps import get_db_service
-from sqlmodel import Session, SQLModel, create_engine, select
-from sqlmodel.pool import StaticPool
-from typer.testing import CliRunner
if TYPE_CHECKING:
from langflow.services.database.service import DatabaseService
diff --git a/tests/test_custom_component.py b/tests/test_custom_component.py
index 725d35564..07796c540 100644
--- a/tests/test_custom_component.py
+++ b/tests/test_custom_component.py
@@ -6,21 +6,15 @@ import pytest
from langchain_core.documents import Document
from langflow.interface.custom.base import CustomComponent
-from langflow.interface.custom.code_parser.code_parser import (
- CodeParser,
- CodeSyntaxError,
-)
-from langflow.interface.custom.custom_component.component import (
- Component,
- ComponentCodeNullError,
-)
+from langflow.interface.custom.code_parser.code_parser import CodeParser, CodeSyntaxError
+from langflow.interface.custom.custom_component.component import Component, ComponentCodeNullError
from langflow.services.database.models.flow import Flow, FlowCreate
code_default = """
from langflow.field_typing import Prompt
from langflow.interface.custom.custom_component import CustomComponent
-from langchain.llms.base import BaseLLM
+from langflow.field_typing import BaseLanguageModel
from langchain.chains import LLMChain
from langchain.prompts import PromptTemplate
from langchain_core.documents import Document
@@ -32,7 +26,7 @@ class YourComponent(CustomComponent):
description: str = "Your description"
field_config = { "url": { "multiline": True, "required": True } }
- def build(self, url: str, llm: BaseLLM, template: Prompt) -> Document:
+ def build(self, url: str, llm: BaseLanguageModel, template: Prompt) -> Document:
response = requests.get(url)
prompt = PromptTemplate.from_template(template)
chain = LLMChain(llm=llm, prompt=prompt)
diff --git a/tests/test_data_components.py b/tests/test_data_components.py
index ca92ee190..1887d6af3 100644
--- a/tests/test_data_components.py
+++ b/tests/test_data_components.py
@@ -6,10 +6,9 @@ import httpx
import pytest
import respx
from httpx import Response
-
from langflow.components import (
data,
-) # Adjust the import according to your project structure
+)
@pytest.fixture
@@ -135,7 +134,7 @@ def test_directory_without_mocks():
from langflow.initial_setup import setup
from langflow.initial_setup.setup import load_starter_projects
- projects = load_starter_projects()
+ _, projects = zip(*load_starter_projects())
# the setup module has a folder where the projects are stored
# the contents of that folder are in the projects variable
# the directory component can be used to load the projects
diff --git a/tests/test_endpoints.py b/tests/test_endpoints.py
index 1430682ae..f5da2c0fc 100644
--- a/tests/test_endpoints.py
+++ b/tests/test_endpoints.py
@@ -472,11 +472,11 @@ def test_successful_run_with_output_type_text(client, starter_project, created_a
assert isinstance(outputs_dict.get("outputs"), list)
assert len(outputs_dict.get("outputs")) == 1
ids = [output.get("component_id") for output in outputs_dict.get("outputs")]
- assert all(["TextOutput" in _id for _id in ids]), ids
+ assert all(["ChatOutput" in _id for _id in ids]), ids
display_names = [output.get("component_display_name") for output in outputs_dict.get("outputs")]
- assert all([name in display_names for name in ["Prompt Output"]]), display_names
+ assert all([name in display_names for name in ["Chat Output"]]), display_names
inner_results = [output.get("results").get("result") for output in outputs_dict.get("outputs")]
- expected_result = "Langflow"
+ expected_result = ""
assert all([expected_result in result for result in inner_results]), inner_results
@@ -501,13 +501,13 @@ def test_successful_run_with_output_type_any(client, starter_project, created_ap
assert "outputs" in outputs_dict
assert outputs_dict.get("inputs") == {"input_value": ""}
assert isinstance(outputs_dict.get("outputs"), list)
- assert len(outputs_dict.get("outputs")) == 2
+ assert len(outputs_dict.get("outputs")) == 1
ids = [output.get("component_id") for output in outputs_dict.get("outputs")]
assert all(["ChatOutput" in _id or "TextOutput" in _id for _id in ids]), ids
display_names = [output.get("component_display_name") for output in outputs_dict.get("outputs")]
- assert all([name in display_names for name in ["Chat Output", "Prompt Output"]]), display_names
+ assert all([name in display_names for name in ["Chat Output"]]), display_names
inner_results = [output.get("results").get("result") for output in outputs_dict.get("outputs")]
- expected_result = "Langflow"
+ expected_result = ""
assert all([expected_result in result for result in inner_results]), inner_results
@@ -533,7 +533,7 @@ def test_successful_run_with_output_type_debug(client, starter_project, created_
assert "outputs" in outputs_dict
assert outputs_dict.get("inputs") == {"input_value": ""}
assert isinstance(outputs_dict.get("outputs"), list)
- assert len(outputs_dict.get("outputs")) == 7
+ assert len(outputs_dict.get("outputs")) == 4
# To test input_type wel'l just set it with output_type debug and check if the value is correct
@@ -559,10 +559,10 @@ def test_successful_run_with_input_type_text(client, starter_project, created_ap
assert "outputs" in outputs_dict
assert outputs_dict.get("inputs") == {"input_value": "value1"}
assert isinstance(outputs_dict.get("outputs"), list)
- assert len(outputs_dict.get("outputs")) == 7
+ assert len(outputs_dict.get("outputs")) == 4
# Now we get all components that contain TextInput in the component_id
text_input_outputs = [output for output in outputs_dict.get("outputs") if "TextInput" in output.get("component_id")]
- assert len(text_input_outputs) == 2
+ assert len(text_input_outputs) == 0
# Now we check if the input_value is correct
assert all([output.get("results").get("result") == "value1" for output in text_input_outputs]), text_input_outputs
@@ -590,7 +590,7 @@ def test_successful_run_with_input_type_chat(client, starter_project, created_ap
assert "outputs" in outputs_dict
assert outputs_dict.get("inputs") == {"input_value": "value1"}
assert isinstance(outputs_dict.get("outputs"), list)
- assert len(outputs_dict.get("outputs")) == 7
+ assert len(outputs_dict.get("outputs")) == 4
# Now we get all components that contain TextInput in the component_id
chat_input_outputs = [output for output in outputs_dict.get("outputs") if "ChatInput" in output.get("component_id")]
assert len(chat_input_outputs) == 1
@@ -620,14 +620,14 @@ def test_successful_run_with_input_type_any(client, starter_project, created_api
assert "outputs" in outputs_dict
assert outputs_dict.get("inputs") == {"input_value": "value1"}
assert isinstance(outputs_dict.get("outputs"), list)
- assert len(outputs_dict.get("outputs")) == 7
+ assert len(outputs_dict.get("outputs")) == 4
# Now we get all components that contain TextInput or ChatInput in the component_id
any_input_outputs = [
output
for output in outputs_dict.get("outputs")
if "TextInput" in output.get("component_id") or "ChatInput" in output.get("component_id")
]
- assert len(any_input_outputs) == 3
+ assert len(any_input_outputs) == 1
# Now we check if the input_value is correct
assert all([output.get("results").get("result") == "value1" for output in any_input_outputs]), any_input_outputs
diff --git a/tests/test_graph.py b/tests/test_graph.py
index 8395c304f..f67c12846 100644
--- a/tests/test_graph.py
+++ b/tests/test_graph.py
@@ -409,7 +409,7 @@ def test_update_source_handle():
@pytest.mark.asyncio
async def test_pickle_graph(json_vector_store):
starter_projects = load_starter_projects()
- data = starter_projects[0]["data"]
+ data = starter_projects[0][1]["data"]
graph = Graph.from_payload(data)
assert isinstance(graph, Graph)
pickled = pickle.dumps(graph)
@@ -421,7 +421,7 @@ async def test_pickle_graph(json_vector_store):
@pytest.mark.asyncio
async def test_pickle_each_vertex(json_vector_store):
starter_projects = load_starter_projects()
- data = starter_projects[0]["data"]
+ data = starter_projects[0][1]["data"]
graph = Graph.from_payload(data)
assert isinstance(graph, Graph)
for vertex in graph.vertices:
@@ -430,15 +430,3 @@ async def test_pickle_each_vertex(json_vector_store):
assert pickled is not None
unpickled = pickle.loads(pickled)
assert unpickled is not None
-
-
-@pytest.mark.asyncio
-async def test_build_ordering(complex_graph_with_groups):
- sorted_vertices = complex_graph_with_groups.sort_vertices(stop_component_id="ChatInput-Ay8QQ")
- assert sorted_vertices == [
- "ChatInput-Ay8QQ",
- "RecordsAsText-vkx2A",
- "FileLoader-Vo1Cq",
- ]
-
- sorted_vertices = complex_graph_with_groups.sort_vertices()
diff --git a/tests/test_initial_setup.py b/tests/test_initial_setup.py
index 33aaf8df3..c506c3275 100644
--- a/tests/test_initial_setup.py
+++ b/tests/test_initial_setup.py
@@ -1,4 +1,5 @@
from datetime import datetime
+from pathlib import Path
import pytest
from langflow.graph.graph.base import Graph
@@ -10,6 +11,7 @@ from langflow.initial_setup.setup import (
load_starter_projects,
)
from langflow.memory import delete_messages
+from langflow.processing.process import process_tweaks
from langflow.services.database.models.flow.model import Flow
from langflow.services.deps import session_scope
from sqlalchemy import func
@@ -19,7 +21,8 @@ from sqlmodel import select
def test_load_starter_projects():
projects = load_starter_projects()
assert isinstance(projects, list)
- assert all(isinstance(project, dict) for project in projects)
+ assert all(isinstance(project[1], dict) for project in projects)
+ assert all(isinstance(project[0], Path) for project in projects)
def test_get_project_data():
@@ -59,7 +62,7 @@ def test_create_or_update_starter_projects(client):
@pytest.mark.asyncio
-async def test_starter_project_can_run_successfully(client):
+async def test_starter_projects_can_run_successfully(client):
with session_scope() as session:
# Run the function to create or update projects
create_or_update_starter_projects()
@@ -75,12 +78,13 @@ async def test_starter_project_can_run_successfully(client):
# Get all the starter projects
projects = session.exec(select(Flow).where(Flow.folder == STARTER_FOLDER_NAME)).all()
-
- graphs: list[tuple[str, Graph]] = [
- (project.name, Graph.from_payload(project.data, flow_id=project.id))
- for project in projects
- if "Document" not in project.name or "RAG" not in project.name
- ]
+ graphs: list[tuple[str, Graph]] = []
+ for project in projects:
+ # Add tweaks to make file_path work
+ tweaks = {"path": __file__}
+ graph_data = process_tweaks(project.data, tweaks)
+ graph_object = Graph.from_payload(graph_data, flow_id=project.id)
+ graphs.append((project.name, graph_object))
assert len(graphs) == len(projects)
for name, graph in graphs:
outputs = await graph.arun(
diff --git a/tests/test_process.py b/tests/test_process.py
index 2548e9215..64404d9dd 100644
--- a/tests/test_process.py
+++ b/tests/test_process.py
@@ -1,5 +1,4 @@
import pytest
-
from langflow.processing.process import process_tweaks
from langflow.services.deps import get_session_service
@@ -284,12 +283,12 @@ async def test_load_langchain_object_with_no_cached_session(client, basic_graph_
session_id = session_service.build_key(session_id1, basic_graph_data)
graph1, artifacts1 = await session_service.load_session(session_id, data_graph=basic_graph_data, flow_id="flow_id")
# Clear the cache
- session_service.clear_session(session_id)
- # Use the new session_id to get the langchain_object again
+ await session_service.clear_session(session_id)
+ # Use the new session_id to get the graph again
graph2, artifacts2 = await session_service.load_session(session_id, data_graph=basic_graph_data, flow_id="flow_id")
- assert id(graph1) != id(graph2)
# Since the cache was cleared, objects should be different
+ assert id(graph1) != id(graph2)
@pytest.mark.asyncio
