Update src/config.py
Browse files- 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)
|