From cb85e2643e3c92c3863e5a0208f2463ed7af7303 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Fri, 31 May 2024 16:09:49 +0200 Subject: [PATCH] Update bibliography.py --- lollms/functions/bibliography.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/lollms/functions/bibliography.py b/lollms/functions/bibliography.py index c7a4137..4cbc9a1 100644 --- a/lollms/functions/bibliography.py +++ b/lollms/functions/bibliography.py @@ -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}, ] - } \ No newline at end of file + }