generated from kyegomez/Python-Package-Template
-
-
Notifications
You must be signed in to change notification settings - Fork 9
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Kye
committed
Apr 12, 2024
1 parent
20c8334
commit 6c83400
Showing
4 changed files
with
92 additions
and
42 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,75 @@ | ||
from swarms import tool | ||
import subprocess | ||
|
||
|
||
# Tools | ||
@tool | ||
def terminal( | ||
code: str, | ||
): | ||
""" | ||
Run code in the terminal. | ||
Args: | ||
code (str): The code to run in the terminal. | ||
Returns: | ||
str: The output of the code. | ||
""" | ||
out = subprocess.run( | ||
code, shell=True, capture_output=True, text=True | ||
).stdout | ||
return str(out) | ||
|
||
|
||
@tool | ||
def browser(query: str): | ||
""" | ||
Search the query in the browser with the `browser` tool. | ||
Args: | ||
query (str): The query to search in the browser. | ||
Returns: | ||
str: The search results. | ||
""" | ||
import webbrowser | ||
|
||
url = f"https://www.google.com/search?q={query}" | ||
webbrowser.open(url) | ||
return f"Searching for {query} in the browser." | ||
|
||
|
||
@tool | ||
def create_file(file_path: str, content: str): | ||
""" | ||
Create a file using the file editor tool. | ||
Args: | ||
file_path (str): The path to the file. | ||
content (str): The content to write to the file. | ||
Returns: | ||
str: The result of the file creation operation. | ||
""" | ||
with open(file_path, "w") as file: | ||
file.write(content) | ||
return f"File {file_path} created successfully." | ||
|
||
|
||
@tool | ||
def file_editor(file_path: str, mode: str, content: str): | ||
""" | ||
Edit a file using the file editor tool. | ||
Args: | ||
file_path (str): The path to the file. | ||
mode (str): The mode to open the file in. | ||
content (str): The content to write to the file. | ||
Returns: | ||
str: The result of the file editing operation. | ||
""" | ||
with open(file_path, mode) as file: | ||
file.write(content) | ||
return f"File {file_path} edited successfully." |