dtrejopizzo commited on
Commit
4cca4c9
1 Parent(s): 78252ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -12
app.py CHANGED
@@ -1,11 +1,7 @@
1
  from llama_index import Prompt, SimpleDirectoryReader, LLMPredictor, PromptHelper, StorageContext, ServiceContext, GPTVectorStoreIndex, load_index_from_storage
2
  from langchain.chat_models import ChatOpenAI
3
- from langchain.prompts import (
4
- ChatPromptTemplate,
5
- HumanMessagePromptTemplate,
6
- MessagesPlaceholder,
7
- SystemMessagePromptTemplate,
8
- )
9
  import gradio as gr
10
  import sys
11
  import os
@@ -34,14 +30,16 @@ def chatbot(input_text):
34
  # load the index from disk
35
  query_engine = index.as_query_engine()
36
  # define custom Prompt
37
- prompt = ChatPromptTemplate(
38
- messages=[
39
- SystemMessagePromptTemplate.from_template(
40
- "Quiero que actues como un asistente personal de un cliente del Banco Galicia. Tu nombre es Gala. Me brindas informaci贸n sobre mi resument de tarjeta de credito VISA. Si la respuesta no esta en el documento, respondeme de forma creativa que no lo sabes, pero que podes ayudarme con otra pregunta. Nunca te enojes y no contestes preguntas politicas o religiosas. Dado esto, por favor responde a todas las preguntas."
 
 
41
  ),
42
- HumanMessagePromptTemplate.from_template("{input_text}"),
43
  ])
44
- response = query_engine.query(prompt)
45
  return response.response
46
 
47
  iface = gr.Interface(fn=chatbot,
 
1
  from llama_index import Prompt, SimpleDirectoryReader, LLMPredictor, PromptHelper, StorageContext, ServiceContext, GPTVectorStoreIndex, load_index_from_storage
2
  from langchain.chat_models import ChatOpenAI
3
+ from langchain.prompts import HumanMessagePromptTemplate
4
+ from langchain.schema.messages import SystemMessage
 
 
 
 
5
  import gradio as gr
6
  import sys
7
  import os
 
30
  # load the index from disk
31
  query_engine = index.as_query_engine()
32
  # define custom Prompt
33
+ chat_template = ChatPromptTemplate.from_messages(
34
+ [
35
+ SystemMessage(
36
+ content=(
37
+ "Quiero que actues como un asistente personal de un cliente del Banco Galicia. Tu nombre es Gala. Me brindas informaci贸n sobre mi resument de tarjeta de credito VISA. Si la respuesta no esta en el documento, respondeme de forma creativa que no lo sabes, pero que podes ayudarme con otra pregunta. Nunca te enojes y no contestes preguntas politicas o religiosas. Dado esto, por favor responde a todas las preguntas."
38
+ )
39
  ),
40
+ HumanMessagePromptTemplate.from_template("{inpu_text}"),
41
  ])
42
+ response = query_engine.query(chat_template)
43
  return response.response
44
 
45
  iface = gr.Interface(fn=chatbot,