yangdx
commited on
Commit
·
24a06f6
1
Parent(s):
938ef6c
解决模型名称返回错误问题
Browse files
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":
|
615 |
-
"model":
|
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,
|