Fix typos and update method names

src/backend/langflow/api/v1/callback.py

@@ -1,19 +1,15 @@
 import asyncio
+from typing import Any, Dict, List, Optional
 from uuid import UUID
 
 from langchain.callbacks.base import AsyncCallbackHandler, BaseCallbackHandler
-
-from langflow.api.v1.schemas import ChatResponse, PromptResponse
-
-
-from typing import Any, Dict, List, Optional
-from langflow.services.deps import get_chat_service
-
-
-from langflow.utils.util import remove_ansi_escape_codes
 from langchain.schema import AgentAction, AgentFinish
 from loguru import logger
 
+from langflow.api.v1.schemas import ChatResponse, PromptResponse
+from langflow.services.deps import get_chat_service
+from langflow.utils.util import remove_ansi_escape_codes
+
 
 # https://github.com/hwchase17/chat-langchain/blob/master/callback.py
 class AsyncStreamingLLMCallbackHandler(AsyncCallbackHandler):
@@ -26,7 +22,7 @@ class AsyncStreamingLLMCallbackHandler(AsyncCallbackHandler):
 
     async def on_llm_new_token(self, token: str, **kwargs: Any) -> None:
         resp = ChatResponse(message=token, type="stream", intermediate_steps="")
-        await self.websocket.send_json(resp.dict())
+        await self.websocket.send_json(resp.model_dump())
 
     async def on_tool_start(self, serialized: Dict[str, Any], input_str: str, **kwargs: Any) -> Any:
         """Run when tool starts running."""
@@ -35,7 +31,7 @@ class AsyncStreamingLLMCallbackHandler(AsyncCallbackHandler):
             type="stream",
             intermediate_steps=f"Tool input: {input_str}",
         )
-        await self.websocket.send_json(resp.dict())
+        await self.websocket.send_json(resp.model_dump())
 
     async def on_tool_end(self, output: str, **kwargs: Any) -> Any:
         """Run when tool ends running."""
@@ -66,7 +62,7 @@ class AsyncStreamingLLMCallbackHandler(AsyncCallbackHandler):
         try:
             # This is to emulate the stream of tokens
             for resp in resps:
-                await self.websocket.send_json(resp.dict())
+                await self.websocket.send_json(resp.model_dump())
         except Exception as exc:
             logger.error(f"Error sending response: {exc}")
 
@@ -92,7 +88,7 @@ class AsyncStreamingLLMCallbackHandler(AsyncCallbackHandler):
             resp = PromptResponse(
                 prompt=text,
             )
-            await self.websocket.send_json(resp.dict())
+            await self.websocket.send_json(resp.model_dump())
             self.chat_service.chat_history.add_message(self.client_id, resp)
 
     async def on_agent_action(self, action: AgentAction, **kwargs: Any):
@@ -103,10 +99,10 @@ class AsyncStreamingLLMCallbackHandler(AsyncCallbackHandler):
             logs = log.split("\n")
             for log in logs:
                 resp = ChatResponse(message="", type="stream", intermediate_steps=log)
-                await self.websocket.send_json(resp.dict())
+                await self.websocket.send_json(resp.model_dump())
         else:
             resp = ChatResponse(message="", type="stream", intermediate_steps=log)
-            await self.websocket.send_json(resp.dict())
+            await self.websocket.send_json(resp.model_dump())
 
     async def on_agent_finish(self, finish: AgentFinish, **kwargs: Any) -> Any:
         """Run on agent end."""
@@ -115,7 +111,7 @@ class AsyncStreamingLLMCallbackHandler(AsyncCallbackHandler):
             type="stream",
             intermediate_steps=finish.log,
         )
-        await self.websocket.send_json(resp.dict())
+        await self.websocket.send_json(resp.model_dump())
 
 
 class StreamingLLMCallbackHandler(BaseCallbackHandler):
@@ -130,5 +126,5 @@ class StreamingLLMCallbackHandler(BaseCallbackHandler):
         resp = ChatResponse(message=token, type="stream", intermediate_steps="")
 
         loop = asyncio.get_event_loop()
-        coroutine = self.websocket.send_json(resp.dict())
+        coroutine = self.websocket.send_json(resp.model_dump())
         asyncio.run_coroutine_threadsafe(coroutine, loop)

src/backend/langflow/interface/custom/code_parser.py

@@ -1,8 +1,8 @@
 import ast
 import inspect
 import traceback
+from typing import Any, Dict, List, Type, Union
 
-from typing import Dict, Any, List, Type, Union
 from fastapi import HTTPException
 from langflow.interface.custom.schema import CallableCodeDetails, ClassCodeDetails
 
