Spaces:
Sleeping
Sleeping
Alexandre Gazola
commited on
Commit
·
0203a60
1
Parent(s):
770ccfe
fix
Browse files- count_max_bird_species_tool.py +1 -1
- langchain_agent.py +2 -1
count_max_bird_species_tool.py
CHANGED
@@ -7,7 +7,7 @@ from utils import decode_base64_to_frame
|
|
7 |
@tool
|
8 |
def count_max_bird_species_in_video(youtube_url: str) -> int:
|
9 |
"""
|
10 |
-
Count the maximum number of bird species to be on camera simultaneously in a YouTube video
|
11 |
|
12 |
Args:
|
13 |
- youtube_url: str: the URL of the YouTube video.
|
|
|
7 |
@tool
|
8 |
def count_max_bird_species_in_video(youtube_url: str) -> int:
|
9 |
"""
|
10 |
+
Count the maximum number of bird species to be on camera simultaneously in a YouTube video.
|
11 |
|
12 |
Args:
|
13 |
- youtube_url: str: the URL of the YouTube video.
|
langchain_agent.py
CHANGED
@@ -50,7 +50,8 @@ class LangChainAgent:
|
|
50 |
agent=agent,
|
51 |
tools=tools,
|
52 |
verbose=True,
|
53 |
-
|
|
|
54 |
|
55 |
def __call__(self, question: str) -> str:
|
56 |
print(f"LangChain agent received: {question[:50]}...")
|
|
|
50 |
agent=agent,
|
51 |
tools=tools,
|
52 |
verbose=True,
|
53 |
+
return_intermediate_steps=True,
|
54 |
+
max_iterations=3)
|
55 |
|
56 |
def __call__(self, question: str) -> str:
|
57 |
print(f"LangChain agent received: {question[:50]}...")
|