File transfer agent
Browse files
app.py
CHANGED
|
@@ -4,6 +4,7 @@ import arxiv
|
|
| 4 |
|
| 5 |
# import packages that are used in our tools
|
| 6 |
import requests
|
|
|
|
| 7 |
from bs4 import BeautifulSoup
|
| 8 |
from huggingface_hub import HfApi
|
| 9 |
from pypdf import PdfReader
|
|
@@ -93,15 +94,59 @@ def read_pdf_file(file_path: str) -> str:
|
|
| 93 |
return content
|
| 94 |
|
| 95 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 96 |
model_id = "Qwen/Qwen2.5-Coder-32B-Instruct"
|
| 97 |
hf_token = os.environ["HF_TOKEN"]
|
| 98 |
|
| 99 |
|
| 100 |
model = HfApiModel(model_id=model_id, token=hf_token)
|
| 101 |
-
agent = CodeAgent(tools=[
|
| 102 |
-
|
| 103 |
-
|
| 104 |
-
|
| 105 |
model=model,
|
| 106 |
add_base_tools=True)
|
| 107 |
|
|
|
|
| 4 |
|
| 5 |
# import packages that are used in our tools
|
| 6 |
import requests
|
| 7 |
+
from typing import Tuple
|
| 8 |
from bs4 import BeautifulSoup
|
| 9 |
from huggingface_hub import HfApi
|
| 10 |
from pypdf import PdfReader
|
|
|
|
| 94 |
return content
|
| 95 |
|
| 96 |
|
| 97 |
+
@tool
|
| 98 |
+
def process_file_transfer_result(log_files_path:) -> Tuple:
|
| 99 |
+
"""
|
| 100 |
+
This function processes the logs sent by the server.
|
| 101 |
+
Args:
|
| 102 |
+
log_files_path: The path to the logs.
|
| 103 |
+
"""
|
| 104 |
+
|
| 105 |
+
return True, ''
|
| 106 |
+
|
| 107 |
+
|
| 108 |
+
@tool
|
| 109 |
+
def request_file_transfer(file_path: str, server: str) -> bool:
|
| 110 |
+
"""
|
| 111 |
+
This function sends a request for a file transfer initiation.
|
| 112 |
+
Args:
|
| 113 |
+
file_path: The path to the source file.
|
| 114 |
+
"""
|
| 115 |
+
|
| 116 |
+
return True
|
| 117 |
+
|
| 118 |
+
|
| 119 |
+
@tool
|
| 120 |
+
def encrypt_file(file_path: str):
|
| 121 |
+
"""
|
| 122 |
+
This function encrypts the source file.
|
| 123 |
+
Args:
|
| 124 |
+
file_path: The path to the source file.
|
| 125 |
+
"""
|
| 126 |
+
|
| 127 |
+
return True
|
| 128 |
+
|
| 129 |
+
@tool
|
| 130 |
+
def validate_file(file_path: str) -> str:
|
| 131 |
+
"""
|
| 132 |
+
This function validates that the source file exists.
|
| 133 |
+
Args:
|
| 134 |
+
file_path: The path to the source file.
|
| 135 |
+
Returns:
|
| 136 |
+
A boolean value indicateing the existance of a file.
|
| 137 |
+
"""
|
| 138 |
+
|
| 139 |
+
return True
|
| 140 |
+
|
| 141 |
model_id = "Qwen/Qwen2.5-Coder-32B-Instruct"
|
| 142 |
hf_token = os.environ["HF_TOKEN"]
|
| 143 |
|
| 144 |
|
| 145 |
model = HfApiModel(model_id=model_id, token=hf_token)
|
| 146 |
+
agent = CodeAgent(tools=[validate_file,
|
| 147 |
+
encrypt_file,
|
| 148 |
+
request_file_transfer,
|
| 149 |
+
process_file_transfer_result],
|
| 150 |
model=model,
|
| 151 |
add_base_tools=True)
|
| 152 |
|