aquibmoin commited on
Commit
a806bea
·
verified ·
1 Parent(s): 19426a7

Update utils/ads_references.py

Browse files
Files changed (1) hide show
  1. 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)