Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -53,7 +53,7 @@ def extract_elements_and_formula(text: str) -> str:
|
|
| 53 |
)
|
| 54 |
|
| 55 |
@tool
|
| 56 |
-
def
|
| 57 |
"""A tool that uses DuckDuckGo to search for recent materials science information on a given topic.
|
| 58 |
|
| 59 |
Args:
|
|
@@ -100,7 +100,7 @@ with open("prompts.yaml", 'r') as stream:
|
|
| 100 |
|
| 101 |
agent = CodeAgent(
|
| 102 |
model=model,
|
| 103 |
-
tools=[final_answer], ## add your tools here (don't remove final answer)
|
| 104 |
max_steps=6,
|
| 105 |
verbosity_level=1,
|
| 106 |
grammar=None,
|
|
|
|
| 53 |
)
|
| 54 |
|
| 55 |
@tool
|
| 56 |
+
def search_tool_custom(query: str) -> str:
|
| 57 |
"""A tool that uses DuckDuckGo to search for recent materials science information on a given topic.
|
| 58 |
|
| 59 |
Args:
|
|
|
|
| 100 |
|
| 101 |
agent = CodeAgent(
|
| 102 |
model=model,
|
| 103 |
+
tools=[final_answer, extract_elements_and_formula, search_tool_custom], ## add your tools here (don't remove final answer)
|
| 104 |
max_steps=6,
|
| 105 |
verbosity_level=1,
|
| 106 |
grammar=None,
|