yangdx commited on
Commit
24a06f6
·
1 Parent(s): 938ef6c

解决模型名称返回错误问题

Browse files
Files changed (1) hide show
  1. lightrag/api/lightrag_ollama.py +2 -4
lightrag/api/lightrag_ollama.py CHANGED
@@ -258,7 +258,6 @@ class OllamaModelDetails(BaseModel):
258
  class OllamaModel(BaseModel):
259
  name: str
260
  model: str
261
- tag: str
262
  size: int
263
  digest: str
264
  modified_at: str
@@ -611,9 +610,8 @@ def create_app(args):
611
  """Get available models"""
612
  return OllamaTagResponse(
613
  models=[{
614
- "name": LIGHTRAG_NAME,
615
- "model": LIGHTRAG_NAME,
616
- "tag": LIGHTRAG_TAG,
617
  "size": LIGHTRAG_SIZE,
618
  "digest": LIGHTRAG_DIGEST,
619
  "modified_at": LIGHTRAG_CREATED_AT,
 
258
  class OllamaModel(BaseModel):
259
  name: str
260
  model: str
 
261
  size: int
262
  digest: str
263
  modified_at: str
 
610
  """Get available models"""
611
  return OllamaTagResponse(
612
  models=[{
613
+ "name": LIGHTRAG_MODEL,
614
+ "model": LIGHTRAG_MODEL,
 
615
  "size": LIGHTRAG_SIZE,
616
  "digest": LIGHTRAG_DIGEST,
617
  "modified_at": LIGHTRAG_CREATED_AT,