Update utils/ads_references.py
Browse files- utils/ads_references.py +1 -3
utils/ads_references.py
CHANGED
@@ -1,8 +1,5 @@
|
|
1 |
from astroquery.nasa_ads import ADS
|
2 |
|
3 |
-
ADS.TOKEN = os.getenv('ADS_API_KEY')
|
4 |
-
|
5 |
-
|
6 |
def extract_keywords_with_gpt(context, client, max_tokens=100, temperature=0.3):
|
7 |
|
8 |
keyword_prompt = f"Extract 3 most important scientific keywords from the following user query:\n\n{context}"
|
@@ -27,6 +24,7 @@ def extract_keywords_with_gpt(context, client, max_tokens=100, temperature=0.3):
|
|
27 |
|
28 |
def fetch_nasa_ads_references(ads_query):
|
29 |
"""Fetch relevant NASA ADS papers and format them for readability."""
|
|
|
30 |
try:
|
31 |
# Query NASA ADS for relevant papers
|
32 |
papers = ADS.query_simple(ads_query)
|
|
|
1 |
from astroquery.nasa_ads import ADS
|
2 |
|
|
|
|
|
|
|
3 |
def extract_keywords_with_gpt(context, client, max_tokens=100, temperature=0.3):
|
4 |
|
5 |
keyword_prompt = f"Extract 3 most important scientific keywords from the following user query:\n\n{context}"
|
|
|
24 |
|
25 |
def fetch_nasa_ads_references(ads_query):
|
26 |
"""Fetch relevant NASA ADS papers and format them for readability."""
|
27 |
+
ADS.TOKEN = os.getenv('ADS_API_KEY')
|
28 |
try:
|
29 |
# Query NASA ADS for relevant papers
|
30 |
papers = ADS.query_simple(ads_query)
|