Update agent.py

agent.py

@@ -38,24 +38,6 @@ with open("prompt.txt", "r", encoding="utf-8") as f:
 # System message
 sys_msg = SystemMessage(content=system_prompt)
 
-# # build a retriever
-# embeddings = HuggingFaceEmbeddings(
-#     model_name="sentence-transformers/all-mpnet-base-v2"
-# )  #  dim=768
-# supabase: Client = create_client(
-#     SUPABASE_URL, SUPABASE_SERVICE_ROLE_KEY
-# )
-# vector_store = SupabaseVectorStore(
-#     client=supabase,
-#     embedding=embeddings,
-#     table_name="documents2",
-#     query_name="match_documents_2",
-# )
-# create_retriever_tool = create_retriever_tool(
-#     retriever=vector_store.as_retriever(),
-#     name="Question Search",
-#     description="A tool to retrieve similar questions from a vector store.",
-# )
 
 # Loading the assistant
 chat = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
@@ -64,7 +46,6 @@ tools = [add,
          substract, 
          multiply, 
          divide, 
-         #DuckDuckGoSearchTool, 
          TavilySearchTool, 
          WikipediaSearchTool, 
          ArxivSearchTool, 
@@ -73,7 +54,10 @@ tools = [add,
          download_file_from_url,
          extract_text_from_image,
          analyze_csv_file, 
-         analyze_excel_file]
+         analyze_excel_file,
+         extract_video_id,
+         get_youtube_transcript
+        ]
 
 chat_with_tools = chat.bind_tools(tools)