Add util functions to renamed chat_source_history.py

app.py

@@ -10,7 +10,6 @@ from langgraph.graph import MessagesState, StateGraph, END
 from langgraph.prebuilt import ToolNode, tools_condition
 
 import structlog
-import uuid
 
 from qdrant_client import QdrantClient
 from qdrant_client.http.models import Distance, VectorParams, SparseVectorParams
@@ -18,7 +17,7 @@ from qdrant_client.http.models import Distance, VectorParams, SparseVectorParams
 import logging_config as _
 # from conversation.main import graph
 from conversation.generate import generate, trigger_ai_message_with_tool_call
-from conversation.source_history import prettify_source_history, build_source_history_object
+from conversation.chat_source_history import prettify_source_history, build_source_history_object, clear_chat_and_source_history, create_chat_hash
 from ingestion.main import ingest_document
 from tools.langfuse_client import get_langfuse_handler
 
@@ -124,19 +123,6 @@ def bot(message, history, source_history, chat_hash) -> list[Any]:
 
     return [response["messages"][-1].content], source_history, chat_hash
 
-def clear_chat_and_source_history(source_history, chat_hash):
-    # Clear the source history
-    source_history = []
-
-    # Return a new chat hash
-    chat_hash = create_chat_hash()
-
-    return source_history, chat_hash
-
-def create_chat_hash():
-    chat_hash = str(uuid.uuid4())
-    logger.info("This is the chat_hash when it is called", chat_hash=chat_hash)
-    return chat_hash
 
 with gr.Blocks() as demo:
     # Initialize source history as session state variable

conversation/{source_history.py → chat_source_history.py}

@@ -1,4 +1,5 @@
 import os
+import uuid
 import structlog
 from langchain_core.messages import ToolMessage
 
@@ -109,4 +110,18 @@ def build_source_history_object(response, source_history: list, user_input_promp
         "source_text_chunk": [doc["source_text_chunk"] for doc in dictionary_document_data]
     })
 
-    return source_history
+    return source_history
+
+def clear_chat_and_source_history(source_history, chat_hash):
+    # Clear the source history
+    source_history = []
+
+    # Return a new chat hash
+    chat_hash = create_chat_hash()
+
+    return source_history, chat_hash
+
+def create_chat_hash():
+    chat_hash = str(uuid.uuid4())
+    logger.info("This is the chat_hash when it is called", chat_hash=chat_hash)
+    return chat_hash