Update
Browse files
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="
|
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 |
)
|