KaiserShultz commited on
Commit
ad62150
·
verified ·
1 Parent(s): 7dcd8a6

Update src/config.py

Browse files
Files changed (1) hide show
  1. src/config.py +2 -1
src/config.py CHANGED
@@ -4,6 +4,7 @@ from src.tools.code_interpreter import safe_code_run
4
  from langgraph.prebuilt import ToolNode
5
  from src.schemas import PlannerPlan
6
  from src.tools.youtube_transcript import extract_youtube_transcript
 
7
  from src.utils.utils import log_stage
8
  from langchain_core.messages import HumanMessage, SystemMessage, AIMessage, ToolMessage
9
  from dotenv import load_dotenv
@@ -15,7 +16,7 @@ TOOLS = [download_file_from_url, web_search,
15
  arxiv_search, wiki_search, add, subtract, multiply, divide,
16
  power, analyze_excel_file, analyze_csv_file, analyze_docx_file,
17
  analyze_pdf_file, analyze_txt_file,
18
- vision_qa_gemma, safe_code_run, web_extract, extract_youtube_transcript, transcribe_audio]
19
 
20
 
21
  TOOL_NODE = ToolNode(TOOLS)
 
4
  from langgraph.prebuilt import ToolNode
5
  from src.schemas import PlannerPlan
6
  from src.tools.youtube_transcript import extract_youtube_transcript
7
+ from tools.video_analyzer import video_qa_gemma
8
  from src.utils.utils import log_stage
9
  from langchain_core.messages import HumanMessage, SystemMessage, AIMessage, ToolMessage
10
  from dotenv import load_dotenv
 
16
  arxiv_search, wiki_search, add, subtract, multiply, divide,
17
  power, analyze_excel_file, analyze_csv_file, analyze_docx_file,
18
  analyze_pdf_file, analyze_txt_file,
19
+ vision_qa_gemma, safe_code_run, web_extract, extract_youtube_transcript, transcribe_audio, video_qa_gemma]
20
 
21
 
22
  TOOL_NODE = ToolNode(TOOLS)