mirror of
https://github.com/tcsenpai/mysides.git
synced 2025-06-04 10:10:05 +00:00
47 lines
978 B
Python
47 lines
978 B
Python
import os
|
|
|
|
import requests
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
pplx_api_key = os.getenv("PPLX_API_KEY")
|
|
model = os.getenv("MODEL")
|
|
|
|
with open("link", "r") as f:
|
|
article_link = f.read().strip()
|
|
|
|
headers = {
|
|
"accept": "application/json",
|
|
"authorization": "Bearer " + pplx_api_key,
|
|
"content-type": "application/json",
|
|
}
|
|
|
|
json_data = {
|
|
"model": model,
|
|
"messages": [
|
|
{
|
|
"role": "system",
|
|
"content": "Be precise, concise and clear",
|
|
},
|
|
{
|
|
"role": "user",
|
|
"content": "Search and summarize: " + article_link,
|
|
},
|
|
],
|
|
}
|
|
|
|
response = requests.post(
|
|
"https://api.perplexity.ai/chat/completions",
|
|
headers=headers,
|
|
json=json_data,
|
|
timeout=5,
|
|
)
|
|
|
|
response = response.json()
|
|
# print(response)
|
|
|
|
# print(response["choices"][0]["message"]["content"])
|
|
with open("response", "w+") as response_file:
|
|
response_file.write(response["choices"][0]["message"]["content"])
|