diff --git a/src/backend/langflow/components/utilities/GetRequest.py b/src/backend/langflow/components/utilities/GetRequest.py index c8182ad0d..13ff0dc23 100644 --- a/src/backend/langflow/components/utilities/GetRequest.py +++ b/src/backend/langflow/components/utilities/GetRequest.py @@ -1,6 +1,6 @@ from langflow import CustomComponent from langchain.schema import Document -from langflow.database.models.base import orjson_dumps +from langflow.services.database.models.base import orjson_dumps import requests from typing import Optional diff --git a/src/backend/langflow/components/utilities/JSONDocumentBuilder.py b/src/backend/langflow/components/utilities/JSONDocumentBuilder.py index e41b6b6af..26a2afd94 100644 --- a/src/backend/langflow/components/utilities/JSONDocumentBuilder.py +++ b/src/backend/langflow/components/utilities/JSONDocumentBuilder.py @@ -13,7 +13,7 @@ from langflow import CustomComponent from langchain.schema import Document -from langflow.database.models.base import orjson_dumps +from langflow.services.database.models.base import orjson_dumps class JSONDocumentBuilder(CustomComponent): diff --git a/src/backend/langflow/components/utilities/PostRequest.py b/src/backend/langflow/components/utilities/PostRequest.py index 9d5ab504f..4f7100d43 100644 --- a/src/backend/langflow/components/utilities/PostRequest.py +++ b/src/backend/langflow/components/utilities/PostRequest.py @@ -1,6 +1,6 @@ from langflow import CustomComponent from langchain.schema import Document -from langflow.database.models.base import orjson_dumps +from langflow.services.database.models.base import orjson_dumps import requests from typing import Optional diff --git a/src/backend/langflow/components/utilities/UpdateRequest.py b/src/backend/langflow/components/utilities/UpdateRequest.py index 569dac013..6e8991794 100644 --- a/src/backend/langflow/components/utilities/UpdateRequest.py +++ b/src/backend/langflow/components/utilities/UpdateRequest.py @@ -2,7 +2,7 @@ from typing import List, Optional import requests from langflow import CustomComponent from langchain.schema import Document -from langflow.database.models.base import orjson_dumps +from langflow.services.database.models.base import orjson_dumps class UpdateRequest(CustomComponent): diff --git a/src/backend/langflow/interface/initialize/utils.py b/src/backend/langflow/interface/initialize/utils.py index 116673645..199626de5 100644 --- a/src/backend/langflow/interface/initialize/utils.py +++ b/src/backend/langflow/interface/initialize/utils.py @@ -1,6 +1,6 @@ import contextlib import json -from langflow.database.models.base import orjson_dumps +from langflow.services.database.models.base import orjson_dumps import orjson from typing import Any, Dict, List diff --git a/src/backend/langflow/services/cache/utils.py b/src/backend/langflow/services/cache/utils.py index e89022167..a36243b75 100644 --- a/src/backend/langflow/services/cache/utils.py +++ b/src/backend/langflow/services/cache/utils.py @@ -8,7 +8,7 @@ from collections import OrderedDict from pathlib import Path from typing import Any, Dict from appdirs import user_cache_dir -from langflow.database.models.base import orjson_dumps +from langflow.services.database.models.base import orjson_dumps CACHE: Dict[str, Any] = {} diff --git a/src/backend/langflow/template/frontend_node/llms.py b/src/backend/langflow/template/frontend_node/llms.py index fdf0a2b5b..01098724e 100644 --- a/src/backend/langflow/template/frontend_node/llms.py +++ b/src/backend/langflow/template/frontend_node/llms.py @@ -1,5 +1,5 @@ from typing import Optional -from langflow.database.models.base import orjson_dumps +from langflow.services.database.models.base import orjson_dumps from langflow.template.field.base import TemplateField from langflow.template.frontend_node.base import FrontendNode diff --git a/src/backend/langflow/template/frontend_node/utilities.py b/src/backend/langflow/template/frontend_node/utilities.py index fa0b55332..9dedacd0f 100644 --- a/src/backend/langflow/template/frontend_node/utilities.py +++ b/src/backend/langflow/template/frontend_node/utilities.py @@ -1,6 +1,6 @@ import ast from typing import Optional -from langflow.database.models.base import orjson_dumps +from langflow.services.database.models.base import orjson_dumps from langflow.template.field.base import TemplateField from langflow.template.frontend_node.base import FrontendNode diff --git a/tests/test_cache.py b/tests/test_cache.py index f3d0cabda..edf205a05 100644 --- a/tests/test_cache.py +++ b/tests/test_cache.py @@ -1,5 +1,5 @@ import json -from langflow.database.models.base import orjson_dumps +from langflow.services.database.models.base import orjson_dumps import orjson from langflow.graph import Graph diff --git a/tests/test_database.py b/tests/test_database.py index d23d67a36..6976f963a 100644 --- a/tests/test_database.py +++ b/tests/test_database.py @@ -1,4 +1,4 @@ -from langflow.database.models.base import orjson_dumps +from langflow.services.database.models.base import orjson_dumps import orjson import pytest