39 lines
924 B
Python
39 lines
924 B
Python
|
from mistralai import Mistral
|
||
|
|
import inbox
|
||
|
|
import os
|
||
|
|
|
||
|
|
api_key = os.environ["MISTRAL_API_KEY"]
|
||
|
|
if api_key is None:
|
||
|
|
raise "Please provide a valid MISTRAL_API_KEY"
|
||
|
|
|
||
|
|
|
||
|
|
client = Mistral(api_key=api_key)
|
||
|
|
|
||
|
|
def process(instructions: str) -> str:
|
||
|
|
messages = [{
|
||
|
|
"role": "system",
|
||
|
|
"content": instructions,
|
||
|
|
},{
|
||
|
|
"role": "system",
|
||
|
|
"content": "Here is the list of emails",
|
||
|
|
}]
|
||
|
|
|
||
|
|
for mail in inbox.fetch_emails():
|
||
|
|
content = f"""
|
||
|
|
Date: {mail["date"]}
|
||
|
|
From: {mail["from"]}
|
||
|
|
To: {mail["to"]}
|
||
|
|
|
||
|
|
Subject: {mail["subject"]}
|
||
|
|
Body: {mail["text"]}
|
||
|
|
"""
|
||
|
|
messages.append({
|
||
|
|
"role": "user",
|
||
|
|
"content": content,
|
||
|
|
})
|
||
|
|
|
||
|
|
chat_response = client.chat.complete(
|
||
|
|
model = "mistral-tiny",
|
||
|
|
messages = messages)
|
||
|
|
|
||
|
|
return chat_response.choices[0].message.content
|