feat(backend): 사내 LLM(FabriX) 게이트웨이 /api/ito + OpenCode FabriX 설정 렌더
- apps/gateway: OpenCode→FabriX 통과 중계. 헤더 3종 인증, x-llm-model-id 로 모델 선택, 401 시 Bearer/날것×클라이언트 헤더 재시도, 모델 허용 목록. ABAP_OPENCODE apps_ito 통째 복사 대신 200줄로 - opencode/opencode.fabrix.json.tmpl + render_opencode.py — .env AAF_* 로 렌더 - .env.example AAF 블록(기본 605 Gemma4, TOKEN_PREFIX=bearer). 파서가 줄 끝 # 주석을 값으로 읽던 것 수정 - tests/test_gateway.py 9개(MockTransport) Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com>
This commit is contained in:
lee-hyeon-cheol
co-authored by
Claude Fable 5.1
Claude Fable 5.1
parent
47d31dbe06
commit
435952f25c
@@ -0,0 +1,6 @@
|
||||
from django.apps import AppConfig
|
||||
|
||||
|
||||
class GatewayConfig(AppConfig):
|
||||
name = "apps.gateway"
|
||||
verbose_name = "사내 LLM(FabriX) 게이트웨이"
|
||||
@@ -0,0 +1,172 @@
|
||||
"""FabriX(사내 LLM, OpenAI 호환 Serving) 접속 규칙 — 순수 함수만. Django 안 봄.
|
||||
|
||||
ABAP_OPENCODE `apps_ito/aaf` 의 passthrough 중계에서 CodeAssist 에 필요한 규칙만 옮겼음:
|
||||
- 와이어는 OpenAI chat/completions 그대로. 인증은 Authorization 이 아니라 커스텀 헤더 3종.
|
||||
- 실제 모델은 body 의 model 이 아니라 `x-llm-model-id` 헤더가 고름. body model 은 고정 경로.
|
||||
- 고객사 게이트웨이가 토큰 형식(Bearer/날것)·클라이언트 헤더 이름을 서버마다 다르게 받아서,
|
||||
401 이면 다른 조합으로 한 번씩 더 찔러보고 통과한 조합을 기억함(2026-09-16 ABAP 쪽 실측).
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from dataclasses import dataclass, field, replace
|
||||
|
||||
DEFAULT_BODY_MODEL = "/mnt/models" # body.model — 공식 예제 고정값. 모델 선택은 헤더가 함
|
||||
TOKEN_HDR = "x-openapi-token"
|
||||
EMAIL_HDR = "x-generative-ai-user-email"
|
||||
MODEL_HDR = "x-llm-model-id"
|
||||
CLIENT_HDRS = ("x-generative-ai-client", "x-fabrix-client") # 고객사는 앞 것만 받음(뒤는 401)
|
||||
|
||||
ENV_KEYS = (
|
||||
"AAF_FABRIX_BASE_URL", # …/openapi/llm 까지. 뒤에 /chat/completions 붙임
|
||||
"AAF_FABRIX_MODEL_ID", # 기본 모델 id (x-llm-model-id)
|
||||
"AAF_FABRIX_MODELS", # "605:Gemma4,339:GaussO Flash,581:GaussO Think" — 화면에서 고를 수 있는 목록
|
||||
"AAF_FABRIX_CLIENT_KEY",
|
||||
"AAF_FABRIX_OPENAPI_TOKEN",
|
||||
"AAF_FABRIX_USER_EMAIL",
|
||||
"AAF_FABRIX_CLIENT_HEADER", # 기본 x-generative-ai-client
|
||||
"AAF_FABRIX_TOKEN_PREFIX", # "bearer" 면 토큰 앞에 "Bearer " 붙임. 기본은 날것
|
||||
"AAF_FABRIX_MODEL", # body.model. 기본 /mnt/models
|
||||
"AAF_FABRIX_MAX_TOKENS", # 있으면 max_completion_tokens 로 실음
|
||||
"AAF_RELAY_STREAM_USAGE", # "1" 이면 stream_options.include_usage — 제공자가 거부하면 끔
|
||||
"AAF_GATEWAY_KEY", # OpenCode 가 Authorization: Bearer 로 보내는 키. 비우면 검사 안 함
|
||||
)
|
||||
|
||||
|
||||
def parse_models(spec: str) -> dict[str, str]:
|
||||
""""339:GaussO Flash,581:GaussO Think" → {"339": "GaussO Flash", "581": "GaussO Think"}."""
|
||||
out: dict[str, str] = {}
|
||||
for item in (x.strip() for x in (spec or "").split(",")):
|
||||
if not item:
|
||||
continue
|
||||
mid, _, name = item.partition(":")
|
||||
if mid.strip():
|
||||
out[mid.strip()] = name.strip() or mid.strip()
|
||||
return out
|
||||
|
||||
|
||||
def token_value(token: str, prefix: str = "") -> str:
|
||||
"""x-openapi-token 값. 저장값에 Bearer 가 붙어 있어도 떼고, prefix=bearer 일 때만 다시 붙임."""
|
||||
t = (token or "").strip()
|
||||
if t.lower().startswith("bearer "):
|
||||
t = t[7:].strip()
|
||||
return f"Bearer {t}" if t and prefix.strip().lower() == "bearer" else t
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class FabrixConfig:
|
||||
base_url: str = ""
|
||||
model_id: str = ""
|
||||
client_key: str = ""
|
||||
token: str = ""
|
||||
email: str = ""
|
||||
client_header: str = CLIENT_HDRS[0]
|
||||
token_prefix: str = ""
|
||||
body_model: str = DEFAULT_BODY_MODEL
|
||||
models: dict[str, str] = field(default_factory=dict)
|
||||
max_tokens: int | None = None
|
||||
stream_usage: bool = False
|
||||
gateway_key: str = ""
|
||||
connect_timeout_s: float = 10.0
|
||||
read_timeout_s: float = 120.0 # 스트림 조각 사이 무수신 한계
|
||||
total_timeout_s: float = 600.0
|
||||
|
||||
@classmethod
|
||||
def from_env(cls, env: dict[str, str]) -> "FabrixConfig":
|
||||
g = lambda k: (env.get(k) or "").strip() # noqa: E731
|
||||
max_tokens = g("AAF_FABRIX_MAX_TOKENS")
|
||||
return cls(
|
||||
base_url=g("AAF_FABRIX_BASE_URL").rstrip("/"),
|
||||
model_id=g("AAF_FABRIX_MODEL_ID"),
|
||||
client_key=g("AAF_FABRIX_CLIENT_KEY"),
|
||||
token=g("AAF_FABRIX_OPENAPI_TOKEN"),
|
||||
email=g("AAF_FABRIX_USER_EMAIL"),
|
||||
client_header=g("AAF_FABRIX_CLIENT_HEADER") or CLIENT_HDRS[0],
|
||||
token_prefix=g("AAF_FABRIX_TOKEN_PREFIX"),
|
||||
body_model=g("AAF_FABRIX_MODEL") or DEFAULT_BODY_MODEL,
|
||||
models=parse_models(g("AAF_FABRIX_MODELS")),
|
||||
max_tokens=int(max_tokens) if max_tokens.isdigit() else None,
|
||||
stream_usage=g("AAF_RELAY_STREAM_USAGE") == "1",
|
||||
gateway_key=g("AAF_GATEWAY_KEY"),
|
||||
)
|
||||
|
||||
def missing(self) -> list[str]:
|
||||
"""비어 있으면 안 되는 설정 이름. 비어 있는 채로 부르면 정체불명 401/404 라 미리 막음."""
|
||||
need = {
|
||||
"AAF_FABRIX_BASE_URL": self.base_url,
|
||||
"AAF_FABRIX_MODEL_ID": self.model_id,
|
||||
"AAF_FABRIX_CLIENT_KEY": self.client_key,
|
||||
"AAF_FABRIX_OPENAPI_TOKEN": self.token,
|
||||
}
|
||||
return [k for k, v in need.items() if not v]
|
||||
|
||||
@property
|
||||
def url(self) -> str:
|
||||
return self.base_url.rstrip("/") + "/chat/completions"
|
||||
|
||||
def model_list(self) -> dict[str, str]:
|
||||
"""화면에 보여줄 모델 목록. AAF_FABRIX_MODELS 없으면 기본 모델 하나."""
|
||||
return dict(self.models) or {self.model_id: "FabriX"}
|
||||
|
||||
def pick_model_id(self, requested: str) -> str:
|
||||
"""OpenCode 가 보낸 model 이 허용 목록에 있으면 그것, 아니면 기본 모델."""
|
||||
rm = (requested or "").strip()
|
||||
return rm if rm and rm in self.models else self.model_id
|
||||
|
||||
def headers(self, model_id: str | None = None) -> dict[str, str]:
|
||||
return {
|
||||
"Content-Type": "application/json",
|
||||
MODEL_HDR: model_id or self.model_id,
|
||||
self.client_header: self.client_key,
|
||||
TOKEN_HDR: token_value(self.token, self.token_prefix),
|
||||
EMAIL_HDR: self.email, # 빈 값이라도 실음 — 공식 예제가 그럼
|
||||
}
|
||||
|
||||
def prepare(self, payload: dict) -> tuple[dict, dict[str, str]]:
|
||||
"""OpenCode 가 보낸 OpenAI 요청 → FabriX 로 보낼 (body, headers)."""
|
||||
body = dict(payload)
|
||||
headers = self.headers(self.pick_model_id(str(payload.get("model") or "")))
|
||||
body["model"] = self.body_model
|
||||
if body.get("stream") and self.stream_usage:
|
||||
so = dict(body.get("stream_options") or {})
|
||||
so["include_usage"] = True
|
||||
body["stream_options"] = so
|
||||
if self.max_tokens:
|
||||
body.setdefault("max_completion_tokens", self.max_tokens)
|
||||
return body, headers
|
||||
|
||||
def with_models(self, models: dict[str, str]) -> "FabrixConfig":
|
||||
return replace(self, models=models)
|
||||
|
||||
|
||||
Variant = tuple[str, str | None] # (토큰 접두 "bearer"|"raw", 클라이언트 헤더 이름)
|
||||
|
||||
|
||||
def auth_variants(headers: dict[str, str], last_ok: Variant | None = None) -> list[tuple[Variant, dict[str, str]]]:
|
||||
"""설정된 헤더에서 시작해 (Bearer/날것) × (클라이언트 헤더 이름) 조합을 만듦.
|
||||
순서: 마지막에 통과한 조합 → 설정된 조합 → 나머지. 토큰이 없으면 원본 하나만."""
|
||||
tok = headers.get(TOKEN_HDR)
|
||||
if not tok:
|
||||
return [(("raw", None), headers)]
|
||||
raw = tok[7:].strip() if tok.lower().startswith("bearer ") else tok.strip()
|
||||
client_val = next((headers[h] for h in CLIENT_HDRS if h in headers), None)
|
||||
client_names: tuple[str | None, ...] = CLIENT_HDRS if client_val is not None else (None,)
|
||||
|
||||
variants: dict[Variant, dict[str, str]] = {}
|
||||
for prefix in ("bearer", "raw"):
|
||||
for chdr in client_names:
|
||||
h = {k: v for k, v in headers.items() if k not in CLIENT_HDRS}
|
||||
h[TOKEN_HDR] = f"Bearer {raw}" if prefix == "bearer" else raw
|
||||
if chdr:
|
||||
h[chdr] = client_val or ""
|
||||
variants[(prefix, chdr)] = h
|
||||
|
||||
current: Variant = (
|
||||
"bearer" if tok.lower().startswith("bearer ") else "raw",
|
||||
next((h for h in CLIENT_HDRS if h in headers), None),
|
||||
)
|
||||
order: list[Variant] = [current]
|
||||
if last_ok and last_ok in variants and last_ok != current:
|
||||
order.insert(0, last_ok)
|
||||
order += [k for k in variants if k not in order]
|
||||
return [(k, variants[k]) for k in order]
|
||||
@@ -0,0 +1,13 @@
|
||||
from django.urls import path
|
||||
|
||||
from .views import chat_completions, healthcheck, models
|
||||
|
||||
# 슬래시 변형도 같이 — POST 는 리다이렉트에 기대면 본문이 유실됨.
|
||||
urlpatterns = [
|
||||
path("api/ito/healthcheck", healthcheck),
|
||||
path("api/ito/healthcheck/", healthcheck),
|
||||
path("api/ito/models", models),
|
||||
path("api/ito/models/", models),
|
||||
path("api/ito/chat/completions", chat_completions),
|
||||
path("api/ito/chat/completions/", chat_completions),
|
||||
]
|
||||
@@ -0,0 +1,128 @@
|
||||
"""`/api/ito/*` — OpenCode 가 붙는 OpenAI 호환 중계. 본문은 손 안 대고 FabriX 로 흘림.
|
||||
|
||||
GET /api/ito/healthcheck 기동 확인(무인증)
|
||||
GET /api/ito/models 설정된 모델 목록(OpenAI 형식)
|
||||
POST /api/ito/chat/completions 스트림/비스트림 통과 중계
|
||||
|
||||
설정이 비어 있어도 서버는 뜸 — 이 엔드포인트만 503 으로 이유를 말함(다른 기능까지 죽이지 않게).
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
from typing import AsyncIterator
|
||||
|
||||
import httpx
|
||||
from django.conf import settings
|
||||
from django.http import HttpRequest, HttpResponse, JsonResponse, StreamingHttpResponse
|
||||
from django.views.decorators.csrf import csrf_exempt
|
||||
|
||||
from .fabrix import FabrixConfig, Variant, auth_variants
|
||||
|
||||
# 테스트가 httpx.MockTransport 를 꽂는 자리. None 이면 진짜 네트워크.
|
||||
TRANSPORT: httpx.AsyncBaseTransport | None = None
|
||||
# 마지막에 통과한 인증 형식 — 프로세스당 하나. 다음 요청은 이것부터 시도.
|
||||
_last_ok: dict[str, Variant | None] = {"variant": None}
|
||||
|
||||
|
||||
def _cfg() -> FabrixConfig:
|
||||
return FabrixConfig.from_env(settings.FABRIX_ENV)
|
||||
|
||||
|
||||
def _err(status: int, message: str, typ: str) -> JsonResponse:
|
||||
"""OpenAI 형식 오류 — OpenCode 가 이 모양을 읽음(우리 envelope 아님)."""
|
||||
return JsonResponse({"error": {"message": message, "type": typ}}, status=status)
|
||||
|
||||
|
||||
def _key_rejected(request: HttpRequest, cfg: FabrixConfig) -> JsonResponse | None:
|
||||
"""AAF_GATEWAY_KEY 가 있으면 Authorization: Bearer <키> 대조. 비우면 검사 안 함(로컬 전용)."""
|
||||
if not cfg.gateway_key:
|
||||
return None
|
||||
auth = request.headers.get("Authorization", "")
|
||||
if auth.startswith("Bearer ") and auth[7:].strip() == cfg.gateway_key:
|
||||
return None
|
||||
return _err(401, "게이트웨이 키가 틀려 — opencode.json 의 apiKey 와 AAF_GATEWAY_KEY 확인", "unauthorized")
|
||||
|
||||
|
||||
def _client(cfg: FabrixConfig) -> httpx.AsyncClient:
|
||||
timeout = httpx.Timeout(cfg.total_timeout_s, connect=cfg.connect_timeout_s, read=cfg.read_timeout_s)
|
||||
return httpx.AsyncClient(timeout=timeout, transport=TRANSPORT)
|
||||
|
||||
|
||||
def _sse_error(status: int, detail: bytes) -> bytes:
|
||||
"""상류 오류를 SSE 오류 청크로. 연결은 정상 종료해 OpenCode 가 본문을 읽게."""
|
||||
payload = {"error": {"message": detail.decode("utf-8", "replace")[:2000], "type": "upstream_error", "status": status}}
|
||||
return b"data: " + json.dumps(payload, ensure_ascii=False).encode() + b"\n\ndata: [DONE]\n\n"
|
||||
|
||||
|
||||
async def healthcheck(_request: HttpRequest) -> JsonResponse:
|
||||
return JsonResponse({"success": True})
|
||||
|
||||
|
||||
async def models(request: HttpRequest) -> HttpResponse:
|
||||
cfg = _cfg()
|
||||
if rejected := _key_rejected(request, cfg):
|
||||
return rejected
|
||||
data = [{"id": mid, "object": "model", "owned_by": "fabrix", "name": name} for mid, name in cfg.model_list().items()]
|
||||
return JsonResponse({"object": "list", "data": data})
|
||||
|
||||
|
||||
@csrf_exempt
|
||||
async def chat_completions(request: HttpRequest) -> HttpResponse:
|
||||
if request.method != "POST":
|
||||
return _err(405, "POST 만 받아", "method_not_allowed")
|
||||
cfg = _cfg()
|
||||
if rejected := _key_rejected(request, cfg):
|
||||
return rejected
|
||||
if missing := cfg.missing():
|
||||
return _err(503, f"FabriX 설정 누락: {', '.join(missing)} (.env 확인)", "configuration_error")
|
||||
try:
|
||||
payload = json.loads(request.body.decode("utf-8"))
|
||||
assert isinstance(payload, dict)
|
||||
except Exception:
|
||||
return _err(400, "본문이 JSON 객체가 아니야", "invalid_request")
|
||||
|
||||
body, headers = cfg.prepare(payload)
|
||||
variants = auth_variants(headers, _last_ok["variant"])
|
||||
|
||||
if body.get("stream"):
|
||||
|
||||
async def gen() -> AsyncIterator[bytes]:
|
||||
async with _client(cfg) as client:
|
||||
resp: httpx.Response | None = None
|
||||
vkey: Variant | None = None
|
||||
for i, (vkey, vh) in enumerate(variants):
|
||||
req = client.build_request("POST", cfg.url, headers=vh, json=body)
|
||||
resp = await client.send(req, stream=True)
|
||||
if resp.status_code == 401 and i + 1 < len(variants):
|
||||
await resp.aclose() # 인증 형식이 안 맞은 것 — 다른 조합으로 한 번 더
|
||||
continue
|
||||
break
|
||||
assert resp is not None
|
||||
try:
|
||||
if resp.status_code >= 400:
|
||||
yield _sse_error(resp.status_code, await resp.aread())
|
||||
return
|
||||
_last_ok["variant"] = vkey
|
||||
async for chunk in resp.aiter_raw():
|
||||
yield chunk
|
||||
finally:
|
||||
await resp.aclose()
|
||||
|
||||
out = StreamingHttpResponse(gen(), content_type="text/event-stream")
|
||||
out["Cache-Control"] = "no-cache"
|
||||
out["X-Accel-Buffering"] = "no"
|
||||
return out
|
||||
|
||||
try:
|
||||
async with _client(cfg) as client:
|
||||
for i, (vkey, vh) in enumerate(variants):
|
||||
resp = await client.post(cfg.url, headers=vh, json=body)
|
||||
if resp.status_code == 401 and i + 1 < len(variants):
|
||||
continue
|
||||
if resp.status_code < 400:
|
||||
_last_ok["variant"] = vkey
|
||||
break
|
||||
except httpx.HTTPError as e:
|
||||
return _err(502, f"FabriX 호출 실패: {type(e).__name__}: {e}", "upstream_error")
|
||||
return HttpResponse(resp.content, status=resp.status_code, content_type="application/json")
|
||||
Reference in New Issue
Block a user