Update api.py
Browse files
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.
|
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 |
-
|
|
|
|
|
|
|
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:
|