KevinHuSh commited on
Commit
0c30cc9
·
1 Parent(s): 0e1a16c

refine readme (#170)

Browse files
README.md CHANGED
@@ -47,19 +47,19 @@
47
 
48
  ## 🤺RagFlow vs. other RAG applications
49
 
50
- | Feature | RagFlow | Langchain-Chatchat | Dify.AI | Assistants API | QAnythig | LangChain |
51
- |---------|:---------:|:----------------:|:-----------:|:-----------:|:-----------:|:-----------:|
52
- | **Well-Founded Answer** | :white_check_mark: | :x: | :x: | :x: | :x: | :x: |
53
- | **Trackable Chunking** | :white_check_mark: | :x: | :x: | :x: | :x: | :x: |
54
- | **Chunking Method** | Rich Variety | Naive | Naive | Naive | Naive | Naive |
55
- | **Table Structure Recognition** | :white_check_mark: | :x: | :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 | API-oriented | Python Code-oriented |
58
- | **RAG Engine** | :white_check_mark: | :white_check_mark: | :white_check_mark: | :white_check_mark: | :x: | :x: |
59
- | **Prompt IDE** | :white_check_mark: | :white_check_mark: | :white_check_mark: | :white_check_mark: | :x: | :x: |
60
- | **Supported LLMs** | Rich Variety | Rich Variety | Rich Variety | OpenAI-only | QwenLLM | Rich Variety |
61
- | **Local Deployment** | :white_check_mark: | :white_check_mark: | :white_check_mark: | :x: | :x: | :x: |
62
- | **Ecosystem Strategy** | Open Source | Open Source | Open Source | Close Source | 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