1cd0aa2b8c
* feat(provider): add GitHub Copilot provider with device-flow auth
Adds GitHub Copilot as a model provider, so Copilot models (gpt-4o/4.1/5,
Claude, Gemini, …) work through the normal chat + agent loop, incl. native
tool calling and vision.
Auth is one-click via the GitHub OAuth device flow; the access token is stored
as the endpoint's (encrypted) api_key and sent directly as `Authorization:
Bearer` (no Copilot-token exchange, no refresh — matching how editors talk to
the Copilot API). Copilot is a normal ModelEndpoint detected by host; the only
provider-specific behaviour is a small set of required request headers,
injected centrally.
Sign-in is available from Settings → model endpoints ("Connect GitHub
Copilot") and from chat via `/setup copilot`.
- src/copilot.py (new), routes/copilot_routes.py (new): constants, header
builders, device-flow start/poll, model discovery, owner-scoped endpoint
provisioning.
- src/llm_core.py, src/endpoint_resolver.py: detect `copilot`, inject headers,
per-request x-initiator/vision.
- src/agent_loop.py: allowlist api.githubcopilot.com for native tool schemas.
- src/model_context.py: known context windows for Copilot (no unauthenticated
/models probe).
- static/, README, tests/test_copilot*.py.
* Tidy copilot_routes: clarify supports_tools, note _PENDING is per-process
81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
"""DB-backed tests for Copilot endpoint provisioning (routes/copilot_routes.py)."""
|
|
import json
|
|
import pytest
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
from core.database import Base, ModelEndpoint
|
|
import routes.copilot_routes as cr
|
|
|
|
|
|
def _mem_db(monkeypatch):
|
|
engine = create_engine("sqlite:///:memory:")
|
|
Base.metadata.create_all(bind=engine)
|
|
TestSessionLocal = sessionmaker(bind=engine)
|
|
monkeypatch.setattr(cr, "SessionLocal", TestSessionLocal)
|
|
return TestSessionLocal
|
|
|
|
|
|
def test_provision_creates_owner_scoped_endpoint(monkeypatch):
|
|
TestSessionLocal = _mem_db(monkeypatch)
|
|
monkeypatch.setattr(
|
|
cr.copilot, "fetch_models",
|
|
lambda base, token: [
|
|
{"id": "gpt-4o", "tool_calls": True, "vision": True},
|
|
{"id": "claude-3.5", "tool_calls": True, "vision": False},
|
|
],
|
|
)
|
|
|
|
res = cr._provision_endpoint("GHTOK", "https://api.githubcopilot.com", "alice")
|
|
|
|
assert res["base_url"] == "https://api.githubcopilot.com"
|
|
assert res["models"] == ["gpt-4o", "claude-3.5"]
|
|
|
|
db = TestSessionLocal()
|
|
try:
|
|
ep = db.query(ModelEndpoint).filter(ModelEndpoint.id == res["id"]).first()
|
|
assert ep is not None
|
|
assert ep.owner == "alice"
|
|
assert ep.is_enabled is True
|
|
assert ep.supports_tools is True
|
|
assert ep.api_key == "GHTOK" # round-trips through EncryptedText
|
|
assert json.loads(ep.cached_models) == ["gpt-4o", "claude-3.5"]
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
def test_provision_refreshes_existing_token(monkeypatch):
|
|
TestSessionLocal = _mem_db(monkeypatch)
|
|
monkeypatch.setattr(cr.copilot, "fetch_models", lambda base, token: [{"id": "gpt-4o", "tool_calls": True}])
|
|
|
|
first = cr._provision_endpoint("OLD", "https://api.githubcopilot.com", "bob")
|
|
second = cr._provision_endpoint("NEW", "https://api.githubcopilot.com", "bob")
|
|
|
|
# Same row reused (no duplicate), token refreshed.
|
|
assert first["id"] == second["id"]
|
|
db = TestSessionLocal()
|
|
try:
|
|
rows = db.query(ModelEndpoint).filter(ModelEndpoint.owner == "bob").all()
|
|
assert len(rows) == 1
|
|
assert rows[0].api_key == "NEW"
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
def test_provision_handles_model_fetch_failure(monkeypatch):
|
|
TestSessionLocal = _mem_db(monkeypatch)
|
|
|
|
def boom(base, token):
|
|
raise RuntimeError("network down")
|
|
|
|
monkeypatch.setattr(cr.copilot, "fetch_models", boom)
|
|
# Should still create the endpoint (login succeeded) with an empty model list.
|
|
res = cr._provision_endpoint("GHTOK", "https://api.githubcopilot.com", "carol")
|
|
assert res["models"] == []
|
|
db = TestSessionLocal()
|
|
try:
|
|
ep = db.query(ModelEndpoint).filter(ModelEndpoint.id == res["id"]).first()
|
|
assert ep is not None and ep.api_key == "GHTOK"
|
|
finally:
|
|
db.close()
|