om4r932 commited on
Commit
1b96641
·
1 Parent(s): 1392287
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -221,7 +221,7 @@ def gen_reqs(req: RequirementsRequest):
221
 
222
  resp_ai = litellm.completion(
223
  model="gemini/gemini-2.0-flash",
224
- api_key="SECRET API HERE",
225
  messages=[{"role":"user","content": f"Here's the document whose ID is {doc_id} with requirements : {full}\n\nI want you to extract all the requirements and give me a context (not giving the section or whatever, a sentence is needed) where that calls for those requirements. If multiples covered contexts is present, make as many requirements list by context as you want."}],
226
  response_format=DocRequirements
227
  )
 
221
 
222
  resp_ai = litellm.completion(
223
  model="gemini/gemini-2.0-flash",
224
+ api_key=os.environ.get("GEMINI"),
225
  messages=[{"role":"user","content": f"Here's the document whose ID is {doc_id} with requirements : {full}\n\nI want you to extract all the requirements and give me a context (not giving the section or whatever, a sentence is needed) where that calls for those requirements. If multiples covered contexts is present, make as many requirements list by context as you want."}],
226
  response_format=DocRequirements
227
  )