Fix orjson_dumps import (#814)

.gitattributes

@@ -15,6 +15,10 @@
 *.svg text
 *.yml text
 *.yaml text
+*.xml text
+*.csv text
+*.json text
+*.sh text
 *.Dockerfile text
 Dockerfile text
 

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
 

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):

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
 

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):

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
 

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] = {}
 

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

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

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
 

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