Files
999/scripts/g4f_local_server.py
T

107 lines
4.1 KiB
Python

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Local G4F API Server for Oracle 999
Provides an OpenAI-compatible /v1/chat/completions endpoint on port 1337
"""
import json
from http.server import HTTPServer, BaseHTTPRequestHandler
import g4f
from g4f.client import Client
client = Client()
class G4FHandler(BaseHTTPRequestHandler):
def do_OPTIONS(self):
self.send_response(200)
self.send_header('Access-Control-Allow-Origin', '*')
self.send_header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS')
self.send_header('Access-Control-Allow-Headers', 'Content-Type, Authorization')
self.end_headers()
def do_GET(self):
self.send_response(200)
self.send_header('Content-Type', 'application/json')
self.send_header('Access-Control-Allow-Origin', '*')
self.end_headers()
self.wfile.write(json.dumps({'status': 'running', 'service': 'Oracle 999 G4F Server'}).encode('utf-8'))
def do_POST(self):
content_length = int(self.headers.get('Content-Length', 0))
post_data = self.rfile.read(content_length).decode('utf-8')
try:
req_json = json.loads(post_data)
messages = req_json.get('messages', [])
model = req_json.get('model', 'gpt-4o-mini')
print(f"[G4F Server] Request received for model: {model}, messages: {len(messages)}")
# 1. Try with requested model
response_text = None
try:
chat_res = client.chat.completions.create(
model=model,
messages=messages,
)
response_text = chat_res.choices[0].message.content
except Exception as e1:
print(f"[G4F Server] Primary model failed ({e1}), falling back to default...")
try:
chat_res = client.chat.completions.create(
model=g4f.models.default,
messages=messages,
)
response_text = chat_res.choices[0].message.content
except Exception as e2:
print(f"[G4F Server] Default model failed ({e2})")
if not response_text or len(response_text.strip()) == 0:
response_text = "Оракул открывает знаки: путь разворачивается перед вами. Сосредоточьтесь на ключевом символе и следуйте зову сердца."
resp_payload = {
"id": "chatcmpl-oracle999",
"object": "chat.completion",
"choices": [{
"index": 0,
"message": {
"role": "assistant",
"content": response_text
},
"finish_reason": "stop"
}]
}
self.send_response(200)
self.send_header('Content-Type', 'application/json; charset=utf-8')
self.send_header('Access-Control-Allow-Origin', '*')
self.end_headers()
self.wfile.write(json.dumps(resp_payload, ensure_ascii=False).encode('utf-8'))
print("[G4F Server] Successfully responded to client!")
except Exception as err:
print(f"[G4F Server] Handler error: {err}")
self.send_response(500)
self.send_header('Content-Type', 'application/json')
self.send_header('Access-Control-Allow-Origin', '*')
self.end_headers()
self.wfile.write(json.dumps({'error': str(err)}).encode('utf-8'))
def log_message(self, format, *args):
# Clean server logs
return
def run(port=1337):
server_address = ('127.0.0.1', port)
httpd = HTTPServer(server_address, G4FHandler)
print(f"🔮 [Oracle 999] Local G4F Server is LIVE on http://127.0.0.1:{port}/v1/chat/completions")
try:
httpd.serve_forever()
except KeyboardInterrupt:
pass
httpd.server_close()
if __name__ == '__main__':
run()