Fix various issues and refactor code (#1671)

* Update setup_mode to "Sync" in AstraDBVectorStoreComponent

* Fix search_kwargs default value in PineconeSearch and QdrantSearch components

* Update Playwright test and ChatMessage component in chat_io.spec.ts and chatMessage/index.tsx

* Refactor test_pickle_each_vertex function in test_graph.py

* Refactor database service to improve performance and readability

src/backend/base/langflow/services/database/service.py

@@ -1,21 +1,22 @@
-from datetime import datetime
 import time
+from datetime import datetime
 from pathlib import Path
 from typing import TYPE_CHECKING
 
 import sqlalchemy as sa
 from alembic import command, util
 from alembic.config import Config
+from loguru import logger
+from sqlalchemy import inspect
+from sqlalchemy.exc import OperationalError
+from sqlmodel import Session, SQLModel, create_engine, select, text
+
 from langflow.services.base import Service
 from langflow.services.database import models  # noqa
 from langflow.services.database.models.user.crud import get_user_by_username
 from langflow.services.database.utils import Result, TableResults
 from langflow.services.deps import get_settings_service
 from langflow.services.utils import teardown_superuser
-from loguru import logger
-from sqlalchemy import inspect
-from sqlalchemy.exc import OperationalError
-from sqlmodel import Session, SQLModel, create_engine, select, text
 
 if TYPE_CHECKING:
     from sqlalchemy.engine import Engine
@@ -195,7 +196,10 @@ class DatabaseService(Service):
         # This method is used for testing purposes only
         # We will check that all models are in the database
         # and that the database is up to date with all columns
-        sql_models = [models.Flow, models.User, models.ApiKey]
+        # get all models that are subclasses of SQLModel
+        sql_models = [
+            model for model in models.__dict__.values() if isinstance(model, type) and issubclass(model, SQLModel)
+        ]
         return [TableResults(sql_model.__tablename__, self.check_table(sql_model)) for sql_model in sql_models]
 
     def check_table(self, model):

src/frontend/src/components/headerComponent/index.tsx

@@ -1,4 +1,4 @@
-import { useContext, useEffect } from "react";
+import { useContext } from "react";
 import { FaDiscord, FaGithub } from "react-icons/fa";
 import { RiTwitterXFill } from "react-icons/ri";
 import { Link, useLocation, useNavigate, useParams } from "react-router-dom";

src/frontend/src/modals/IOModal/components/chatView/chatMessage/index.tsx

@@ -164,7 +164,12 @@ export default function ChatMessage({
               {chat.thought && chat.thought !== "" && !hidden && <br></br>}
               <div className="flex w-full flex-col">
                 <div className="flex w-full flex-col dark:text-white">
-                  <div data-testid={"chat-message-"+chat.sender_name+"-"+chatMessage} className="flex w-full flex-col">
+                  <div
+                    data-testid={
+                      "chat-message-" + chat.sender_name + "-" + chatMessage
+                    }
+                    className="flex w-full flex-col"
+                  >
                     {useMemo(
                       () =>
                         chatMessage === "" && lockChat ? (
@@ -313,7 +318,13 @@ dark:prose-invert"
                 </span>
               </>
             ) : (
-              <span data-testid={"chat-message-"+chat.sender_name+"-"+chatMessage}>{chatMessage}</span>
+              <span
+                data-testid={
+                  "chat-message-" + chat.sender_name + "-" + chatMessage
+                }
+              >
+                {chatMessage}
+              </span>
             )}
           </div>
         )}

src/frontend/tests/end-to-end/chat_io.spec.ts

@@ -1,4 +1,4 @@
-import { test,expect } from "@playwright/test";
+import { expect, test } from "@playwright/test";
 import { readFileSync } from "fs";
 
 test("chat_io_teste", async ({ page }) => {