From c1f4f4a20e1e5474fb56bb1443a0664881fda995 Mon Sep 17 00:00:00 2001 From: yangdx Date: Wed, 15 Jan 2025 15:06:28 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96ollama=E8=BF=94=E5=9B=9E?= =?UTF-8?q?=E6=95=B0=E6=8D=AE=E7=9A=84=E6=A0=BC=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- lightrag/api/lightrag_ollama.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/lightrag/api/lightrag_ollama.py b/lightrag/api/lightrag_ollama.py index 6f1ec9a4..e03c91bd 100644 --- a/lightrag/api/lightrag_ollama.py +++ b/lightrag/api/lightrag_ollama.py @@ -27,6 +27,7 @@ load_dotenv() LIGHTRAG_NAME = "lightrag" LIGHTRAG_TAG = "latest" LIGHTRAG_MODEL = "{LIGHTRAG_NAME}:{LIGHTRAG_TAG}" +LIGHTRAG_SIZE = 7365960935 LIGHTRAG_CREATED_AT = "2024-01-15T00:00:00Z" LIGHTRAG_DIGEST = "sha256:lightrag" @@ -245,7 +246,6 @@ class OllamaChatResponse(BaseModel): class OllamaVersionResponse(BaseModel): version: str - build: str = "default" class OllamaTagResponse(BaseModel): models: List[Dict[str, str]] @@ -586,7 +586,7 @@ def create_app(args): async def get_version(): """Get Ollama version information""" return OllamaVersionResponse( - version="0.1.0" + version="0.5.4" ) @app.get("/api/tags") @@ -595,10 +595,19 @@ def create_app(args): return OllamaTagResponse( models=[{ "name": LIGHTRAG_NAME, + "model": LIGHTRAG_NAME, "tag": LIGHTRAG_TAG, - "size": 0, + "size": LIGHTRAG_SIZE, "digest": LIGHTRAG_DIGEST, - "modified_at": LIGHTRAG_CREATED_AT + "modified_at": LIGHTRAG_CREATED_AT, + "details": { + "parent_model": "", + "format": "gguf", + "family": LIGHTRAG_NAME, + "families": [LIGHTRAG_NAME], + "parameter_size": "13B", + "quantization_level": "Q4_0" + } }] )