wt002 commited on
Commit
119459e
·
verified ·
1 Parent(s): d956180

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -136,7 +136,7 @@ def python_execution(code: str) -> str:
136
  except Exception as e:
137
  return f"Python execution error: {str(e)}"
138
 
139
- class VideoTranscriptionTool(BaseTool):
140
  name: str = "transcript_video"
141
  # CORRECTED LINE BELOW: Added '=' for assignment
142
  description: str = "Fetch text transcript from YouTube videos using URL or ID. Use for any question involving video or audio. Input is the YouTube URL or ID."
@@ -532,12 +532,12 @@ class BasicAgent:
532
  def __init__(self):
533
  # Instantiate tools (implementation not shown - should be defined elsewhere)
534
  self.tools = [
535
- # duckduckgo_search,
536
- # wikipedia_search,
537
- # arxiv_search,
538
- # document_qa,
539
- # python_execution,
540
- # VideoTranscriptionTool()
541
  ]
542
 
543
  # Pre-initialize RAG vector store
 
136
  except Exception as e:
137
  return f"Python execution error: {str(e)}"
138
 
139
+ class VideoTranscriptionTool(Tool):
140
  name: str = "transcript_video"
141
  # CORRECTED LINE BELOW: Added '=' for assignment
142
  description: str = "Fetch text transcript from YouTube videos using URL or ID. Use for any question involving video or audio. Input is the YouTube URL or ID."
 
532
  def __init__(self):
533
  # Instantiate tools (implementation not shown - should be defined elsewhere)
534
  self.tools = [
535
+ duckduckgo_search,
536
+ wikipedia_search,
537
+ arxiv_search,
538
+ document_qa,
539
+ python_execution,
540
+ VideoTranscriptionTool()
541
  ]
542
 
543
  # Pre-initialize RAG vector store