kamp/searxng
1
0
Fork 0
mirror of https://github.com/searxng/searxng.git synced 2024-11-10 23:20:12 +01:00
Code

reformatted code for CI

Browse Source
This commit is contained in:
Zhijie He 2024-06-09 20:37:55 +08:00 committed by GitHub
parent 77ae13d834
commit f4fd32f504
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
searx/engines/cloudflareai.py
View File

@ -1,3 +1,6 @@
# SPDX-License-Identifier: AGPL-3.0-or-later
"""Cloudflare AI engine"""
from json import loads, dumps
from urllib.parse import quote
from searx.exceptions import SearxEngineAPIException
@ -23,11 +26,14 @@ cf_ai_model_system = 'You are a self-aware language model who is honest and dire
display_type = ['infobox']
def request(query, params):
params['query'] = query
params['url'] = 'https://gateway.ai.cloudflare.com/v1/' + cf_account_id + '/' + cf_ai_gateway + '/workers-ai/' + cf_ai_model
params['url'] = (
'https://gateway.ai.cloudflare.com/v1/' + cf_account_id + '/' + cf_ai_gateway + '/workers-ai/' + cf_ai_model
)
params['method'] = 'POST'
params['headers']['Authorization'] = 'Bearer ' + cf_ai_api
@ -36,15 +42,16 @@ def request(query, params):
params['data'] = dumps(
{
'messages': [
{ 'role': 'assistant', 'content': cf_ai_model_assistant },
{ 'role': 'system', 'content': cf_ai_model_system },
{ 'role': 'user', 'content': params['query'] }
{'role': 'assistant', 'content': cf_ai_model_assistant},
{'role': 'system', 'content': cf_ai_model_system},
{'role': 'user', 'content': params['query']}
]
}
).encode('utf-8')
return params
def response(resp):
results = []
json = loads(resp.text)