mirror of https://github.com/decolua/9router.git synced 2026-05-08 12:01:28 +00:00

Files

HXD.VN e1b836168a feat: add /v1/embeddings endpoint (OpenAI-compatible) (#146 )

* feat: implement /v1/embeddings endpoint (#117)

Add OpenAI-compatible POST /v1/embeddings endpoint that routes through
the existing provider credential + fallback infrastructure.

Changes:
- open-sse/handlers/embeddingsCore.js: core handler (handleEmbeddingsCore)
  * Validates input (string or array), encoding_format
  * Builds provider-specific URL and headers for openai, openrouter,
    and openai-compatible providers
  * Handles 401/403 token refresh via executor.refreshCredentials
  * Returns normalized OpenAI-format response { object: 'list', data, model, usage }
- cloud/src/handlers/embeddings.js: cloud Worker handler (handleEmbeddings)
  * Auth + machineId resolution identical to handleChat
  * Provider credential fallback loop with rate-limit tracking
- cloud/src/index.js: wire new routes
  * POST /v1/embeddings  (new format — machineId from API key)
  * POST /{machineId}/v1/embeddings  (old format — machineId from URL)

* test: add unit tests for /v1/embeddings endpoint

- Setup vitest as test framework (tests/ directory)
- embeddingsCore.test.js (36 tests):
  - buildEmbeddingsBody: single string, array, encoding_format, default float
  - buildEmbeddingsUrl: openai, openrouter, openai-compatible-*, unsupported
  - buildEmbeddingsHeaders: per-provider headers, accessToken fallback
  - handleEmbeddingsCore: input validation, success path, provider errors,
    network errors, invalid JSON, token refresh 401 handling
- embeddings.cloud.test.js (23 tests):
  - CORS OPTIONS preflight
  - Auth: missing/invalid/old-format/wrong key → 401/400
  - Body validation: bad JSON, missing model, missing input, bad model → 400
  - Happy path: single string, array, delegation, CORS header, machineId override
  - Rate limiting: all-rate-limited → 429 + Retry-After, no credentials → 400
  - Error propagation: non-fallback errors, 429 exhausts accounts

Total: 59/59 tests passing
Framework: vitest v4.0.18, Node v22.22.0

* feat: add Next.js API route for /v1/embeddings endpoint

Wire the embeddings handler into Next.js App Router.

- src/app/api/v1/embeddings/route.js: Next.js API route (POST + OPTIONS)
- src/sse/handlers/embeddings.js: SSE-layer handler mirroring chat.js pattern

Uses handleEmbeddingsCore from open-sse/handlers/embeddingsCore.js with
the same auth, credential fallback, and token refresh logic as the chat
handler. Supports REQUIRE_API_KEY env var, provider fallback loop, and
consistent logging.

2026-02-18 13:24:02 +07:00

2.0 KiB

Raw Blame History

9Router Embeddings Tests

Unit tests for the /v1/embeddings endpoint implementation.

Setup

Vitest must be installed globally or in /tmp/node_modules (due to npm workspace hoisting from the root Next.js project):

cd /tmp && npm install vitest

Running Tests

cd tests/
NODE_PATH=/tmp/node_modules /tmp/node_modules/.bin/vitest run --reporter=verbose --config ./vitest.config.js

Or using the package script (from the tests/ directory):

npm test

Test Files

File	What it tests
`unit/embeddingsCore.test.js`	`open-sse/handlers/embeddingsCore.js` — core logic: body builder, URL router, headers, handler flow
`unit/embeddings.cloud.test.js`	`cloud/src/handlers/embeddings.js` — cloud worker handler: auth, validation, rate limits, CORS

Coverage Summary (59 tests)

`embeddingsCore.test.js` (36 tests)

buildEmbeddingsBody: single string, array, encoding_format, default float
buildEmbeddingsUrl: openai, openrouter, openai-compatible-*, unsupported providers
buildEmbeddingsHeaders: per-provider header sets, fallback to accessToken
handleEmbeddingsCore input validation: missing, wrong type, null, empty
handleEmbeddingsCore success: response format, CORS, Content-Type, callbacks
handleEmbeddingsCore errors: 400/429/500, network error, invalid JSON
handleEmbeddingsCore token refresh: 401 retry, graceful fallback

`embeddings.cloud.test.js` (23 tests)

CORS OPTIONS: 200 response, empty body, correct headers
Authentication: missing key, bad format, old-format key, wrong key value, valid key
Body validation: invalid JSON, missing model, missing input, bad model
Happy path: single string, array, correct delegation, CORS header, machineId override
Rate limiting: all accounts rate-limited → 503 + Retry-After, no credentials → 400
Error propagation: non-fallback errors passed through, 429 exhausts accounts
machineId override: validates key, rejects wrong key

2.0 KiB Raw Blame History