Улучшение API wrapper для HuggingFace Space: добавлена надежная инициализация графов

api_wrapper.py

@@ -38,18 +38,33 @@ class TENAgentHandler(http.server.BaseHTTPRequestHandler):
                 try:
                     property_file = Path(AGENT_DIR) / "property.json"
                     if not property_file.exists():
-                        self.send_error(404, "Property file not found")
-                        return
+                        print("⚠️ Property file not found at", property_file)
+                        # Создаем базовый файл свойств на месте
+                        create_basic_property_file(property_file)
                     
                     with open(property_file, "r") as f:
                         property_data = json.load(f)
                     
                     graphs = property_data.get("graphs", [])
+                    print(f"✅ Returning {len(graphs)} graphs from property.json")
+                    print(f"Graphs: {json.dumps(graphs, indent=2)}")
+                    
                     self._set_headers()
                     self.wfile.write(json.dumps(graphs).encode())
                 except Exception as e:
                     print(f"Error reading property.json: {e}")
-                    self.send_error(500, f"Internal error: {e}")
+                    traceback.print_exc()
+                    
+                    # Возвращаем резервный список графов
+                    fallback_graphs = [
+                        {
+                            "name": "Voice Agent (Fallback)",
+                            "description": "Basic voice agent with OpenAI",
+                            "file": "voice_agent.json"
+                        }
+                    ]
+                    self._set_headers()
+                    self.wfile.write(json.dumps(fallback_graphs).encode())
             
             elif self.path in ["/health", "/"]:
                 # Просто возвращаем, что API сервер работает
@@ -77,8 +92,31 @@ class TENAgentHandler(http.server.BaseHTTPRequestHandler):
             
             # Обработка запросов к API TEN Graph Designer
             elif self.path.startswith("/api/designer/") or self.path.startswith("/api/dev/"):
-                self._set_headers()
-                self.wfile.write(json.dumps({"data": [], "status": 200, "message": "Success"}).encode())
+                if "/packages/reload" in self.path:
+                    # Возвращаем структурированный ответ с графами
+                    property_file = Path(AGENT_DIR) / "property.json"
+                    if property_file.exists():
+                        with open(property_file, "r") as f:
+                            property_data = json.load(f)
+                        
+                        graphs = property_data.get("graphs", [])
+                        response_data = {
+                            "data": graphs,
+                            "status": 200,
+                            "message": "Success"
+                        }
+                    else:
+                        response_data = {
+                            "data": [],
+                            "status": 200,
+                            "message": "Success"
+                        }
+                        
+                    self._set_headers()
+                    self.wfile.write(json.dumps(response_data).encode())
+                else:
+                    self._set_headers()
+                    self.wfile.write(json.dumps({"data": [], "status": 200, "message": "Success"}).encode())
                 
             else:
                 # Для всех остальных запросов возвращаем 404
@@ -153,6 +191,137 @@ class TENAgentHandler(http.server.BaseHTTPRequestHandler):
             traceback.print_exc()
             self.send_error(500, f"Internal server error: {e}")
 
+def create_basic_property_file(filepath):
+    """Создает базовый property.json файл"""
+    print(f"Creating basic property.json file at {filepath}")
+    
+    # Создаем базовый property.json
+    property_data = {
+        "name": "TEN Agent Example",
+        "version": "0.0.1",
+        "extensions": ["openai_chatgpt"],
+        "description": "A basic voice agent with OpenAI",
+        "graphs": [
+            {
+                "name": "Voice Agent",
+                "description": "Basic voice agent with OpenAI",
+                "file": "voice_agent.json"
+            },
+            {
+                "name": "Chat Agent",
+                "description": "Simple chat agent",
+                "file": "chat_agent.json"
+            }
+        ]
+    }
+    
+    try:
+        # Убедимся, что директория существует
+        filepath.parent.mkdir(exist_ok=True, parents=True)
+        
+        with open(filepath, "w") as f:
+            json.dump(property_data, f, indent=2)
+        print(f"✅ Created property.json at {filepath}")
+        
+        # Также создаем базовые файлы графов
+        create_basic_graph_files(filepath.parent)
+        
+        return True
+    except Exception as e:
+        print(f"❌ Error creating property.json: {e}")
+        traceback.print_exc()
+        return False
+
+def create_basic_graph_files(dir_path):
+    """Создает базовые файлы графов"""
+    print(f"Creating basic graph files in {dir_path}")
+    
+    # Создаем voice_agent.json
+    voice_agent = {
+        "_ten": {"version": "0.0.1"},
+        "nodes": [
+            {
+                "id": "start",
+                "type": "start",
+                "data": {"x": 100, "y": 100}
+            },
+            {
+                "id": "openai_chatgpt",
+                "type": "openai_chatgpt",
+                "data": {
+                    "x": 300,
+                    "y": 200,
+                    "properties": {
+                        "model": "gpt-3.5-turbo",
+                        "temperature": 0.7,
+                        "system_prompt": "You are a helpful assistant."
+                    }
+                }
+            },
+            {
+                "id": "end",
+                "type": "end",
+                "data": {"x": 500, "y": 100}
+            }
+        ],
+        "edges": [
+            {
+                "id": "start_to_chatgpt",
+                "source": "start",
+                "target": "openai_chatgpt"
+            },
+            {
+                "id": "chatgpt_to_end",
+                "source": "openai_chatgpt",
+                "target": "end"
+            }
+        ],
+        "groups": [],
+        "templates": [],
+        "root": "start"
+    }
+    
+    try:
+        with open(dir_path / "voice_agent.json", "w") as f:
+            json.dump(voice_agent, f, indent=2)
+        print(f"✅ Created voice_agent.json")
+        
+        # Создаем chat_agent.json (аналогичный voice_agent.json)
+        chat_agent = dict(voice_agent)
+        chat_agent["nodes"][1]["data"]["properties"]["system_prompt"] = "You are a helpful chat assistant."
+        
+        with open(dir_path / "chat_agent.json", "w") as f:
+            json.dump(chat_agent, f, indent=2)
+        print(f"✅ Created chat_agent.json")
+        
+        # Создаем manifest.json
+        manifest = {
+            "_ten": {"version": "0.0.1"},
+            "name": "default",
+            "agents": [
+                {
+                    "name": "voice_agent",
+                    "description": "A simple voice agent",
+                    "type": "voice"
+                },
+                {
+                    "name": "chat_agent",
+                    "description": "A text chat agent",
+                    "type": "chat"
+                }
+            ]
+        }
+        
+        with open(dir_path / "manifest.json", "w") as f:
+            json.dump(manifest, f, indent=2)
+        print(f"✅ Created manifest.json")
+        
+        return True
+    except Exception as e:
+        print(f"❌ Error creating graph files: {e}")
+        traceback.print_exc()
+        return False
+
 def run(server_class=http.server.HTTPServer, handler_class=TENAgentHandler, port=8080):
     server_address = ('', port)
     httpd = server_class(server_address, handler_class)
@@ -169,31 +338,19 @@ def run(server_class=http.server.HTTPServer, handler_class=TENAgentHandler, port
     property_file = agent_dir_path / "property.json"
     if not property_file.exists():
         print(f"WARNING: property.json not found in {AGENT_DIR}")
-        print("Creating minimal property.json...")
-        
-        # Создаем базовый property.json
-        property_data = {
-            "name": "TEN Agent Example",
-            "version": "0.0.1",
-            "extensions": ["openai_chatgpt"],
-            "description": "A basic voice agent with OpenAI",
-            "graphs": [
-                {
-                    "name": "Voice Agent",
-                    "description": "Basic voice agent with OpenAI",
-                    "file": "voice_agent.json"
-                }
-            ]
-        }
-        
+        create_basic_property_file(property_file)
+    else:
+        print(f"✅ Using existing property.json at {property_file}")
+        # Выводим содержимое
         try:
-            with open(property_file, "w") as f:
-                json.dump(property_data, f, indent=2)
-            print(f"Created {property_file}")
+            with open(property_file, "r") as f:
+                property_data = json.load(f)
+            print(f"Property.json content: {json.dumps(property_data, indent=2)}")
         except Exception as e:
-            print(f"ERROR: Could not create property.json: {e}")
+            print(f"Error reading property.json: {e}")
     
     try:
+        print(f"✅ API server is ready to receive requests!")
         httpd.serve_forever()
     except KeyboardInterrupt:
         print("Shutting down API server...")