2_marvin.py

"""
Data Scientist.: Dr. Eddy Giusepe Chirinos Isidro
"""
from marvin import ai_fn
from typing_extensions import TypedDict


# Substitua sua chave de API OpenAI:
import openai
import os
from dotenv import load_dotenv, find_dotenv
_ = load_dotenv(find_dotenv()) # read local .env file
openai.api_key  = os.environ['OPENAI_API_KEY']


class GrantInfo(TypedDict):
    organization_name: str
    grant_amount: str
    rationale: str


@ai_fn(model="openai/gpt-3.5-turbo-1106")
def extract_grant_info(text: str) -> GrantInfo:
    '''
    Infere corretamente informações importantes sobre uma concessão da organização: nome_da_organização, \
    valor_da_doação e justificativa do texto transmitido.
    '''

if __name__=='__main__':
    response = extract_grant_info('A empresa Eddy DS dou 100000 dólares para pesquisa em Inteligência Artificial.')
    print(response)