feat: migrated old chat to new chat

surfsense_backend/app/tasks/chat/stream_connector_search_results.py

@@ -1,75 +0,0 @@
-from collections.abc import AsyncGenerator
-from typing import Any
-from uuid import UUID
-
-from sqlalchemy.ext.asyncio import AsyncSession
-
-from app.agents.researcher.graph import graph as researcher_graph
-from app.agents.researcher.state import State
-from app.services.streaming_service import StreamingService
-
-
-async def stream_connector_search_results(
-    user_query: str,
-    user_id: str | UUID,
-    search_space_id: int,
-    session: AsyncSession,
-    research_mode: str,
-    selected_connectors: list[str],
-    langchain_chat_history: list[Any],
-    document_ids_to_add_in_context: list[int],
-    language: str | None = None,
-    top_k: int = 10,
-) -> AsyncGenerator[str, None]:
-    """
-    Stream connector search results to the client
-
-    Args:
-        user_query: The user's query
-        user_id: The user's ID (can be UUID object or string)
-        search_space_id: The search space ID
-        session: The database session
-        research_mode: The research mode
-        selected_connectors: List of selected connectors
-
-    Yields:
-        str: Formatted response strings
-    """
-    streaming_service = StreamingService()
-
-    # Convert UUID to string if needed
-    user_id_str = str(user_id) if isinstance(user_id, UUID) else user_id
-
-    # Sample configuration
-    config = {
-        "configurable": {
-            "user_query": user_query,
-            "connectors_to_search": selected_connectors,
-            "user_id": user_id_str,
-            "search_space_id": search_space_id,
-            "document_ids_to_add_in_context": document_ids_to_add_in_context,
-            "language": language,  # Add language to the configuration
-            "top_k": top_k,  # Add top_k to the configuration
-        }
-    }
-    # print(f"Researcher configuration: {config['configurable']}")  # Debug print
-    # Initialize state with database session and streaming service
-    initial_state = State(
-        db_session=session,
-        streaming_service=streaming_service,
-        chat_history=langchain_chat_history,
-    )
-
-    # Run the graph directly
-    print("\nRunning the complete researcher workflow...")
-
-    # Use streaming with config parameter
-    async for chunk in researcher_graph.astream(
-        initial_state,
-        config=config,
-        stream_mode="custom",
-    ):
-        if isinstance(chunk, dict) and "yield_value" in chunk:
-            yield chunk["yield_value"]
-
-    yield streaming_service.format_completion()

surfsense_backend/app/tasks/chat/stream_new_chat.py

@@ -18,7 +18,7 @@ from app.agents.new_chat.llm_config import (
     create_chat_litellm_from_config,
     load_llm_config_from_yaml,
 )
-from app.schemas.chats import ChatMessage
+from app.schemas.new_chat import ChatMessage
 from app.services.connector_service import ConnectorService
 from app.services.new_streaming_service import VercelStreamingService