gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化ollama返回数据的格式

Browse Source
This commit is contained in:
yangdx
2025-01-15 15:06:28 +08:00
parent be134878fe
commit c1f4f4a20e
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
lightrag/api/lightrag_ollama.py
View File

@@ -27,6 +27,7 @@ load_dotenv()
LIGHTRAG_NAME = "lightrag" LIGHTRAG_NAME = "lightrag"
LIGHTRAG_TAG = "latest" LIGHTRAG_TAG = "latest"
LIGHTRAG_MODEL = "{LIGHTRAG_NAME}:{LIGHTRAG_TAG}" LIGHTRAG_MODEL = "{LIGHTRAG_NAME}:{LIGHTRAG_TAG}"
LIGHTRAG_SIZE = 7365960935
LIGHTRAG_CREATED_AT = "2024-01-15T00:00:00Z" LIGHTRAG_CREATED_AT = "2024-01-15T00:00:00Z"
LIGHTRAG_DIGEST = "sha256:lightrag" LIGHTRAG_DIGEST = "sha256:lightrag"
@@ -245,7 +246,6 @@ class OllamaChatResponse(BaseModel):
class OllamaVersionResponse(BaseModel): class OllamaVersionResponse(BaseModel):
version: str version: str
build: str = "default"
class OllamaTagResponse(BaseModel): class OllamaTagResponse(BaseModel):
models: List[Dict[str, str]] models: List[Dict[str, str]]
@@ -586,7 +586,7 @@ def create_app(args):
async def get_version(): async def get_version():
"""Get Ollama version information""" """Get Ollama version information"""
return OllamaVersionResponse( return OllamaVersionResponse(
version="0.1.0" version="0.5.4"
) )
@app.get("/api/tags") @app.get("/api/tags")
@@ -595,10 +595,19 @@ def create_app(args):
return OllamaTagResponse( return OllamaTagResponse(
models=[{ models=[{
"name": LIGHTRAG_NAME, "name": LIGHTRAG_NAME,
"model": LIGHTRAG_NAME,
"tag": LIGHTRAG_TAG, "tag": LIGHTRAG_TAG,
"size": 0, "size": LIGHTRAG_SIZE,
"digest": LIGHTRAG_DIGEST, "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"
}
}] }]
) )