Spaces:
Sleeping
Sleeping
Update aisearch_v1.py
Browse files- aisearch_v1.py +3 -3
aisearch_v1.py
CHANGED
|
@@ -175,15 +175,15 @@ class aiSearch:
|
|
| 175 |
# try:
|
| 176 |
response = await gemini_client.aio.models.generate_content(
|
| 177 |
model='gemini-1.5-flash',
|
| 178 |
-
contents =
|
| 179 |
config=types.GenerateContentConfig(
|
| 180 |
-
system_instruction=
|
| 181 |
max_output_tokens= 4096,
|
| 182 |
temperature= 0.1
|
| 183 |
)
|
| 184 |
)
|
| 185 |
|
| 186 |
-
return response
|
| 187 |
|
| 188 |
# except Exception as e:
|
| 189 |
# print(e)
|
|
|
|
| 175 |
# try:
|
| 176 |
response = await gemini_client.aio.models.generate_content(
|
| 177 |
model='gemini-1.5-flash',
|
| 178 |
+
contents = prompt,
|
| 179 |
config=types.GenerateContentConfig(
|
| 180 |
+
system_instruction=self.research_prompt,
|
| 181 |
max_output_tokens= 4096,
|
| 182 |
temperature= 0.1
|
| 183 |
)
|
| 184 |
)
|
| 185 |
|
| 186 |
+
return response.text
|
| 187 |
|
| 188 |
# except Exception as e:
|
| 189 |
# print(e)
|