Enhance Ollama interface compatibility

2025-06-11 15:16:32 +08:00 · 2025-06-11 15:16:32 +08:00 · 9351b09cc7
commit 9351b09cc7
parent 7b07d4c917
1 changed files with 5 additions and 0 deletions
--- a/lightrag/api/routers/ollama_api.py
+++ b/lightrag/api/routers/ollama_api.py
@ -579,6 +579,11 @@ class OllamaAPI:
                                data = {
                                    "model": self.ollama_server_infos.LIGHTRAG_MODEL,
                                    "created_at": self.ollama_server_infos.LIGHTRAG_CREATED_AT,
+                                    "message": {
+                                        "role": "assistant",
+                                        "content": "",
+                                        "images": None,
+                                    },
                                    "done": True,
                                    "total_duration": total_time,
                                    "load_duration": 0,