KevinHuSh
commited on
Commit
·
0c30cc9
1
Parent(s):
0e1a16c
refine readme (#170)
Browse files- README.md +12 -12
- api/db/services/llm_service.py +2 -2
- rag/llm/embedding_model.py +1 -1
README.md
CHANGED
@@ -47,19 +47,19 @@
|
|
47 |
|
48 |
## 🤺RagFlow vs. other RAG applications
|
49 |
|
50 |
-
| Feature | RagFlow | Langchain-Chatchat |
|
51 |
-
|
52 |
-
| **Well-Founded Answer** | :white_check_mark: | :x: | :x: | :x: | :x: |
|
53 |
-
| **Trackable Chunking** | :white_check_mark: | :x: | :x: | :x: | :x: |
|
54 |
-
| **Chunking Method** | Rich Variety | Naive | Naive |
|
55 |
-
| **Table Structure Recognition** | :white_check_mark: | :x: |
|
56 |
| **Structured Data Lookup** | :white_check_mark: | :x: | :x: | :x: | :x: | :x: |
|
57 |
-
| **Programming Approach** | API-oriented | API-oriented | API-oriented | API-oriented |
|
58 |
-
| **RAG Engine** | :white_check_mark: | :white_check_mark: | :white_check_mark: | :
|
59 |
-
| **Prompt IDE** | :white_check_mark: | :white_check_mark: | :white_check_mark: | :
|
60 |
-
| **Supported LLMs** | Rich Variety | Rich Variety |
|
61 |
-
| **Local Deployment** | :white_check_mark: | :white_check_mark: | :
|
62 |
-
| **Ecosystem Strategy** | Open Source | Open Source |
|
63 |
|
64 |
## 🔎 System Architecture
|
65 |
|
|
|
47 |
|
48 |
## 🤺RagFlow vs. other RAG applications
|
49 |
|
50 |
+
| Feature | RagFlow | Langchain-Chatchat | Assistants API | QAnythig | LangChain |
|
51 |
+
|---------|:---------:|:----------------:|:-----------:|:-----------:|:-----------:|
|
52 |
+
| **Well-Founded Answer** | :white_check_mark: | :x: | :x: | :x: | :x: |
|
53 |
+
| **Trackable Chunking** | :white_check_mark: | :x: | :x: | :x: | :x: |
|
54 |
+
| **Chunking Method** | Rich Variety | Naive | Naive | | Naive | Naive |
|
55 |
+
| **Table Structure Recognition** | :white_check_mark: | :x: | | :x: | :x: | :x: |
|
56 |
| **Structured Data Lookup** | :white_check_mark: | :x: | :x: | :x: | :x: | :x: |
|
57 |
+
| **Programming Approach** | API-oriented | API-oriented | API-oriented | API-oriented | Python Code-oriented |
|
58 |
+
| **RAG Engine** | :white_check_mark: | :white_check_mark: | :white_check_mark: | :x: | :x: |
|
59 |
+
| **Prompt IDE** | :white_check_mark: | :white_check_mark: | :white_check_mark: | :x: | :x: |
|
60 |
+
| **Supported LLMs** | Rich Variety | Rich Variety | OpenAI-only | QwenLLM | Rich Variety |
|
61 |
+
| **Local Deployment** | :white_check_mark: | :white_check_mark: | :x: | :x: | :x: |
|
62 |
+
| **Ecosystem Strategy** | Open Source | Open Source | Close Source | Open Source | Open Source |
|
63 |
|
64 |
## 🔎 System Architecture
|
65 |
|
api/db/services/llm_service.py
CHANGED
@@ -84,7 +84,7 @@ class TenantLLMService(CommonService):
|
|
84 |
if model_config["llm_factory"] not in EmbeddingModel:
|
85 |
return
|
86 |
return EmbeddingModel[model_config["llm_factory"]](
|
87 |
-
model_config["api_key"], model_config["llm_name"], model_config["api_base"])
|
88 |
|
89 |
if llm_type == LLMType.IMAGE2TEXT.value:
|
90 |
if model_config["llm_factory"] not in CvModel:
|
@@ -98,7 +98,7 @@ class TenantLLMService(CommonService):
|
|
98 |
if model_config["llm_factory"] not in ChatModel:
|
99 |
return
|
100 |
return ChatModel[model_config["llm_factory"]](
|
101 |
-
model_config["api_key"], model_config["llm_name"], model_config["api_base"])
|
102 |
|
103 |
@classmethod
|
104 |
@DB.connection_context()
|
|
|
84 |
if model_config["llm_factory"] not in EmbeddingModel:
|
85 |
return
|
86 |
return EmbeddingModel[model_config["llm_factory"]](
|
87 |
+
model_config["api_key"], model_config["llm_name"], base_url=model_config["api_base"])
|
88 |
|
89 |
if llm_type == LLMType.IMAGE2TEXT.value:
|
90 |
if model_config["llm_factory"] not in CvModel:
|
|
|
98 |
if model_config["llm_factory"] not in ChatModel:
|
99 |
return
|
100 |
return ChatModel[model_config["llm_factory"]](
|
101 |
+
model_config["api_key"], model_config["llm_name"], base_url=model_config["api_base"])
|
102 |
|
103 |
@classmethod
|
104 |
@DB.connection_context()
|
rag/llm/embedding_model.py
CHANGED
@@ -51,7 +51,7 @@ class Base(ABC):
|
|
51 |
|
52 |
|
53 |
class HuEmbedding(Base):
|
54 |
-
def __init__(self, **kwargs):
|
55 |
"""
|
56 |
If you have trouble downloading HuggingFace models, -_^ this might help!!
|
57 |
|
|
|
51 |
|
52 |
|
53 |
class HuEmbedding(Base):
|
54 |
+
def __init__(self, *args, **kwargs):
|
55 |
"""
|
56 |
If you have trouble downloading HuggingFace models, -_^ this might help!!
|
57 |
|