Update agent.py

agent.py

@@ -19,7 +19,7 @@ from langchain_openai import ChatOpenAI
 
 from langchain.tools import Tool
 from code_interpreter import CodeInterpreter
-#from langchain_chroma import Chroma
+from langchain_chroma import Chroma
 from langchain_core.messages import AIMessage
 
 interpreter_instance = CodeInterpreter()
@@ -231,7 +231,7 @@ execute_code_multilang_tool = Tool(
 ### ======================================== RETRIEVER TOOLS ======================================== ###
 
 embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
-"""
+
 vector_store = Chroma(
     collection_name="gaia_dataset",
     embedding_function=embeddings,
@@ -243,6 +243,7 @@ create_retriever_tool = create_retriever_tool(
     name="Question Search",
     description="A tool to retrieve similar questions from a vector store.",
 )
+
 """
 # build a retriever
 embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2") #  dim=768
@@ -260,6 +261,7 @@ create_retriever_tool = create_retriever_tool(
     name="Question Search",
     description="A tool to retrieve similar questions from a vector store.",
 )
+"""
 
 
 # load the system prompt from the file