Spaces:
Sleeping
Sleeping
vincentmin
commited on
Commit
·
7f82885
1
Parent(s):
d3e5a67
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ from langchain.prompts import PromptTemplate
|
|
10 |
from langchain.schema import Document
|
11 |
|
12 |
|
13 |
-
|
14 |
FORMAT = '%Y%m%d%H%M%S'
|
15 |
embeddings = HuggingFaceEmbeddings()
|
16 |
|
@@ -56,7 +56,7 @@ def get_documents(category: str, min_date: datetime, max_date: datetime):
|
|
56 |
query = f"cat:{category} AND submittedDate:[{min_date.strftime(FORMAT)} TO {max_date.strftime(FORMAT)}]"
|
57 |
search = arxiv.Search(
|
58 |
query=query,
|
59 |
-
max_results=
|
60 |
sort_by=arxiv.SortCriterion.SubmittedDate
|
61 |
)
|
62 |
docs = [Document(
|
|
|
10 |
from langchain.schema import Document
|
11 |
|
12 |
|
13 |
+
MAX_RESULTS = 100
|
14 |
FORMAT = '%Y%m%d%H%M%S'
|
15 |
embeddings = HuggingFaceEmbeddings()
|
16 |
|
|
|
56 |
query = f"cat:{category} AND submittedDate:[{min_date.strftime(FORMAT)} TO {max_date.strftime(FORMAT)}]"
|
57 |
search = arxiv.Search(
|
58 |
query=query,
|
59 |
+
max_results=MAX_RESULTS,
|
60 |
sort_by=arxiv.SortCriterion.SubmittedDate
|
61 |
)
|
62 |
docs = [Document(
|