Spaces:
Sleeping
Sleeping
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 |
)
|