Spaces:
Sleeping
Sleeping
Samuel Thomas
commited on
Commit
·
26abdc4
1
Parent(s):
d5d4678
duckduckgo debug
Browse files
tools.py
CHANGED
|
@@ -794,7 +794,7 @@ class EnhancedDuckDuckGoSearchTool(BaseTool):
|
|
| 794 |
"Ideal for topics that require the latest news, recent developments, or information not covered in static sources."
|
| 795 |
)
|
| 796 |
max_results: int = 3
|
| 797 |
-
max_chars_per_page: int =
|
| 798 |
session: Any = None # Now it's optional and defaults to None
|
| 799 |
|
| 800 |
|
|
@@ -945,8 +945,8 @@ Found {len(search_results)} results, successfully processed {processed_count} pa
|
|
| 945 |
"""
|
| 946 |
|
| 947 |
# Ensure the response isn't too long
|
| 948 |
-
if len(response) >
|
| 949 |
-
response = response[:
|
| 950 |
|
| 951 |
return response
|
| 952 |
|
|
@@ -1413,8 +1413,8 @@ class EnhancedYoutubeScreenshotQA(BaseTool):
|
|
| 1413 |
def _get_config(self, key: str, default_value=None, input_data: Dict[str, Any] = None):
|
| 1414 |
"""Get configuration value with fallback to defaults"""
|
| 1415 |
defaults = {
|
| 1416 |
-
'frame_interval_seconds':
|
| 1417 |
-
'max_frames':
|
| 1418 |
'use_scene_detection': True,
|
| 1419 |
'resize_frames': True,
|
| 1420 |
'parallel_processing': True,
|
|
@@ -3333,7 +3333,7 @@ def call_tool_with_memory_management(state: AgentState) -> AgentState:
|
|
| 3333 |
if not isinstance(tool_result, str):
|
| 3334 |
tool_result = str(tool_result)
|
| 3335 |
|
| 3336 |
-
max_length =
|
| 3337 |
if len(tool_result) > max_length:
|
| 3338 |
original_length = len(tool_result)
|
| 3339 |
tool_result = tool_result[:max_length] + f"... [Result truncated from {original_length} to {max_length} chars to prevent memory issues]"
|
|
@@ -3591,7 +3591,7 @@ def run_agent(agent, state: AgentState):
|
|
| 3591 |
|
| 3592 |
# Initialize tools
|
| 3593 |
WIKIPEDIA_TOOL = WikipediaSearchToolWithFAISS()
|
| 3594 |
-
SEARCH_TOOL = EnhancedDuckDuckGoSearchTool(max_results=3, max_chars_per_page=
|
| 3595 |
YOUTUBE_TOOL = EnhancedYoutubeScreenshotQA()
|
| 3596 |
YOUTUBE_AUDIO_TOOL = YouTubeTranscriptExtractor()
|
| 3597 |
AUDIO_TRANSCRIPTION_TOOL = AudioTranscriptionTool()
|
|
|
|
| 794 |
"Ideal for topics that require the latest news, recent developments, or information not covered in static sources."
|
| 795 |
)
|
| 796 |
max_results: int = 3
|
| 797 |
+
max_chars_per_page: int = 12000
|
| 798 |
session: Any = None # Now it's optional and defaults to None
|
| 799 |
|
| 800 |
|
|
|
|
| 945 |
"""
|
| 946 |
|
| 947 |
# Ensure the response isn't too long
|
| 948 |
+
if len(response) > 12000:
|
| 949 |
+
response = response[:12000] + "\n[Response truncated to prevent memory issues]"
|
| 950 |
|
| 951 |
return response
|
| 952 |
|
|
|
|
| 1413 |
def _get_config(self, key: str, default_value=None, input_data: Dict[str, Any] = None):
|
| 1414 |
"""Get configuration value with fallback to defaults"""
|
| 1415 |
defaults = {
|
| 1416 |
+
'frame_interval_seconds': 20,
|
| 1417 |
+
'max_frames': 50,
|
| 1418 |
'use_scene_detection': True,
|
| 1419 |
'resize_frames': True,
|
| 1420 |
'parallel_processing': True,
|
|
|
|
| 3333 |
if not isinstance(tool_result, str):
|
| 3334 |
tool_result = str(tool_result)
|
| 3335 |
|
| 3336 |
+
max_length = 12000 if "wikipedia" in tool_name.lower() else 12000
|
| 3337 |
if len(tool_result) > max_length:
|
| 3338 |
original_length = len(tool_result)
|
| 3339 |
tool_result = tool_result[:max_length] + f"... [Result truncated from {original_length} to {max_length} chars to prevent memory issues]"
|
|
|
|
| 3591 |
|
| 3592 |
# Initialize tools
|
| 3593 |
WIKIPEDIA_TOOL = WikipediaSearchToolWithFAISS()
|
| 3594 |
+
SEARCH_TOOL = EnhancedDuckDuckGoSearchTool(max_results=3, max_chars_per_page=12000)
|
| 3595 |
YOUTUBE_TOOL = EnhancedYoutubeScreenshotQA()
|
| 3596 |
YOUTUBE_AUDIO_TOOL = YouTubeTranscriptExtractor()
|
| 3597 |
AUDIO_TRANSCRIPTION_TOOL = AudioTranscriptionTool()
|