ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-02-01 08:48:19 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update bibliography.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-05-31 16:09:49 +02:00 committed by GitHub
parent fa98d94184
commit cb85e2643e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lollms/functions/bibliography.py
View File

@ -14,7 +14,7 @@ from typing import List, Optional, Any, Tuple, Dict
from lollms.utilities import PackageManager, find_first_available_file_index, discussion_path_to_url
# ascii_colors offers advanced console coloring and bug tracing
from ascii_colors import trace_exception
from ascii_colors import trace_exception, ASCIIColors
# Import Client from lollms.client_session
from lollms.client_session import Client
@ -43,6 +43,7 @@ def arxiv_pdf_search(query: str, max_results: Optional[int] = 5, sort_by: Option
if author:
url += f'&author={author}'
ASCIIColors.multicolor(["URL:",url],[ASCIIColors.red, ASCIIColors.yellow])
response = requests.get(url)
response.raise_for_status()
@ -355,4 +356,4 @@ def search_and_rank_function(llm, score_threshold:float, client: Optional[Any] =
{"name": "end_date", "type": "str", "description": "The end date for the search results in the format YYYY-MM-DD. (Optional)", "optional": True},
{"name": "author", "type": "str", "description": "The author name for the search results. (Optional)", "optional": True},
]
}
}