InnaV commited on
Commit
67d8b62
·
verified ·
1 Parent(s): 56aca12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -14
app.py CHANGED
@@ -9,19 +9,7 @@ from bertopic import BERTopic
9
  from Gradio_UI import GradioUI
10
 
11
  @tool
12
- def my_custom_tool(text: str) -> str:
13
- """A tool that processes input text and returns a modified version.
14
-
15
- Args:
16
- text (str): The input text to process.
17
-
18
- Returns:
19
- str: The processed text output.
20
- """
21
- return f"Processed text: {text}"
22
-
23
- @tool
24
- def extract_main_topic(text: str) -> str:
25
  """Extracts the main topic from a given text using BERTopic.
26
 
27
  Args:
@@ -64,7 +52,7 @@ with open("prompts.yaml", 'r') as stream:
64
 
65
  agent = CodeAgent(
66
  model=model,
67
- tools=[final_answer, my_custom_tool, extract_main_topic],
68
  max_steps=6,
69
  verbosity_level=1,
70
  grammar=None,
 
9
  from Gradio_UI import GradioUI
10
 
11
  @tool
12
+ def extract_main_topic(arg1: str) -> str:
 
 
 
 
 
 
 
 
 
 
 
 
13
  """Extracts the main topic from a given text using BERTopic.
14
 
15
  Args:
 
52
 
53
  agent = CodeAgent(
54
  model=model,
55
+ tools=[final_answer, extract_main_topic],
56
  max_steps=6,
57
  verbosity_level=1,
58
  grammar=None,