sk1502 commited on
Commit
cababea
·
verified ·
1 Parent(s): 48bdb9c

Update aisearch_v1.py

Browse files
Files changed (1) hide show
  1. 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 = self.sp_final_report,
179
  config=types.GenerateContentConfig(
180
- system_instruction=prompt,
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)