EddyGiusepe commited on
Commit
8e89862
·
1 Parent(s): baa1080

Usando o classificador @ai_classifier

Browse files
Files changed (1) hide show
  1. 3_marvin.py +39 -0
3_marvin.py ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ Data Scientist.: Dr. Eddy Giusepe Chirinos Isidro
3
+
4
+ Link de estudo --> https://www.askmarvin.ai/components/ai_classifier/
5
+
6
+ Classificador de IA
7
+ ===================
8
+ '@ai_classifier' é um decorador que permite usar LLMs para escolher opções, ferramentas ou classificar entradas.
9
+ Usaremos este Classificador para analisar o sentimento das queries do usuário.
10
+ """
11
+ # Substitua sua chave de API OpenAI:
12
+ import openai
13
+ import os
14
+ from dotenv import load_dotenv, find_dotenv
15
+ _ = load_dotenv(find_dotenv()) # read local .env file
16
+ openai.api_key = os.environ['OPENAI_API_KEY']
17
+
18
+
19
+ from marvin import ai_classifier
20
+ from enum import Enum
21
+
22
+
23
+ @ai_classifier(model="openai/gpt-3.5-turbo-1106", temperature = 0)
24
+ class Sentimento(Enum):
25
+ POSITIVE = "POSITIVO"
26
+ NEGATIVE = "NEGATIVO"
27
+ NEUTRO = "NEUTRO"
28
+
29
+
30
+ while True:
31
+ query = input("\033[033mQuery do usuário:🤓\033[m")
32
+ print("")
33
+ resposta = Sentimento(query)
34
+ print(resposta.value)
35
+ if resposta.value == "NEGATIVO":
36
+ print("Desculpe pelo transtorno. Entraremos em contato, o mais breve possível.")
37
+ if not query:
38
+ break
39
+ # Hoje foi na DETRAN cedinho e ainda não consegui informações de como renovar a minha CNH?