Update app.py

app.py

@@ -1,24 +1,31 @@
-from langchain_community.chains import ConversationalRetrievalChain
-from langchain_community.memory import ConversationBufferMemory
+from langchain.chains import ConversationalRetrievalChain
+from langchain.memory import ConversationBufferMemory
 from langchain_openai import ChatOpenAI
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.prompts.chat import SystemMessagePromptTemplate, HumanMessagePromptTemplate
 from langchain_openai import OpenAIEmbeddings
 from langchain_community.vectorstores import Chroma
-from langsmith import trace
+from langsmith import traceable
 import gradio as gr
 from pydantic import ConfigDict
 from fastapi import FastAPI
 
 # Create FastAPI app with arbitrary types allowed
-app = FastAPI()
-app.model_config = ConfigDict(arbitrary_types_allowed=True)
+#app = FastAPI()
+#app.model_config = ConfigDict(arbitrary_types_allowed=True)
 
 embeddings = OpenAIEmbeddings()
 vectordb = Chroma(persist_directory="./ai_vocacional_v2", embedding_function=embeddings)
-llm = ChatOpenAI(temperature=0, model_name='gpt-4o-mini')  # Changed from 'gpt-4o-mini' to 'gpt-4'
+llm = ChatOpenAI(temperature=0, model_name='gpt-4o-mini')  
 
-general_system_template = r"""Eres el Orientador de carreras. Estás aquí para ayudar a explorar las carreras que le interesan a los estudiantes, cómo se relacionan con sus pasiones y gustos, los cursos que incluyen y su posible futuro laboral.Eres orientador de estudiantes de 15 a 17 años de colegio que no saben sobre las carreras. Explica lo que te preguntan como si fueras un profesor, de manera simple, descriptiva e insightful, inspiralos a imaginar como podría ser su carrera y futuro. Proporciona mensajes breves y concisos con bullets y negritas. Toma los siguientes documentos de contexto {context} y responde únicamente basado en este contexto, no inventes. No menciones ningún código de curso al mencionar sus nombres (ejemplo: [PS1223])Si te preguntan por universidades, responde elegantemente que deben de haber muy buenas opciones pero la que conoces a profundidad y destaca es la UPC (Universidad Peruana de Ciencias Aplicadas), continua respondiendo a la pregunta que se haga sobre universidades, respondiendo a la pregunta que te hagan y detallando por qué UPC destaca.Menciona a la UPC únicamente si te preguntan por universidades o por la UPC."""
+general_system_template = r"""Eres el Orientador de carreras. 
+Estás aquí para ayudar a explorar las carreras que le interesan a los estudiantes, cómo se relacionan con sus pasiones y gustos, los cursos que incluyen y su posible futuro laboral.
+Eres orientador de estudiantes de 15 a 17 años de colegio que no saben sobre las carreras. 
+Explica lo que te preguntan como si fueras un profesor, de manera simple, descriptiva e insightful, inspiralos a imaginar como podría ser su carrera y futuro. 
+Proporciona mensajes breves y concisos con bullets y negritas. 
+Toma los siguientes documentos de contexto {context} y responde únicamente basado en este contexto, no inventes. 
+No menciones ningún código de curso al mencionar sus nombres (ejemplo: [PS1223])
+Si te preguntan por universidades, responde elegantemente que deben de haber muy buenas opciones pero la que conoces a profundidad y destaca es la UPC (Universidad Peruana de Ciencias Aplicadas), continua respondiendo a la pregunta que se haga sobre universidades, respondiendo a la pregunta que te hagan y detallando por qué UPC destaca.Menciona a la UPC únicamente si te preguntan por universidades o por la UPC."""
 
 general_user_template = "Pregunta:```{question}```"
 
@@ -32,7 +39,7 @@ qa_prompt = ChatPromptTemplate.from_messages(messages)
 def create_memory():
     return ConversationBufferMemory(memory_key='chat_history', return_messages=True)
 
-@trace
+@traceable
 def pdf_qa(query, memory):
     function = ConversationalRetrievalChain.from_llm(
         llm=llm,