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

2025-01-15 18:19:39 +08:00
parent fd50c3a240
commit 882da88603
1 changed files with 2 additions and 4 deletions
--- a/lightrag/api/lightrag_ollama.py
+++ b/lightrag/api/lightrag_ollama.py
@@ -258,7 +258,6 @@ class OllamaModelDetails(BaseModel):
 class OllamaModel(BaseModel):
    name: str
    model: str
-    tag: str
    size: int
    digest: str
    modified_at: str
@@ -611,9 +610,8 @@ def create_app(args):
        """Get available models"""
        return OllamaTagResponse(
            models=[{
-                "name": LIGHTRAG_NAME,
-                "model": LIGHTRAG_NAME,
-                "tag": LIGHTRAG_TAG,
+                "name": LIGHTRAG_MODEL,
+                "model": LIGHTRAG_MODEL,
                "size": LIGHTRAG_SIZE,
                "digest": LIGHTRAG_DIGEST,
                "modified_at": LIGHTRAG_CREATED_AT,