gzdaniel commited on
Commit
0366449
·
1 Parent(s): 68d3ef4

Enhance Ollama interface compatibility

Browse files
Files changed (1) hide show
  1. lightrag/api/routers/ollama_api.py +5 -0
lightrag/api/routers/ollama_api.py CHANGED
@@ -579,6 +579,11 @@ class OllamaAPI:
579
  data = {
580
  "model": self.ollama_server_infos.LIGHTRAG_MODEL,
581
  "created_at": self.ollama_server_infos.LIGHTRAG_CREATED_AT,
 
 
 
 
 
582
  "done": True,
583
  "total_duration": total_time,
584
  "load_duration": 0,
 
579
  data = {
580
  "model": self.ollama_server_infos.LIGHTRAG_MODEL,
581
  "created_at": self.ollama_server_infos.LIGHTRAG_CREATED_AT,
582
+ "message": {
583
+ "role": "assistant",
584
+ "content": "",
585
+ "images": None,
586
+ },
587
  "done": True,
588
  "total_duration": total_time,
589
  "load_duration": 0,