Don't use answer_with_citations() for local models

graph.py

@@ -132,7 +132,7 @@ def BuildGraph(
     top_k=6,
     think_query=False,
     think_answer=False,
-    local_citations=True,
+    local_citations=False,
     embedding_ckpt_dir=None,
 ):
     """
@@ -258,7 +258,7 @@ def BuildGraph(
         if is_local:
             # Don't include the system message here because it's defined in ToolCallingLLM
             messages = state["messages"]
-            messages = normalize_messages(messages, "query")
+            messages = normalize_messages(messages)
         else:
             messages = [SystemMessage(query_prompt(chat_model))] + state["messages"]
         response = query_model.invoke(messages)
@@ -269,7 +269,7 @@ def BuildGraph(
         """Generates an answer with the chat model"""
         if is_local:
             messages = state["messages"]
-            messages = normalize_messages(messages, "answer")
+            messages = normalize_messages(messages)
             if not local_citations:
                 # Add the system message here if we're not using tools
                 messages = [