akiko19191 commited on
Commit
e954578
·
verified ·
1 Parent(s): 212b67b

Update tests.py

Browse files
Files changed (1) hide show
  1. tests.py +2 -1
tests.py CHANGED
@@ -363,8 +363,9 @@ def scrape_websites(url_list:list,query:str) -> list:
363
  res = conn.getresponse()
364
  data = res.read()
365
  content=str(data.decode("utf-8"))
 
366
  response = completion(
367
- model="gemini/gemini-2.0-flash-exp",
368
  messages=[
369
  {"role": "user", "content": f"Output the following content in the human readable format.Try to conserve all the links and the text.Try to ouput the entire content.Remove the html codes so its human readable.Also answer this question about the content in a seperate paragraph:{query}.Here is the content:{content}"}
370
  ],
 
363
  res = conn.getresponse()
364
  data = res.read()
365
  content=str(data.decode("utf-8"))
366
+
367
  response = completion(
368
+ model="gemini/gemini-1.5-pro",
369
  messages=[
370
  {"role": "user", "content": f"Output the following content in the human readable format.Try to conserve all the links and the text.Try to ouput the entire content.Remove the html codes so its human readable.Also answer this question about the content in a seperate paragraph:{query}.Here is the content:{content}"}
371
  ],