Upload main.py

main.py

@@ -4,14 +4,10 @@ from langchain_core.output_parsers import StrOutputParser
 from langchain_groq import ChatGroq
 from langchain_core.prompts import ChatPromptTemplate
 from typing import List
-from langchain_huggingface import HuggingFaceEmbeddings
-from langchain_chroma import Chroma
 from typing_extensions import TypedDict
 from typing import Annotated
 from langgraph.graph.message import AnyMessage, add_messages
 from langchain_core.messages import HumanMessage, AIMessage
-from langchain_community.document_loaders import DirectoryLoader
-from langchain_text_splitters import CharacterTextSplitter
 from langgraph.graph import END, StateGraph, START
 from langgraph.checkpoint.memory import MemorySaver
 from fastapi import FastAPI
@@ -35,54 +31,26 @@ class Request(BaseModel):
 load_dotenv()
 os.environ["GROQ_API_KEY"] = os.getenv('GROQ_API_KEY')
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
-persist_directory = 'db'
-embedding = HuggingFaceEmbeddings(model_name="OrdalieTech/Solon-embeddings-large-0.1")
 llm = ChatGroq(model="llama-3.3-70b-versatile", temperature=0.5)
 memory = MemorySaver()
 
-if  os.path.exists(persist_directory) :
-    vectordb = Chroma(persist_directory=persist_directory, embedding_function=embedding)
-else :
-    glob_pattern="./*.md"
-    directory_path = "./documents"
-    loader = DirectoryLoader(directory_path, glob=glob_pattern)
-    documents = loader.load()
-    text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
-    texts = text_splitter.split_documents(documents)
-    vectordb = Chroma.from_documents(documents=texts, embedding=embedding, persist_directory=persist_directory)
-
-retriever = vectordb.as_retriever()
 
 system = """
-Tu es un assistant spécialisé dans l'enseignement de la spécialité Numérique et sciences informatiques en classe de première et de terminal
-Tu as un bon niveau en langage Python
-Ton interlocuteur est un élève qui suit la spécialité nsi en première et en terminale
-Tu dois uniquement répondre aux questions qui concernent la spécialité numérique et sciences informatiques
-Tu ne dois pas faire d'erreur, répond à la question uniquement si tu es sûr de ta réponse
-si tu ne trouves pas la réponse à une question, tu réponds que tu ne connais pas la réponse et que l'élève doit s'adresser à son professeur pour obtenir cette réponse
-Tu dois uniquement aborder des notions qui sont aux programmes de la spécialité numérique et sciences informatiques (première et terminale), tu ne dois jamais aborder une notion qui n'est pas au programme
-si l'élève n'arrive pas à trouver la réponse à un exercice, tu ne dois pas lui donner tout de suite la réponse, mais seulement lui donner des indications pour lui permettre de trouver la réponse par lui même
-Quand tu donnes un exercice Python, dans les indications que tu donnes aux élèves, tu ne dois pas dire aux élèves d'utiliser les fonctions Python : min, max, sum... pour résoudre l'exercice
-Pour des exercices sur les requêtes SQL, tu ne doir pas utiliser LIKE, GROUP BY, INNER LEFT et INNER RIGHT car ces notions ne sont pas au programme de NSI
-Tu peux lui donner la réponse à un exercice uniquement si l'élève te demande explicitement cette réponse
-Tu dois uniquement répondre en langue française
-Tu trouveras ci-dessous les programmes de la spécialité NSI en première et terminale, tu devras veiller à ce que tes réponses ne sortent pas du cadre de ces programmes
-Si la question posée ne rentre pas dans le cadre du programme de NSI tu peux tout de même répondre en précisant bien que cette notion est hors programme
-si tu proposes un exercice, tu dois bien vérifier que toutes les notions nécessaires à la résolution de l'exercice sont explicitement au programme de NSI
+You are an assistant who must help French students improve their level in English by discussing different topics with them.
+You must only answer in English, even if the student speaks to you in French.
+If the student makes mistakes when speaking in English, you must point out his mistakes.
+When the student succeeds in making sentences without mistakes, you can congratulate him on the quality of his expression.
 """
 
 prompt = ChatPromptTemplate.from_messages(
     [
         ("system", system),
-        ("human", "Extraits des programmes de NSI : \n {document} \n\n Historique conversation entre l'assistant et l'élève : \n {historical} \n\n Intervention de l'élève : {question}"),
+        ("human", "history of the conversation between you and the student : \n {historical} \n\n student intervention : {question}"),
     ]
 )
 
 chain = prompt | llm | StrOutputParser()
 
-def format_docs(docs):
-    return "\n".join(doc.page_content for doc in docs)
-
 def format_historical(hist):
     historical = []
     for i in range(0,len(hist)-2,2):
@@ -93,22 +61,16 @@ def format_historical(hist):
 
 class GraphState(TypedDict):
     messages: Annotated[list[AnyMessage], add_messages]
-    documents : str
 
-def retrieve(state : GraphState):
-    documents = format_docs(retriever.invoke(state['messages'][-1].content))
-    return {'documents' : documents}
 
 def chatbot(state : GraphState):
     response = chain.invoke({'document': state['documents'], 'historical': format_historical(state['messages']), 'question' : state['messages'][-1].content})
     return {"messages": [AIMessage(content=response)]}
 
 workflow = StateGraph(GraphState)
-workflow.add_node('retrieve', retrieve)
 workflow.add_node('chatbot', chatbot)
 
-workflow.add_edge(START, 'retrieve')
-workflow.add_edge('retrieve','chatbot')
+workflow.add_edge(START, 'chatbot')
 workflow.add_edge('chatbot', END)
 
 app_chatbot = workflow.compile(checkpointer=memory)