From 2719e07107db9f3f30b96d0776cdc8c828d78ece Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Sat, 25 Jan 2025 22:33:09 +0800
Subject: [PATCH] Convert OllamaMessage to dict in conversation history

---
 lightrag/api/lightrag_server.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py
index b94e0d17..e000dc0e 100644
--- a/lightrag/api/lightrag_server.py
+++ b/lightrag/api/lightrag_server.py
@@ -1585,7 +1585,10 @@ def create_app(args):
 
             # Get the last message as query and previous messages as history
             query = messages[-1].content
-            conversation_history = messages[:-1]  # 所有之前的消息作为历史记录
+            # Convert OllamaMessage objects to dictionaries
+            conversation_history = [
+                {"role": msg.role, "content": msg.content} for msg in messages[:-1]
+            ]
 
             # Check for query prefix
             cleaned_query, mode = parse_query_mode(query)