gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
yangdx 2025-01-15 18:19:39 +08:00
parent fd50c3a240
commit 882da88603
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
lightrag/api/lightrag_ollama.py
View file

@ -258,7 +258,6 @@ class OllamaModelDetails(BaseModel):
class OllamaModel(BaseModel): class OllamaModel(BaseModel):
name: str name: str
model: str model: str
tag: str
size: int size: int
digest: str digest: str
modified_at: str modified_at: str
@ -611,9 +610,8 @@ def create_app(args):
"""Get available models""" """Get available models"""
return OllamaTagResponse( return OllamaTagResponse(
models=[{ models=[{
"name": LIGHTRAG_NAME, "name": LIGHTRAG_MODEL,
"model": LIGHTRAG_NAME, "model": LIGHTRAG_MODEL,
"tag": LIGHTRAG_TAG,
"size": LIGHTRAG_SIZE, "size": LIGHTRAG_SIZE,
"digest": LIGHTRAG_DIGEST, "digest": LIGHTRAG_DIGEST,
"modified_at": LIGHTRAG_CREATED_AT, "modified_at": LIGHTRAG_CREATED_AT,