@@ -104,7 +104,7 @@ class CodeParser:
         func.args = self.parse_function_args(node)
         func.body = self.parse_function_body(node)
 
-        return func.dict()
+        return func.model_dump()
 
     def parse_function_args(self, node: ast.FunctionDef) -> List[Dict[str, Any]]:
         """
@@ -233,7 +233,7 @@ class CodeParser:
                 else:
                     class_details.methods.append(method)
 
-        self.data["classes"].append(class_details.dict())
+        self.data["classes"].append(class_details.model_dump())
 
     def parse_global_vars(self, node: ast.Assign) -> None:
         """

src/backend/langflow/services/chat/service.py

@@ -1,20 +1,20 @@
-from collections import defaultdict
+import asyncio
 import uuid
+from collections import defaultdict
+from typing import Any, Dict, List
+
+import orjson
 from fastapi import WebSocket, status
-from starlette.websockets import WebSocketState
 from langflow.api.v1.schemas import ChatMessage, ChatResponse, FileResponse
 from langflow.interface.utils import pil_to_base64
+from langflow.services import ServiceType, service_manager
 from langflow.services.base import Service
 from langflow.services.chat.cache import Subject
 from langflow.services.chat.utils import process_graph
 from loguru import logger
+from starlette.websockets import WebSocketState
 
 from .cache import cache_service
-import asyncio
-from typing import Any, Dict, List
-
-from langflow.services import service_manager, ServiceType
-import orjson
 
 
 class ChatHistory(Subject):
@@ -104,7 +104,7 @@ class ChatService(Service):
 
     async def send_json(self, client_id: str, message: ChatMessage):
         websocket = self.active_connections[client_id]
-        await websocket.send_json(message.dict())
+        await websocket.send_json(message.model_dump())
 
     async def close_connection(self, client_id: str, code: int, reason: str):
         if websocket := self.active_connections[client_id]:
@@ -191,7 +191,7 @@ class ChatService(Service):
         try:
             chat_history = self.chat_history.get_history(client_id)
             # iterate and make BaseModel into dict
-            chat_history = [chat.dict() for chat in chat_history]
+            chat_history = [chat.model_dump() for chat in chat_history]
             await websocket.send_json(chat_history)
 
             while True:

src/backend/langflow/services/settings/base.py

@@ -205,7 +205,7 @@ class Settings(BaseSettings):
 
 def save_settings_to_yaml(settings: Settings, file_path: str):
     with open(file_path, "w") as f:
-        settings_dict = settings.dict()
+        settings_dict = settings.model_dump()
         yaml.dump(settings_dict, f)
 
 

src/backend/langflow/settings.py

@@ -1,12 +1,13 @@
 import contextlib
 import json
 import os
-from typing import Optional, List
 from pathlib import Path
+from typing import List, Optional
 
 import yaml
-from pydantic import validator, model_validator
+from pydantic import model_validator, validator
 from pydantic_settings import BaseSettings
+
 from langflow.utils.logger import logger
 
 BASE_COMPONENTS_PATH = str(Path(__file__).parent / "components")
@@ -141,7 +142,7 @@ class Settings(BaseSettings):
 
 def save_settings_to_yaml(settings: Settings, file_path: str):
     with open(file_path, "w") as f:
-        settings_dict = settings.dict()
+        settings_dict = settings.model_dump()
         yaml.dump(settings_dict, f)
 
 

src/backend/langflow/template/field/base.py

@@ -60,7 +60,7 @@ class TemplateFieldCreator(BaseModel, ABC):
     """Additional information about the field to be shown in the tooltip. Defaults to an empty string."""
 
     def to_dict(self):
-        result = self.dict()
+        result = self.model_dump()
         # Remove key if it is None
         for key in list(result.keys()):
             if result[key] is None or result[key] == []:

src/backend/langflow/worker.py

@@ -1,15 +1,15 @@
-from langflow.core.celery_app import celery_app
-from typing import Any, Dict, Optional
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, Dict, Optional
 
 from celery.exceptions import SoftTimeLimitExceeded  # type: ignore
+
+from langflow.core.celery_app import celery_app
 from langflow.processing.process import (
     Result,
     generate_result,
     process_inputs,
 )
-from langflow.services.manager import initialize_session_service
 from langflow.services.deps import get_session_service
+from langflow.services.manager import initialize_session_service
 
 if TYPE_CHECKING:
     from langflow.graph.vertex.base import Vertex
@@ -55,4 +55,4 @@ def process_graph_cached_task(
     # we need to update the cache with the updated langchain_object
     session_service.update_session(session_id, (graph, artifacts))
 
-    return Result(result=result, session_id=session_id).dict()
+    return Result(result=result, session_id=session_id).model_dump()