Исправление проблемы отображения графов: добавление поля predefined_graphs и прокси-сервер для модификации ответов API
Browse files- app.py +181 -338
- proxy_server.py +229 -0
app.py
CHANGED
@@ -14,10 +14,7 @@ import urllib.error
|
|
14 |
import tempfile
|
15 |
import http.server
|
16 |
import socketserver
|
17 |
-
import
|
18 |
-
from threading import Thread
|
19 |
-
from http import HTTPStatus
|
20 |
-
from typing import Dict, Any, Optional
|
21 |
|
22 |
# Настройка логирования
|
23 |
logging.basicConfig(level=logging.INFO,
|
@@ -35,263 +32,6 @@ CHAT_AGENT_JSON = AGENTS_DIR / "chat_agent.json"
|
|
35 |
API_BINARY = Path("/app/server/bin/api")
|
36 |
PLAYGROUND_DIR = Path("/app/playground")
|
37 |
BACKUP_DIR = Path("/app/backup")
|
38 |
-
MOCK_API_DIR = Path("/app/mock-api")
|
39 |
-
|
40 |
-
# Загружаем мок-ответы для дизайнера
|
41 |
-
DESIGNER_PACKAGES_JSON = MOCK_API_DIR / "designer-packages.json"
|
42 |
-
|
43 |
-
# Мок-данные для ответов дизайнера
|
44 |
-
mock_responses = {}
|
45 |
-
|
46 |
-
def load_mock_responses():
|
47 |
-
"""Загружает предварительно созданные мок-ответы"""
|
48 |
-
global mock_responses
|
49 |
-
|
50 |
-
try:
|
51 |
-
if DESIGNER_PACKAGES_JSON.exists():
|
52 |
-
with open(DESIGNER_PACKAGES_JSON, 'r') as f:
|
53 |
-
mock_responses['packages'] = json.load(f)
|
54 |
-
logger.info(f"Загружен мок-ответ для дизайнера: {mock_responses['packages']}")
|
55 |
-
except Exception as e:
|
56 |
-
logger.error(f"Ошибка при загрузке мок-ответов: {e}")
|
57 |
-
# Создаём запасной ответ
|
58 |
-
mock_responses['packages'] = {
|
59 |
-
"success": True,
|
60 |
-
"packages": [
|
61 |
-
{
|
62 |
-
"name": "default",
|
63 |
-
"description": "Default package",
|
64 |
-
"graphs": [
|
65 |
-
{
|
66 |
-
"name": "Voice Agent",
|
67 |
-
"description": "Voice Agent with OpenAI",
|
68 |
-
"file": "voice_agent.json",
|
69 |
-
"id": "voice_agent",
|
70 |
-
"package": "default"
|
71 |
-
},
|
72 |
-
{
|
73 |
-
"name": "Chat Agent",
|
74 |
-
"description": "Chat Agent",
|
75 |
-
"file": "chat_agent.json",
|
76 |
-
"id": "chat_agent",
|
77 |
-
"package": "default"
|
78 |
-
}
|
79 |
-
]
|
80 |
-
}
|
81 |
-
]
|
82 |
-
}
|
83 |
-
|
84 |
-
class ProxyHTTPRequestHandler(http.server.BaseHTTPRequestHandler):
|
85 |
-
"""HTTP-прокси для отладки и исправления запросов между UI и API"""
|
86 |
-
|
87 |
-
def __init__(self, *args, **kwargs):
|
88 |
-
# Целевой API сервер
|
89 |
-
self.api_host = "localhost"
|
90 |
-
self.api_port = 8080
|
91 |
-
super().__init__(*args, **kwargs)
|
92 |
-
|
93 |
-
def do_GET(self):
|
94 |
-
"""Обработка GET запросов"""
|
95 |
-
logger.info(f"PROXY: GET запрос: {self.path}")
|
96 |
-
|
97 |
-
# Специальная обработка для запроса графов
|
98 |
-
if self.path == "/graphs" or self.path == "/api/agents/graphs":
|
99 |
-
self._handle_graphs_request()
|
100 |
-
return
|
101 |
-
|
102 |
-
# Проксирование запроса к API
|
103 |
-
target_url = f"http://{self.api_host}:{self.api_port}{self.path}"
|
104 |
-
try:
|
105 |
-
response = requests.get(target_url, headers=self._get_headers())
|
106 |
-
self._send_response(response)
|
107 |
-
except Exception as e:
|
108 |
-
logger.error(f"PROXY: Ошибка при проксировании GET-запроса: {e}")
|
109 |
-
self.send_error(HTTPStatus.INTERNAL_SERVER_ERROR, str(e))
|
110 |
-
|
111 |
-
def do_POST(self):
|
112 |
-
"""Обработка POST запросов"""
|
113 |
-
content_length = int(self.headers.get('Content-Length', 0))
|
114 |
-
post_data = self.rfile.read(content_length) if content_length > 0 else b''
|
115 |
-
|
116 |
-
logger.info(f"PROXY: POST запрос: {self.path}")
|
117 |
-
|
118 |
-
# Специальные запросы для дизайнера
|
119 |
-
if "/api/designer/v1/packages/reload" in self.path or "/api/dev/v1/packages/reload" in self.path:
|
120 |
-
self._handle_designer_reload()
|
121 |
-
return
|
122 |
-
|
123 |
-
# Проксирование запроса к API
|
124 |
-
target_url = f"http://{self.api_host}:{self.api_port}{self.path}"
|
125 |
-
try:
|
126 |
-
response = requests.post(target_url, data=post_data, headers=self._get_headers())
|
127 |
-
self._send_response(response)
|
128 |
-
except Exception as e:
|
129 |
-
logger.error(f"PROXY: Ошибка при проксировании POST-запроса: {e}")
|
130 |
-
self.send_error(HTTPStatus.INTERNAL_SERVER_ERROR, str(e))
|
131 |
-
|
132 |
-
def _handle_graphs_request(self):
|
133 |
-
"""Специальная обработка для запроса графов"""
|
134 |
-
try:
|
135 |
-
# Читаем графы из property.json
|
136 |
-
with open(PROPERTY_JSON, 'r') as f:
|
137 |
-
property_data = json.load(f)
|
138 |
-
|
139 |
-
graphs = property_data.get('graphs', [])
|
140 |
-
logger.info(f"PROXY: Возвращаем графы напрямую: {json.dumps(graphs)}")
|
141 |
-
|
142 |
-
self.send_response(HTTPStatus.OK)
|
143 |
-
self.send_header('Content-Type', 'application/json')
|
144 |
-
self.end_headers()
|
145 |
-
self.wfile.write(json.dumps(graphs).encode('utf-8'))
|
146 |
-
except Exception as e:
|
147 |
-
logger.error(f"PROXY: Ошибка при обработке запроса графов: {e}")
|
148 |
-
self.send_error(HTTPStatus.INTERNAL_SERVER_ERROR, str(e))
|
149 |
-
|
150 |
-
def _handle_designer_reload(self):
|
151 |
-
"""Обработка запросов дизайнера"""
|
152 |
-
# Новый формат ответа для designer API
|
153 |
-
response_data = {
|
154 |
-
"status": 200,
|
155 |
-
"packages": [
|
156 |
-
{
|
157 |
-
"name": "default",
|
158 |
-
"description": "Default package",
|
159 |
-
"agents": [
|
160 |
-
{
|
161 |
-
"name": "voice_agent",
|
162 |
-
"description": "A simple voice agent"
|
163 |
-
},
|
164 |
-
{
|
165 |
-
"name": "chat_agent",
|
166 |
-
"description": "A simple chat agent"
|
167 |
-
}
|
168 |
-
],
|
169 |
-
"graphs": []
|
170 |
-
}
|
171 |
-
]
|
172 |
-
}
|
173 |
-
|
174 |
-
# Читаем графы из property.json
|
175 |
-
try:
|
176 |
-
with open(PROPERTY_JSON, 'r') as f:
|
177 |
-
property_data = json.load(f)
|
178 |
-
|
179 |
-
graphs = property_data.get('graphs', [])
|
180 |
-
# Обеспечиваем форматирование совместимое с designer API
|
181 |
-
formatted_graphs = []
|
182 |
-
for graph in graphs:
|
183 |
-
# Копируем граф и добавляем дополнительные поля, которые могут ожидаться
|
184 |
-
graph_copy = graph.copy()
|
185 |
-
if 'id' not in graph_copy:
|
186 |
-
graph_copy['id'] = graph_copy.get('name', '').lower().replace(' ', '_')
|
187 |
-
if 'package' not in graph_copy:
|
188 |
-
graph_copy['package'] = 'default'
|
189 |
-
formatted_graphs.append(graph_copy)
|
190 |
-
|
191 |
-
response_data["packages"][0]["graphs"] = formatted_graphs
|
192 |
-
|
193 |
-
# Добавим альтернативные форматы в лог
|
194 |
-
alternative_format = {
|
195 |
-
"success": True,
|
196 |
-
"packages": [
|
197 |
-
{
|
198 |
-
"name": "default",
|
199 |
-
"description": "Default package",
|
200 |
-
"graphs": formatted_graphs
|
201 |
-
}
|
202 |
-
]
|
203 |
-
}
|
204 |
-
|
205 |
-
alternate_format2 = formatted_graphs
|
206 |
-
|
207 |
-
logger.info(f"PROXY: Ответ для designer API, формат 1: {json.dumps(response_data)}")
|
208 |
-
logger.info(f"PROXY: Ответ для designer API, формат 2: {json.dumps(alternative_format)}")
|
209 |
-
logger.info(f"PROXY: Ответ для designer API, формат 3: {json.dumps(alternate_format2)}")
|
210 |
-
|
211 |
-
# Тестируем разные варианты ответов
|
212 |
-
# 1. Отправляем третий формат (простой массив графов)
|
213 |
-
self.send_response(HTTPStatus.OK)
|
214 |
-
self.send_header('Content-Type', 'application/json')
|
215 |
-
self.end_headers()
|
216 |
-
self.wfile.write(json.dumps(alternate_format2).encode('utf-8'))
|
217 |
-
|
218 |
-
# В следующих вызовах попробуем другие форматы
|
219 |
-
|
220 |
-
except Exception as e:
|
221 |
-
logger.error(f"PROXY: Ошибка при чтении графов для дизайнера: {e}")
|
222 |
-
self.send_error(HTTPStatus.INTERNAL_SERVER_ERROR, str(e))
|
223 |
-
|
224 |
-
def _get_headers(self):
|
225 |
-
"""Получение заголовков для проксирования"""
|
226 |
-
headers = {}
|
227 |
-
for header in self.headers:
|
228 |
-
headers[header] = self.headers[header]
|
229 |
-
return headers
|
230 |
-
|
231 |
-
def _send_response(self, response):
|
232 |
-
"""Отправка ответа клиенту"""
|
233 |
-
self.send_response(response.status_code)
|
234 |
-
|
235 |
-
# Копируем заголовки
|
236 |
-
for header, value in response.headers.items():
|
237 |
-
self.send_header(header, value)
|
238 |
-
self.end_headers()
|
239 |
-
|
240 |
-
# Отправляем тело ответа
|
241 |
-
self.wfile.write(response.content)
|
242 |
-
|
243 |
-
def log_message(self, format, *args):
|
244 |
-
"""Переопределение логирования"""
|
245 |
-
logger.debug(f"PROXY: {self.address_string()} - {format % args}")
|
246 |
-
|
247 |
-
class MockDesignerAPIHandler(http.server.BaseHTTPRequestHandler):
|
248 |
-
"""Обработчик для мок-API дизайнера"""
|
249 |
-
|
250 |
-
def do_GET(self):
|
251 |
-
"""Обработка GET запросов"""
|
252 |
-
logger.info(f"MOCK API: GET запрос: {self.path}")
|
253 |
-
|
254 |
-
if self.path == "/api/designer/v1/packages" or self.path == "/api/dev/v1/packages":
|
255 |
-
self._send_designer_packages()
|
256 |
-
elif self.path.startswith("/api/designer/v1/") or self.path.startswith("/api/dev/v1/"):
|
257 |
-
self._send_success_response({"success": True})
|
258 |
-
else:
|
259 |
-
self.send_error(404, "Endpoint not found")
|
260 |
-
|
261 |
-
def do_POST(self):
|
262 |
-
"""Обработка POST запросов"""
|
263 |
-
logger.info(f"MOCK API: POST запрос: {self.path}")
|
264 |
-
|
265 |
-
if self.path == "/api/designer/v1/packages/reload" or self.path == "/api/dev/v1/packages/reload":
|
266 |
-
self._send_designer_packages()
|
267 |
-
elif self.path.startswith("/api/designer/v1/") or self.path.startswith("/api/dev/v1/"):
|
268 |
-
self._send_success_response({"success": True})
|
269 |
-
else:
|
270 |
-
self.send_error(404, "Endpoint not found")
|
271 |
-
|
272 |
-
def _send_designer_packages(self):
|
273 |
-
"""Отправляет информацию о пакетах и графах"""
|
274 |
-
self._send_success_response(mock_responses['packages'])
|
275 |
-
|
276 |
-
def _send_success_response(self, data: Dict[str, Any]):
|
277 |
-
"""Отправляет успешный ответ с данными"""
|
278 |
-
self.send_response(200)
|
279 |
-
self.send_header('Content-Type', 'application/json')
|
280 |
-
self.end_headers()
|
281 |
-
self.wfile.write(json.dumps(data).encode('utf-8'))
|
282 |
-
|
283 |
-
def log_message(self, format, *args):
|
284 |
-
"""Настраиваем логирование для сервера"""
|
285 |
-
logger.debug(f"MOCK API: {self.address_string()} - {format % args}")
|
286 |
-
|
287 |
-
def run_proxy_server(port=9090):
|
288 |
-
"""Запуск прокси-сервера"""
|
289 |
-
try:
|
290 |
-
with socketserver.TCPServer(("", port), ProxyHTTPRequestHandler) as httpd:
|
291 |
-
logger.info(f"Запуск прокси-сервера на порту {port}")
|
292 |
-
httpd.serve_forever()
|
293 |
-
except Exception as e:
|
294 |
-
logger.error(f"Ошибка при запуске прокси-сервера: {e}")
|
295 |
|
296 |
def ensure_directory_permissions(directory_path):
|
297 |
"""Обеспечиваем правильные разрешения для директории"""
|
@@ -320,40 +60,107 @@ def backup_file(filepath):
|
|
320 |
except Exception as e:
|
321 |
logger.error(f"Ошибка при создании резервной копии {filepath}: {e}")
|
322 |
|
323 |
-
def
|
324 |
-
"""Проверяет
|
|
|
|
|
|
|
|
|
325 |
try:
|
326 |
-
|
327 |
-
|
328 |
-
|
329 |
-
|
330 |
-
|
331 |
-
|
332 |
-
|
333 |
-
|
334 |
-
|
335 |
-
|
336 |
-
|
337 |
-
|
338 |
-
|
339 |
-
|
340 |
-
|
341 |
-
|
342 |
-
|
343 |
-
|
344 |
-
|
345 |
-
|
346 |
-
|
347 |
-
|
348 |
-
|
349 |
-
|
350 |
-
|
351 |
-
|
352 |
-
|
353 |
-
|
354 |
-
logger.info(f"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
355 |
except Exception as e:
|
356 |
-
logger.error(f"Ошибка при
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
357 |
|
358 |
def check_and_create_agent_files():
|
359 |
"""Проверяет наличие всех необходимых файлов агентов и создает их при необходимости"""
|
@@ -448,6 +255,41 @@ def check_files():
|
|
448 |
subprocess.run(["stat", str(AGENTS_DIR)])
|
449 |
subprocess.run(["stat", str(PROPERTY_JSON)])
|
450 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
451 |
def test_api():
|
452 |
"""Делает запрос к API для получения списка графов"""
|
453 |
logger.info("=== Тестирование API ===")
|
@@ -458,47 +300,46 @@ def test_api():
|
|
458 |
data = response.read().decode('utf-8')
|
459 |
logger.info(f"Ответ /graphs: {data}")
|
460 |
|
461 |
-
#
|
462 |
-
|
463 |
-
|
464 |
-
|
|
|
|
|
|
|
|
|
465 |
logger.info(f"API вернул {len(json_data)} графов")
|
466 |
-
# Если API вернул пустой список, исправляем это
|
467 |
-
if len(json_data) == 0:
|
468 |
-
logger.warning("API вернул пустой список графов, исправляем property.json")
|
469 |
-
backup_file(PROPERTY_JSON)
|
470 |
-
ensure_property_json()
|
471 |
-
check_and_create_agent_files()
|
472 |
-
ensure_directory_permissions(AGENTS_DIR)
|
473 |
-
# Перезапускаем API сервер
|
474 |
-
logger.info("Перезапускаем API сервер...")
|
475 |
-
subprocess.run(["pkill", "-f", str(API_BINARY)])
|
476 |
-
time.sleep(1)
|
477 |
-
subprocess.Popen([str(API_BINARY)])
|
478 |
else:
|
479 |
logger.warning("API вернул пустой список графов")
|
480 |
-
|
481 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
482 |
except urllib.error.URLError as e:
|
483 |
logger.error(f"Ошибка запроса к API: {e}")
|
484 |
except Exception as e:
|
485 |
logger.error(f"Неизвестная ошибка при запросе к API: {e}")
|
486 |
|
487 |
-
def run_mock_api_server(port=8090):
|
488 |
-
"""Запускает мок-сервер для API дизайнера"""
|
489 |
-
try:
|
490 |
-
with socketserver.TCPServer(("", port), MockDesignerAPIHandler) as httpd:
|
491 |
-
logger.info(f"Запуск мок-API сервера на порту {port}")
|
492 |
-
httpd.serve_forever()
|
493 |
-
except Exception as e:
|
494 |
-
logger.error(f"Ошибка при запуске мок-API сервера: {e}")
|
495 |
-
|
496 |
def main():
|
497 |
processes = []
|
498 |
try:
|
499 |
-
# Загружаем мок-ответы
|
500 |
-
load_mock_responses()
|
501 |
-
|
502 |
# Проверяем существование файлов
|
503 |
if not API_BINARY.exists():
|
504 |
logger.error(f"API binary не найден: {API_BINARY}")
|
@@ -513,7 +354,10 @@ def main():
|
|
513 |
ensure_directory_permissions(BACKUP_DIR)
|
514 |
|
515 |
# Проверяем и создаем property.json
|
516 |
-
|
|
|
|
|
|
|
517 |
|
518 |
# Проверяем и создаем файлы агентов
|
519 |
check_and_create_agent_files()
|
@@ -521,12 +365,6 @@ def main():
|
|
521 |
# Проверка файлов перед запуском
|
522 |
check_files()
|
523 |
|
524 |
-
# Запускаем мок-сервер для API дизайнера
|
525 |
-
mock_api_port = 8090
|
526 |
-
mock_api_thread = threading.Thread(target=run_mock_api_server, args=(mock_api_port,))
|
527 |
-
mock_api_thread.daemon = True
|
528 |
-
mock_api_thread.start()
|
529 |
-
|
530 |
# Запускаем API сервер
|
531 |
logger.info("Запуск TEN-Agent API сервера на порту 8080...")
|
532 |
api_process = subprocess.Popen([str(API_BINARY)])
|
@@ -537,23 +375,28 @@ def main():
|
|
537 |
test_thread.daemon = True
|
538 |
test_thread.start()
|
539 |
|
540 |
-
# Запускаем прокси-сервер
|
541 |
proxy_port = 9090
|
542 |
-
|
543 |
-
|
544 |
-
|
|
|
|
|
|
|
|
|
|
|
545 |
|
546 |
# Запускаем Playground UI в режиме dev на порту 7860 (порт Hugging Face)
|
547 |
logger.info("Запуск Playground UI в режиме разработки на порту 7860...")
|
548 |
os.environ["PORT"] = "7860"
|
549 |
-
os.environ["AGENT_SERVER_URL"] = f"http://localhost:{proxy_port}" #
|
550 |
os.environ["NEXT_PUBLIC_EDIT_GRAPH_MODE"] = "true" # Включаем расширенный режим редактирования
|
551 |
os.environ["NEXT_PUBLIC_DISABLE_CAMERA"] = "true" # Отключаем запрос на использование камеры
|
552 |
|
553 |
-
#
|
554 |
-
os.environ["NEXT_PUBLIC_DEV_MODE"] = "
|
555 |
os.environ["NEXT_PUBLIC_API_BASE_URL"] = "/api/agents"
|
556 |
-
os.environ["NEXT_PUBLIC_DESIGNER_API_URL"] = f"http://localhost:{
|
557 |
|
558 |
# Запускаем Playground UI
|
559 |
playground_process = subprocess.Popen(
|
|
|
14 |
import tempfile
|
15 |
import http.server
|
16 |
import socketserver
|
17 |
+
from typing import Dict, Any, Optional, Union
|
|
|
|
|
|
|
18 |
|
19 |
# Настройка логирования
|
20 |
logging.basicConfig(level=logging.INFO,
|
|
|
32 |
API_BINARY = Path("/app/server/bin/api")
|
33 |
PLAYGROUND_DIR = Path("/app/playground")
|
34 |
BACKUP_DIR = Path("/app/backup")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
35 |
|
36 |
def ensure_directory_permissions(directory_path):
|
37 |
"""Обеспечиваем правильные разрешения для директории"""
|
|
|
60 |
except Exception as e:
|
61 |
logger.error(f"Ошибка при создании резервной копии {filepath}: {e}")
|
62 |
|
63 |
+
def update_property_json():
|
64 |
+
"""Проверяет существующий property.json и добавляет поле predefined_graphs, если его нет"""
|
65 |
+
if not PROPERTY_JSON.exists():
|
66 |
+
logger.info(f"{PROPERTY_JSON} не существует, нет необходимости в обновлении")
|
67 |
+
return False
|
68 |
+
|
69 |
try:
|
70 |
+
with open(PROPERTY_JSON, 'r') as f:
|
71 |
+
data = json.load(f)
|
72 |
+
|
73 |
+
if "predefined_graphs" in data:
|
74 |
+
logger.info(f"{PROPERTY_JSON} уже содержит поле predefined_graphs")
|
75 |
+
return False
|
76 |
+
|
77 |
+
# Создаем резервную копию перед изменением
|
78 |
+
backup_file(PROPERTY_JSON)
|
79 |
+
|
80 |
+
# Добавляем поле predefined_graphs на основе поля graphs или создаем его
|
81 |
+
if "graphs" in data:
|
82 |
+
data["predefined_graphs"] = data["graphs"]
|
83 |
+
logger.info(f"Добавлено поле predefined_graphs на основе поля graphs")
|
84 |
+
else:
|
85 |
+
# Создаем стандартные графы
|
86 |
+
data["predefined_graphs"] = [
|
87 |
+
{
|
88 |
+
"name": "Voice Agent",
|
89 |
+
"description": "Basic voice agent with OpenAI",
|
90 |
+
"file": "voice_agent.json"
|
91 |
+
},
|
92 |
+
{
|
93 |
+
"name": "Chat Agent",
|
94 |
+
"description": "Simple chat agent",
|
95 |
+
"file": "chat_agent.json"
|
96 |
+
}
|
97 |
+
]
|
98 |
+
logger.info(f"Добавлено стандартное поле predefined_graphs")
|
99 |
+
|
100 |
+
# Сохраняем обновленный файл
|
101 |
+
with open(PROPERTY_JSON, 'w') as f:
|
102 |
+
json.dump(data, f, indent=2)
|
103 |
+
|
104 |
+
logger.info(f"{PROPERTY_JSON} успешно обновлен с добавлением поля predefined_graphs")
|
105 |
+
return True
|
106 |
except Exception as e:
|
107 |
+
logger.error(f"Ошибка при обновлении {PROPERTY_JSON}: {e}")
|
108 |
+
return False
|
109 |
+
|
110 |
+
def check_and_create_property_json():
|
111 |
+
"""Проверяет наличие property.json и создает его при необходимости"""
|
112 |
+
if not PROPERTY_JSON.exists():
|
113 |
+
logger.warning(f"{PROPERTY_JSON} не найден, создаем файл...")
|
114 |
+
|
115 |
+
property_data = {
|
116 |
+
"_ten": {}, # Важное поле для TEN формата
|
117 |
+
"name": "TEN Agent Example",
|
118 |
+
"version": "0.0.1",
|
119 |
+
"extensions": ["openai_chatgpt"],
|
120 |
+
"description": "A basic voice agent with OpenAI",
|
121 |
+
"predefined_graphs": [ # Важное поле, которое ожидает API
|
122 |
+
{
|
123 |
+
"name": "Voice Agent",
|
124 |
+
"description": "Basic voice agent with OpenAI",
|
125 |
+
"file": "voice_agent.json"
|
126 |
+
},
|
127 |
+
{
|
128 |
+
"name": "Chat Agent",
|
129 |
+
"description": "Simple chat agent",
|
130 |
+
"file": "chat_agent.json"
|
131 |
+
}
|
132 |
+
],
|
133 |
+
"graphs": [
|
134 |
+
{
|
135 |
+
"name": "Voice Agent",
|
136 |
+
"description": "Basic voice agent with OpenAI",
|
137 |
+
"file": "voice_agent.json"
|
138 |
+
},
|
139 |
+
{
|
140 |
+
"name": "Chat Agent",
|
141 |
+
"description": "Simple chat agent",
|
142 |
+
"file": "chat_agent.json"
|
143 |
+
}
|
144 |
+
]
|
145 |
+
}
|
146 |
+
|
147 |
+
# Проверяем и создаем директории
|
148 |
+
PROPERTY_JSON.parent.mkdir(parents=True, exist_ok=True)
|
149 |
+
|
150 |
+
# Создаем временный файл и затем перемещаем его
|
151 |
+
with tempfile.NamedTemporaryFile(mode='w', delete=False) as temp_file:
|
152 |
+
json.dump(property_data, temp_file, indent=2)
|
153 |
+
temp_path = temp_file.name
|
154 |
+
|
155 |
+
# Копируем временный файл в целевой
|
156 |
+
try:
|
157 |
+
shutil.copy2(temp_path, PROPERTY_JSON)
|
158 |
+
os.chmod(PROPERTY_JSON, 0o666) # Устанавливаем права доступа rw-rw-rw-
|
159 |
+
logger.info(f"Файл {PROPERTY_JSON} создан успешно")
|
160 |
+
except Exception as e:
|
161 |
+
logger.error(f"Ошибка при создании {PROPERTY_JSON}: {e}")
|
162 |
+
finally:
|
163 |
+
os.unlink(temp_path) # Удаляем временный файл
|
164 |
|
165 |
def check_and_create_agent_files():
|
166 |
"""Проверяет наличие всех необходимых файлов агентов и создает их при необходимости"""
|
|
|
255 |
subprocess.run(["stat", str(AGENTS_DIR)])
|
256 |
subprocess.run(["stat", str(PROPERTY_JSON)])
|
257 |
|
258 |
+
def analyze_api_response(response_data):
|
259 |
+
"""Анализирует ответ API и выводит диагностическую информацию"""
|
260 |
+
try:
|
261 |
+
# Пробуем распарсить JSON
|
262 |
+
if not response_data or response_data.strip() == "":
|
263 |
+
logger.error("API вернул пустой ответ")
|
264 |
+
return None
|
265 |
+
|
266 |
+
json_data = json.loads(response_data)
|
267 |
+
|
268 |
+
# Проверяем тип данных
|
269 |
+
if isinstance(json_data, list):
|
270 |
+
logger.info(f"API вернул список с {len(json_data)} элементами")
|
271 |
+
if len(json_data) > 0:
|
272 |
+
logger.info(f"Структура первого элемента: {json.dumps(json_data[0], indent=2)}")
|
273 |
+
return json_data
|
274 |
+
elif isinstance(json_data, dict):
|
275 |
+
logger.info(f"API вернул словарь с ключами: {list(json_data.keys())}")
|
276 |
+
# Проверяем наличие кода ошибки
|
277 |
+
if "code" in json_data:
|
278 |
+
logger.error(f"API вернул ошибку с кодом: {json_data['code']}")
|
279 |
+
if "msg" in json_data:
|
280 |
+
logger.error(f"Сообщение ошибки: {json_data['msg']}")
|
281 |
+
return json_data
|
282 |
+
else:
|
283 |
+
logger.warning(f"API вернул неожиданный тип данных: {type(json_data)}")
|
284 |
+
return json_data
|
285 |
+
except json.JSONDecodeError as e:
|
286 |
+
logger.error(f"Ошибка декодирования JSON: {e}")
|
287 |
+
logger.error(f"Сырые данные: {response_data}")
|
288 |
+
return None
|
289 |
+
except Exception as e:
|
290 |
+
logger.error(f"Ошибка при анализе ответа API: {e}")
|
291 |
+
return None
|
292 |
+
|
293 |
def test_api():
|
294 |
"""Делает запрос к API для получения списка графов"""
|
295 |
logger.info("=== Тестирование API ===")
|
|
|
300 |
data = response.read().decode('utf-8')
|
301 |
logger.info(f"Ответ /graphs: {data}")
|
302 |
|
303 |
+
# Анализируем ответ API
|
304 |
+
json_data = analyze_api_response(data)
|
305 |
+
|
306 |
+
# Проверяем структуру ответа и реагируем на проблемы
|
307 |
+
if json_data is None:
|
308 |
+
logger.error("Не удалось проанализировать ответ API")
|
309 |
+
elif isinstance(json_data, list):
|
310 |
+
if len(json_data) > 0:
|
311 |
logger.info(f"API вернул {len(json_data)} графов")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
312 |
else:
|
313 |
logger.warning("API вернул пустой список графов")
|
314 |
+
# Пробуем исправить property.json, если он уже существует
|
315 |
+
if PROPERTY_JSON.exists():
|
316 |
+
logger.info("Пробуем обновить существующий property.json...")
|
317 |
+
if update_property_json():
|
318 |
+
# Перезапускаем API сервер после обновления
|
319 |
+
logger.info("Перезапускаем API сервер после обновления property.json...")
|
320 |
+
subprocess.run(["pkill", "-f", str(API_BINARY)])
|
321 |
+
time.sleep(1)
|
322 |
+
subprocess.Popen([str(API_BINARY)])
|
323 |
+
elif isinstance(json_data, dict) and "code" in json_data:
|
324 |
+
logger.warning("API вернул ошибку, исправляем property.json")
|
325 |
+
backup_file(PROPERTY_JSON)
|
326 |
+
check_and_create_property_json()
|
327 |
+
update_property_json() # Добавляем вызов функции обновления
|
328 |
+
check_and_create_agent_files()
|
329 |
+
ensure_directory_permissions(AGENTS_DIR)
|
330 |
+
# Перезапускаем API сервер
|
331 |
+
logger.info("Перезапускаем API сервер...")
|
332 |
+
subprocess.run(["pkill", "-f", str(API_BINARY)])
|
333 |
+
time.sleep(1)
|
334 |
+
subprocess.Popen([str(API_BINARY)])
|
335 |
except urllib.error.URLError as e:
|
336 |
logger.error(f"Ошибка запроса к API: {e}")
|
337 |
except Exception as e:
|
338 |
logger.error(f"Неизвестная ошибка при запросе к API: {e}")
|
339 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
340 |
def main():
|
341 |
processes = []
|
342 |
try:
|
|
|
|
|
|
|
343 |
# Проверяем существование файлов
|
344 |
if not API_BINARY.exists():
|
345 |
logger.error(f"API binary не найден: {API_BINARY}")
|
|
|
354 |
ensure_directory_permissions(BACKUP_DIR)
|
355 |
|
356 |
# Проверяем и создаем property.json
|
357 |
+
check_and_create_property_json()
|
358 |
+
|
359 |
+
# Проверяем и обновляем существующий property.json, если он не содержит predefined_graphs
|
360 |
+
update_property_json()
|
361 |
|
362 |
# Проверяем и создаем файлы агентов
|
363 |
check_and_create_agent_files()
|
|
|
365 |
# Проверка файлов перед запуском
|
366 |
check_files()
|
367 |
|
|
|
|
|
|
|
|
|
|
|
|
|
368 |
# Запускаем API сервер
|
369 |
logger.info("Запуск TEN-Agent API сервера на порту 8080...")
|
370 |
api_process = subprocess.Popen([str(API_BINARY)])
|
|
|
375 |
test_thread.daemon = True
|
376 |
test_thread.start()
|
377 |
|
378 |
+
# Запускаем прокси-сервер
|
379 |
proxy_port = 9090
|
380 |
+
proxy_script = Path("proxy_server.py")
|
381 |
+
if proxy_script.exists():
|
382 |
+
logger.info(f"Запуск прокси-сервера на порту {proxy_port}")
|
383 |
+
proxy_process = subprocess.Popen([sys.executable, str(proxy_script)], env=dict(os.environ, PROXY_PORT=str(proxy_port)))
|
384 |
+
processes.append(proxy_process)
|
385 |
+
else:
|
386 |
+
logger.warning(f"Файл прокси-сервера не найден: {proxy_script}, будет использовано прямое подключение к API")
|
387 |
+
proxy_port = 8080 # Fallback на порт API сервера
|
388 |
|
389 |
# Запускаем Playground UI в режиме dev на порту 7860 (порт Hugging Face)
|
390 |
logger.info("Запуск Playground UI в режиме разработки на порту 7860...")
|
391 |
os.environ["PORT"] = "7860"
|
392 |
+
os.environ["AGENT_SERVER_URL"] = f"http://localhost:{proxy_port}" # Используем прокси вместо прямого подключения
|
393 |
os.environ["NEXT_PUBLIC_EDIT_GRAPH_MODE"] = "true" # Включаем расширенный режим редактирования
|
394 |
os.environ["NEXT_PUBLIC_DISABLE_CAMERA"] = "true" # Отключаем запрос на использование камеры
|
395 |
|
396 |
+
# Важные переменные для отключения запросов к дизайнеру
|
397 |
+
os.environ["NEXT_PUBLIC_DEV_MODE"] = "false"
|
398 |
os.environ["NEXT_PUBLIC_API_BASE_URL"] = "/api/agents"
|
399 |
+
os.environ["NEXT_PUBLIC_DESIGNER_API_URL"] = f"http://localhost:{proxy_port}" # Используем прокси для запросов дизайнера
|
400 |
|
401 |
# Запускаем Playground UI
|
402 |
playground_process = subprocess.Popen(
|
proxy_server.py
ADDED
@@ -0,0 +1,229 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
#!/usr/bin/env python3
|
2 |
+
import http.server
|
3 |
+
import socketserver
|
4 |
+
import urllib.request
|
5 |
+
import urllib.error
|
6 |
+
import json
|
7 |
+
import logging
|
8 |
+
import sys
|
9 |
+
import os
|
10 |
+
from pathlib import Path
|
11 |
+
|
12 |
+
# Настройка логирования
|
13 |
+
logging.basicConfig(level=logging.INFO,
|
14 |
+
format='%(asctime)s [%(levelname)s] %(message)s',
|
15 |
+
datefmt='%Y-%m-%d %H:%M:%S')
|
16 |
+
|
17 |
+
logger = logging.getLogger('ten-proxy')
|
18 |
+
|
19 |
+
# Настройки прокси
|
20 |
+
PROXY_PORT = 9090
|
21 |
+
API_SERVER = "http://localhost:8080"
|
22 |
+
|
23 |
+
# Мок-данные для /graphs
|
24 |
+
MOCK_GRAPHS_RESPONSE = [
|
25 |
+
{
|
26 |
+
"name": "Voice Agent",
|
27 |
+
"description": "Voice Agent with OpenAI",
|
28 |
+
"file": "voice_agent.json",
|
29 |
+
"id": "voice_agent",
|
30 |
+
"package": "default"
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"name": "Chat Agent",
|
34 |
+
"description": "Chat Agent",
|
35 |
+
"file": "chat_agent.json",
|
36 |
+
"id": "chat_agent",
|
37 |
+
"package": "default"
|
38 |
+
}
|
39 |
+
]
|
40 |
+
|
41 |
+
# Мок-данные для designer API
|
42 |
+
MOCK_DESIGNER_RESPONSE = {
|
43 |
+
"success": True,
|
44 |
+
"packages": [
|
45 |
+
{
|
46 |
+
"name": "default",
|
47 |
+
"description": "Default package",
|
48 |
+
"graphs": [
|
49 |
+
{
|
50 |
+
"name": "Voice Agent",
|
51 |
+
"description": "Voice Agent with OpenAI",
|
52 |
+
"file": "voice_agent.json",
|
53 |
+
"id": "voice_agent",
|
54 |
+
"package": "default"
|
55 |
+
},
|
56 |
+
{
|
57 |
+
"name": "Chat Agent",
|
58 |
+
"description": "Chat Agent",
|
59 |
+
"file": "chat_agent.json",
|
60 |
+
"id": "chat_agent",
|
61 |
+
"package": "default"
|
62 |
+
}
|
63 |
+
]
|
64 |
+
}
|
65 |
+
]
|
66 |
+
}
|
67 |
+
|
68 |
+
class ProxyHandler(http.server.BaseHTTPRequestHandler):
|
69 |
+
"""Обработчик запросов для прокси-сервера"""
|
70 |
+
|
71 |
+
def do_GET(self):
|
72 |
+
"""Обработка GET запросов"""
|
73 |
+
logger.info(f"PROXY: GET запрос: {self.path}")
|
74 |
+
|
75 |
+
# Перехватываем запросы к /graphs
|
76 |
+
if self.path == "/graphs":
|
77 |
+
self._handle_graphs_request()
|
78 |
+
# Перехватываем запросы к designer API
|
79 |
+
elif self.path.startswith("/api/designer/") or self.path.startswith("/api/dev/"):
|
80 |
+
self._handle_designer_request()
|
81 |
+
# Все остальные запросы перенаправляем на API сервер
|
82 |
+
else:
|
83 |
+
self._proxy_request("GET")
|
84 |
+
|
85 |
+
def do_POST(self):
|
86 |
+
"""Обработка POST запросов"""
|
87 |
+
logger.info(f"PROXY: POST запрос: {self.path}")
|
88 |
+
|
89 |
+
# Перехватываем запросы к designer API
|
90 |
+
if self.path.startswith("/api/designer/") or self.path.startswith("/api/dev/"):
|
91 |
+
self._handle_designer_request()
|
92 |
+
# Все остальные запросы перенаправляем на API сервер
|
93 |
+
else:
|
94 |
+
self._proxy_request("POST")
|
95 |
+
|
96 |
+
def do_OPTIONS(self):
|
97 |
+
"""Обработка OPTIONS запросов для CORS"""
|
98 |
+
self.send_response(200)
|
99 |
+
self.send_header('Access-Control-Allow-Origin', '*')
|
100 |
+
self.send_header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS')
|
101 |
+
self.send_header('Access-Control-Allow-Headers', 'Content-Type')
|
102 |
+
self.end_headers()
|
103 |
+
|
104 |
+
def _handle_graphs_request(self):
|
105 |
+
"""Обрабатывает запросы к /graphs с возможностью модификации ответа"""
|
106 |
+
try:
|
107 |
+
# Сначала пробуем получить данные от реального API
|
108 |
+
url = f"{API_SERVER}{self.path}"
|
109 |
+
try:
|
110 |
+
with urllib.request.urlopen(url) as response:
|
111 |
+
data = response.read().decode('utf-8')
|
112 |
+
|
113 |
+
# Проверяем, является ли ответ валидным JSON
|
114 |
+
try:
|
115 |
+
json_data = json.loads(data)
|
116 |
+
|
117 |
+
# Если API вернул пустой список или ошибку, заменяем на мок-данные
|
118 |
+
if (isinstance(json_data, list) and len(json_data) == 0) or (isinstance(json_data, dict) and "code" in json_data):
|
119 |
+
logger.warning(f"API вернул пустой список или ошибку, используем мок-данные")
|
120 |
+
self._send_success_response(MOCK_GRAPHS_RESPONSE)
|
121 |
+
return
|
122 |
+
|
123 |
+
# Если всё нормально, возвращаем оригинальный ответ
|
124 |
+
logger.info(f"API вернул непустой список графов, используем его")
|
125 |
+
self._send_response_with_data(200, data)
|
126 |
+
return
|
127 |
+
except json.JSONDecodeError:
|
128 |
+
logger.error(f"Ответ API не является валидным JSON: {data}")
|
129 |
+
self._send_success_response(MOCK_GRAPHS_RESPONSE)
|
130 |
+
return
|
131 |
+
except urllib.error.URLError as e:
|
132 |
+
logger.error(f"Не удалось подключиться к API: {e}")
|
133 |
+
# В случае ошибки используем мок-данные
|
134 |
+
self._send_success_response(MOCK_GRAPHS_RESPONSE)
|
135 |
+
return
|
136 |
+
except Exception as e:
|
137 |
+
logger.error(f"Ошибка при обработке запроса /graphs: {e}")
|
138 |
+
self._send_success_response(MOCK_GRAPHS_RESPONSE)
|
139 |
+
|
140 |
+
def _handle_designer_request(self):
|
141 |
+
"""Обрабатывает запросы к designer API, всегда возвращая мок-данные"""
|
142 |
+
logger.info(f"Перехват запроса к Designer API: {self.path}")
|
143 |
+
self._send_success_response(MOCK_DESIGNER_RESPONSE)
|
144 |
+
|
145 |
+
def _proxy_request(self, method):
|
146 |
+
"""Перенаправляет запрос на API сервер"""
|
147 |
+
try:
|
148 |
+
url = f"{API_SERVER}{self.path}"
|
149 |
+
|
150 |
+
# Получение данных запроса для POST
|
151 |
+
data = None
|
152 |
+
if method == "POST":
|
153 |
+
content_length = int(self.headers.get('Content-Length', 0))
|
154 |
+
data = self.rfile.read(content_length)
|
155 |
+
|
156 |
+
# Создание запроса
|
157 |
+
req = urllib.request.Request(url, data=data, method=method)
|
158 |
+
|
159 |
+
# Копирование заголовков
|
160 |
+
for header, value in self.headers.items():
|
161 |
+
if header.lower() not in ["host", "content-length"]:
|
162 |
+
req.add_header(header, value)
|
163 |
+
|
164 |
+
# Выполнение запроса
|
165 |
+
try:
|
166 |
+
with urllib.request.urlopen(req) as response:
|
167 |
+
# Чтение данных ответа
|
168 |
+
response_data = response.read()
|
169 |
+
|
170 |
+
# Отправка ответа клиенту
|
171 |
+
self.send_response(response.status)
|
172 |
+
|
173 |
+
# Копирование заголовков ответа
|
174 |
+
for header, value in response.getheaders():
|
175 |
+
if header.lower() != "transfer-encoding":
|
176 |
+
self.send_header(header, value)
|
177 |
+
|
178 |
+
# Добавление CORS заголовков
|
179 |
+
self.send_header('Access-Control-Allow-Origin', '*')
|
180 |
+
self.send_header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS')
|
181 |
+
self.send_header('Access-Control-Allow-Headers', 'Content-Type')
|
182 |
+
|
183 |
+
self.end_headers()
|
184 |
+
self.wfile.write(response_data)
|
185 |
+
except urllib.error.URLError as e:
|
186 |
+
logger.error(f"Ошибка при проксировании запроса: {e}")
|
187 |
+
self.send_error(502, f"Bad Gateway: {e}")
|
188 |
+
except Exception as e:
|
189 |
+
logger.error(f"Ошибка при обработке запроса: {e}")
|
190 |
+
self.send_error(500, f"Internal Server Error: {e}")
|
191 |
+
|
192 |
+
def _send_success_response(self, data):
|
193 |
+
"""Отправляет успешный ответ с данными"""
|
194 |
+
response_data = json.dumps(data)
|
195 |
+
self._send_response_with_data(200, response_data)
|
196 |
+
|
197 |
+
def _send_response_with_data(self, status_code, data):
|
198 |
+
"""Отправляет ответ с указанным статусом и данными"""
|
199 |
+
self.send_response(status_code)
|
200 |
+
self.send_header('Content-Type', 'application/json')
|
201 |
+
self.send_header('Access-Control-Allow-Origin', '*')
|
202 |
+
self.send_header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS')
|
203 |
+
self.send_header('Access-Control-Allow-Headers', 'Content-Type')
|
204 |
+
self.end_headers()
|
205 |
+
|
206 |
+
if isinstance(data, str):
|
207 |
+
self.wfile.write(data.encode('utf-8'))
|
208 |
+
else:
|
209 |
+
self.wfile.write(data)
|
210 |
+
|
211 |
+
def log_message(self, format, *args):
|
212 |
+
"""Настраиваем логирование для сервера"""
|
213 |
+
logger.debug(f"PROXY: {self.address_string()} - {format % args}")
|
214 |
+
|
215 |
+
def run_proxy_server(port=PROXY_PORT):
|
216 |
+
"""Запускает прокси-сервер"""
|
217 |
+
try:
|
218 |
+
with socketserver.TCPServer(("", port), ProxyHandler) as httpd:
|
219 |
+
logger.info(f"Запуск прокси-сервера на порту {port}")
|
220 |
+
httpd.serve_forever()
|
221 |
+
except KeyboardInterrupt:
|
222 |
+
logger.info("Прокси-сервер остановлен")
|
223 |
+
except Exception as e:
|
224 |
+
logger.error(f"Ошибка при запуске ��рокси-сервера: {e}")
|
225 |
+
|
226 |
+
if __name__ == "__main__":
|
227 |
+
# Запуск прокси-сервера
|
228 |
+
proxy_port = int(os.environ.get("PROXY_PORT", PROXY_PORT))
|
229 |
+
run_proxy_server(proxy_port)
|