adrienbrdne commited on
Commit
1309b91
·
verified ·
1 Parent(s): 7c7e8ff

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +5 -3
api.py CHANGED
@@ -33,7 +33,7 @@ async def lifespan(app: FastAPI):
33
  api_key = os.getenv("GEMINI_API_KEY") or getattr(settings, "GEMINI_API_KEY", None)
34
  if not api_key:
35
  raise ValueError("GEMINI_API_KEY not found in environment or settings.")
36
- genai.configure(api_key=api_key)
37
  logger.info("Gemini client configured successfully.")
38
  except Exception as e:
39
  logger.error(f"Failed to configure Gemini client: {e}", exc_info=True)
@@ -140,9 +140,11 @@ def extract_research_paper_arxiv(rp_number: str, node_type: str) -> dict:
140
 
141
  try:
142
  model_name = "gemini-2.5-flash-preview-05-20"
143
- model = genai.GenerativeModel(model_name)
 
 
 
144
 
145
- response = model.generate_content(prompt)
146
  rp_data["summary"] = response.text
147
  logger.info(f"Summary generated for Arxiv ID: {rp_number}")
148
  except Exception as e:
 
33
  api_key = os.getenv("GEMINI_API_KEY") or getattr(settings, "GEMINI_API_KEY", None)
34
  if not api_key:
35
  raise ValueError("GEMINI_API_KEY not found in environment or settings.")
36
+ client = genai.Client(api_key=api_key)
37
  logger.info("Gemini client configured successfully.")
38
  except Exception as e:
39
  logger.error(f"Failed to configure Gemini client: {e}", exc_info=True)
 
140
 
141
  try:
142
  model_name = "gemini-2.5-flash-preview-05-20"
143
+ response = client.generate_content(
144
+ model=model_name,
145
+ contents=prompt
146
+ )
147
 
 
148
  rp_data["summary"] = response.text
149
  logger.info(f"Summary generated for Arxiv ID: {rp_number}")
150
  except Exception as e: