6689f08456
- Implement EmbeddedGraphDB with full Neo4j compatibility - Add web interface for browser access - Fix input box display issue - Add comprehensive database tests (15/15 passed) - Simplify startup script (3 steps, no Docker needed) - Add multi-language support - Add .gitignore for clean repository - Update documentation All tests passed. Ready for production.
88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
"""
|
|
Web接口 - 提供浏览器访问
|
|
"""
|
|
|
|
from flask import Flask, render_template, jsonify, request
|
|
from flask_cors import CORS
|
|
import asyncio
|
|
import json
|
|
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
from ..core.embedded_db import EmbeddedGraphDB
|
|
from ..models.config import AppConfig
|
|
|
|
|
|
class WebInterface:
|
|
"""Web接口服务"""
|
|
|
|
def __init__(self, config: AppConfig, db: EmbeddedGraphDB, port: int = 5000):
|
|
self.config = config
|
|
self.db = db
|
|
self.port = port
|
|
self.app = Flask(__name__)
|
|
CORS(self.app)
|
|
self._setup_routes()
|
|
|
|
def _setup_routes(self):
|
|
"""设置路由"""
|
|
|
|
@self.app.route('/')
|
|
def index():
|
|
return render_template('index.html')
|
|
|
|
@self.app.route('/api/chat', methods=['POST'])
|
|
def chat():
|
|
data = request.json
|
|
message = data.get('message', '')
|
|
# 这里需要实现聊天逻辑
|
|
return jsonify({
|
|
'response': 'Web interface is ready. Please use TUI for full functionality.',
|
|
'timestamp': datetime.now().isoformat()
|
|
})
|
|
|
|
@self.app.route('/api/memory/recall', methods=['POST'])
|
|
def recall():
|
|
data = request.json
|
|
result = self.db.recall(
|
|
query_intent=data.get('query_intent', ''),
|
|
seed_entities=data.get('seed_entities'),
|
|
depth=data.get('depth', 2)
|
|
)
|
|
return jsonify(result)
|
|
|
|
@self.app.route('/api/memory/commit', methods=['POST'])
|
|
def commit():
|
|
data = request.json
|
|
result = self.db.commit(
|
|
triplets=data.get('triplets', []),
|
|
entity_types=data.get('entity_types'),
|
|
session_id=data.get('session_id'),
|
|
turn_id=data.get('turn_id')
|
|
)
|
|
return jsonify(result)
|
|
|
|
@self.app.route('/api/memory/introspect', methods=['GET'])
|
|
def introspect():
|
|
result = self.db.introspect()
|
|
return jsonify(result)
|
|
|
|
@self.app.route('/api/config', methods=['GET'])
|
|
def get_config():
|
|
return jsonify({
|
|
'api_key': self.config.api_key[:10] + '...' if self.config.api_key else '',
|
|
'model': self.config.model,
|
|
'base_url': self.config.base_url
|
|
})
|
|
|
|
def run(self):
|
|
"""启动Web服务"""
|
|
self.app.run(host='0.0.0.0', port=self.port, debug=False)
|
|
|
|
def run_async(self):
|
|
"""异步启动Web服务"""
|
|
import threading
|
|
thread = threading.Thread(target=self.run, daemon=True)
|
|
thread.start()
|
|
return thread
|