Add huggingface model support

2024-10-15 19:40:08 +08:00
parent 2d0d25005b
commit 1e78068268
11 changed files with 100 additions and 56 deletions
--- a/lightrag/base.py
+++ b/lightrag/base.py
@@ -14,7 +14,7 @@ T = TypeVar("T")

@dataclass
 class QueryParam:
-    mode: Literal["local", "global", "hybird", "naive"] = "global"
+    mode: Literal["local", "global", "hybrid", "naive"] = "global"
    only_need_context: bool = False
    response_type: str = "Multiple Paragraphs"
    top_k: int = 60