Spaces:
Paused
Paused
lanny xu
commited on
Commit
·
3141e61
1
Parent(s):
7a4bc96
delete vectara
Browse files- entity_extractor.py +3 -3
- evaluate_retrieval.py +3 -3
- graph_indexer.py +6 -6
- retrieval_evaluation.py +3 -3
- test_reranking.py +3 -3
- workflow_nodes.py +9 -9
entity_extractor.py
CHANGED
|
@@ -12,9 +12,9 @@ try:
|
|
| 12 |
from langchain_core.prompts import PromptTemplate
|
| 13 |
except ImportError:
|
| 14 |
try:
|
| 15 |
-
|
| 16 |
-
except ImportError:
|
| 17 |
-
|
| 18 |
|
| 19 |
from langchain_community.chat_models import ChatOllama
|
| 20 |
from langchain_core.output_parsers import JsonOutputParser
|
|
|
|
| 12 |
from langchain_core.prompts import PromptTemplate
|
| 13 |
except ImportError:
|
| 14 |
try:
|
| 15 |
+
from langchain_core.prompts import PromptTemplate
|
| 16 |
+
except ImportError:
|
| 17 |
+
from langchain.prompts import PromptTemplate
|
| 18 |
|
| 19 |
from langchain_community.chat_models import ChatOllama
|
| 20 |
from langchain_core.output_parsers import JsonOutputParser
|
evaluate_retrieval.py
CHANGED
|
@@ -22,9 +22,9 @@ try:
|
|
| 22 |
from langchain_core.documents import Document
|
| 23 |
except ImportError:
|
| 24 |
try:
|
| 25 |
-
|
| 26 |
-
except ImportError:
|
| 27 |
-
|
| 28 |
|
| 29 |
# 导入LangChain相关模块
|
| 30 |
from langchain_community.vectorstores import FAISS, Chroma
|
|
|
|
| 22 |
from langchain_core.documents import Document
|
| 23 |
except ImportError:
|
| 24 |
try:
|
| 25 |
+
from langchain_core.documents import Document
|
| 26 |
+
except ImportError:
|
| 27 |
+
from langchain.schema import Document
|
| 28 |
|
| 29 |
# 导入LangChain相关模块
|
| 30 |
from langchain_community.vectorstores import FAISS, Chroma
|
graph_indexer.py
CHANGED
|
@@ -9,12 +9,12 @@ try:
|
|
| 9 |
from langchain_core.documents import Document
|
| 10 |
except ImportError:
|
| 11 |
try:
|
| 12 |
-
|
| 13 |
-
except ImportError:
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
except ImportError:
|
| 17 |
-
|
| 18 |
|
| 19 |
try:
|
| 20 |
from langchain_text_splitters import RecursiveCharacterTextSplitter
|
|
|
|
| 9 |
from langchain_core.documents import Document
|
| 10 |
except ImportError:
|
| 11 |
try:
|
| 12 |
+
from langchain_core.documents import Document
|
| 13 |
+
except ImportError:
|
| 14 |
+
try:
|
| 15 |
+
from langchain_core.documents import Document
|
| 16 |
+
except ImportError:
|
| 17 |
+
from langchain.schema import Document
|
| 18 |
|
| 19 |
try:
|
| 20 |
from langchain_text_splitters import RecursiveCharacterTextSplitter
|
retrieval_evaluation.py
CHANGED
|
@@ -12,9 +12,9 @@ try:
|
|
| 12 |
from langchain_core.documents import Document
|
| 13 |
except ImportError:
|
| 14 |
try:
|
| 15 |
-
|
| 16 |
-
except ImportError:
|
| 17 |
-
|
| 18 |
from sklearn.metrics import ndcg_score, precision_score, recall_score, f1_score
|
| 19 |
from sentence_transformers import SentenceTransformer, util
|
| 20 |
import matplotlib.pyplot as plt
|
|
|
|
| 12 |
from langchain_core.documents import Document
|
| 13 |
except ImportError:
|
| 14 |
try:
|
| 15 |
+
from langchain_core.documents import Document
|
| 16 |
+
except ImportError:
|
| 17 |
+
from langchain.schema import Document
|
| 18 |
from sklearn.metrics import ndcg_score, precision_score, recall_score, f1_score
|
| 19 |
from sentence_transformers import SentenceTransformer, util
|
| 20 |
import matplotlib.pyplot as plt
|
test_reranking.py
CHANGED
|
@@ -14,9 +14,9 @@ try:
|
|
| 14 |
from langchain_core.documents import Document
|
| 15 |
except ImportError:
|
| 16 |
try:
|
| 17 |
-
|
| 18 |
-
except ImportError:
|
| 19 |
-
|
| 20 |
import time
|
| 21 |
|
| 22 |
|
|
|
|
| 14 |
from langchain_core.documents import Document
|
| 15 |
except ImportError:
|
| 16 |
try:
|
| 17 |
+
from langchain_core.documents import Document
|
| 18 |
+
except ImportError:
|
| 19 |
+
from langchain.schema import Document
|
| 20 |
import time
|
| 21 |
|
| 22 |
|
workflow_nodes.py
CHANGED
|
@@ -10,12 +10,12 @@ try:
|
|
| 10 |
from langchain_core.documents import Document
|
| 11 |
except ImportError:
|
| 12 |
try:
|
| 13 |
-
|
| 14 |
-
except ImportError:
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
except ImportError:
|
| 18 |
-
|
| 19 |
from langchain_community.chat_models import ChatOllama
|
| 20 |
from langchain_core.output_parsers import StrOutputParser
|
| 21 |
from langchain_community.tools.tavily_search import TavilySearchResults
|
|
@@ -23,9 +23,9 @@ try:
|
|
| 23 |
from langchain_core.prompts import PromptTemplate
|
| 24 |
except ImportError:
|
| 25 |
try:
|
| 26 |
-
|
| 27 |
-
except ImportError:
|
| 28 |
-
|
| 29 |
|
| 30 |
from config import LOCAL_LLM, WEB_SEARCH_RESULTS_COUNT, ENABLE_HYBRID_SEARCH, ENABLE_QUERY_EXPANSION, ENABLE_MULTIMODAL
|
| 31 |
from document_processor import DocumentProcessor
|
|
|
|
| 10 |
from langchain_core.documents import Document
|
| 11 |
except ImportError:
|
| 12 |
try:
|
| 13 |
+
from langchain_core.documents import Document
|
| 14 |
+
except ImportError:
|
| 15 |
+
try:
|
| 16 |
+
from langchain_core.documents import Document
|
| 17 |
+
except ImportError:
|
| 18 |
+
from langchain.schema import Document
|
| 19 |
from langchain_community.chat_models import ChatOllama
|
| 20 |
from langchain_core.output_parsers import StrOutputParser
|
| 21 |
from langchain_community.tools.tavily_search import TavilySearchResults
|
|
|
|
| 23 |
from langchain_core.prompts import PromptTemplate
|
| 24 |
except ImportError:
|
| 25 |
try:
|
| 26 |
+
from langchain_core.prompts import PromptTemplate
|
| 27 |
+
except ImportError:
|
| 28 |
+
from langchain.prompts import PromptTemplate
|
| 29 |
|
| 30 |
from config import LOCAL_LLM, WEB_SEARCH_RESULTS_COUNT, ENABLE_HYBRID_SEARCH, ENABLE_QUERY_EXPANSION, ENABLE_MULTIMODAL
|
| 31 |
from document_processor import DocumentProcessor
|