@effect-uai/jina 0.12.1 → 0.14.0

package/dist/JinaReranker.d.mts

@@ -0,0 +1,53 @@
+import { JinaRerankerModel } from "./models.mjs";
+import { Context, Effect, Layer, Redacted } from "effect";
+import { HttpClient } from "effect/unstable/http";
+import * as AiError from "@effect-uai/core/AiError";
+import { CommonRerankRequest, RerankResponse, Reranker } from "@effect-uai/core/Reranker";
+import { ImageSource } from "@effect-uai/core/Image";
+declare namespace JinaReranker_d_exports {
+  export { Config, JinaRerankDocument, JinaRerankRequest, JinaReranker, JinaRerankerService, layer, make };
+}
+/**
+ * A candidate document. Text models take strings; `jina-reranker-m0` also
+ * takes `{ text }` / `{ image }`. One entry is one modality, and image
+ * *queries* are local-inference only, so `query` stays a string.
+ */
+type JinaRerankDocument = string | {
+  readonly text: string;
+} | {
+  readonly image: ImageSource;
+};
+/** Widens {@link CommonRerankRequest} with Jina's multimodal documents. */
+type JinaRerankRequest = Omit<CommonRerankRequest, "model" | "documents"> & {
+  readonly model: JinaRerankerModel;
+  readonly documents: ReadonlyArray<JinaRerankDocument>;
+};
+type JinaRerankerService = {
+  readonly rerank: (request: JinaRerankRequest) => Effect.Effect<RerankResponse, AiError.AiError>;
+};
+declare const JinaReranker_base: Context.ServiceClass<JinaReranker, "@betalyra/effect-uai/providers/jina/JinaReranker", JinaRerankerService>;
+/**
+ * Provider-typed service tag. Yield this for multimodal documents; yield the
+ * generic `Reranker` tag for provider-portable code. Both are registered by
+ * {@link layer}.
+ */
+declare class JinaReranker extends JinaReranker_base {}
+type Config = {
+  readonly apiKey: Redacted.Redacted;
+  readonly baseUrl?: string;
+};
+/**
+ * Build a `JinaRerankerService` value. For Layer-based setup, prefer
+ * {@link layer}.
+ */
+declare const make: (cfg: Config) => Effect.Effect<JinaRerankerService, never, HttpClient.HttpClient>;
+/**
+ * Layer registering both the provider-typed `JinaReranker` tag and the generic
+ * `Reranker` tag over one implementation. A `CommonRerankRequest` is
+ * structurally a `JinaRerankRequest` with string documents, so the generic
+ * registration forwards directly.
+ */
+declare const layer: (cfg: Config) => Layer.Layer<JinaReranker | Reranker, never, HttpClient.HttpClient>;
+//#endregion
+export { Config, JinaRerankDocument, JinaRerankRequest, JinaReranker, JinaRerankerService, layer, make, JinaReranker_d_exports as t };
+//# sourceMappingURL=JinaReranker.d.mts.map

package/dist/JinaReranker.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"JinaReranker.d.mts","names":[],"sources":["../src/JinaReranker.ts"],"mappings":";;;;;;;;;;;;;;KAsBY;WAEG;;WACA,OAAO;;;KAGV,oBAAoB,KAAK;WAC1B,OAAO;WACP,WAAW,cAAc;;KAGxB;WACD,SAAS,SAAS,sBAAsB,OAAO,OAAO,gBAAgB,QAAQ;;;;;;;;cAQ5E,qBAAqB;KAItB;WACD,QAAQ,SAAS;WACjB;;;;;;cAmIE,OAAI,KACV,WACJ,OAAO,OAAO,4BAA4B,WAAW;;;;;;;cAY3C,QAAK,KACX,WACJ,MAAM,MAAM,eAAe,iBAAiB,WAAW"}

package/dist/JinaReranker.mjs

@@ -0,0 +1,120 @@
+import { t as __exportAll } from "./rolldown-runtime-D7D4PA-g.mjs";
+import { Context, Effect, Encoding, Layer, Match, Schema } from "effect";
+import { HttpClient, HttpClientRequest } from "effect/unstable/http";
+import * as AiError from "@effect-uai/core/AiError";
+import { Reranker } from "@effect-uai/core/Reranker";
+//#region src/JinaReranker.ts
+var JinaReranker_exports = /* @__PURE__ */ __exportAll({
+	JinaReranker: () => JinaReranker,
+	layer: () => layer,
+	make: () => make
+});
+/**
+* Provider-typed service tag. Yield this for multimodal documents; yield the
+* generic `Reranker` tag for provider-portable code. Both are registered by
+* {@link layer}.
+*/
+var JinaReranker = class extends Context.Service()("@betalyra/effect-uai/providers/jina/JinaReranker") {};
+const imageSourceToWire = Match.type().pipe(Match.tag("url", (s) => ({ image: s.url })), Match.tag("base64", (s) => ({ image: s.base64 })), Match.tag("bytes", (s) => ({ image: Encoding.encodeBase64(s.bytes) })), Match.exhaustive);
+const documentToWire = Match.type().pipe(Match.when(Match.string, (s) => s), Match.when({ text: Match.string }, ({ text }) => ({ text })), Match.when({ image: Match.any }, ({ image }) => imageSourceToWire(image)), Match.exhaustive);
+const buildBody = (request) => ({
+	model: request.model,
+	query: request.query,
+	documents: request.documents.map(documentToWire),
+	...request.topN !== void 0 && { top_n: request.topN }
+});
+const WireResult = Schema.Struct({
+	index: Schema.Number,
+	relevance_score: Schema.Number
+});
+const WireUsage = Schema.Struct({ total_tokens: Schema.optional(Schema.Number) });
+const WireResponse = Schema.Struct({
+	model: Schema.optional(Schema.String),
+	usage: Schema.optional(WireUsage),
+	results: Schema.Array(WireResult)
+});
+const usageOf = (usage) => usage?.total_tokens !== void 0 ? { totalTokens: usage.total_tokens } : {};
+const toResponse = (wire) => ({
+	results: [...wire.results].sort((a, b) => b.relevance_score - a.relevance_score).map((r) => ({
+		index: r.index,
+		score: r.relevance_score
+	})),
+	usage: usageOf(wire.usage)
+});
+const transportFailure = (cause) => new AiError.Unavailable({
+	provider: "jina",
+	raw: cause
+});
+const httpStatusError = (status, body) => {
+	const provider = "jina";
+	const raw = body;
+	if (status === 429) return new AiError.RateLimited({
+		provider,
+		raw
+	});
+	if (status === 408 || status === 504) return new AiError.Timeout({
+		provider,
+		raw
+	});
+	if (status === 401) return new AiError.AuthFailed({
+		provider,
+		subtype: "auth",
+		raw
+	});
+	if (status === 403) return new AiError.AuthFailed({
+		provider,
+		subtype: "permission",
+		raw
+	});
+	if (status === 402) return new AiError.AuthFailed({
+		provider,
+		subtype: "billing",
+		raw
+	});
+	if (status === 413) return new AiError.ContextLengthExceeded({
+		provider,
+		raw
+	});
+	if (status >= 500) return new AiError.Unavailable({
+		provider,
+		status,
+		raw
+	});
+	return new AiError.InvalidRequest({
+		provider,
+		raw
+	});
+};
+const baseUrl = (cfg) => cfg.baseUrl ?? "https://api.jina.ai/v1";
+const postRerank = (cfg, body) => Effect.gen(function* () {
+	const client = yield* HttpClient.HttpClient;
+	const httpRequest = HttpClientRequest.post(`${baseUrl(cfg)}/rerank`).pipe(HttpClientRequest.bearerToken(cfg.apiKey), HttpClientRequest.bodyJsonUnsafe(body));
+	const response = yield* client.execute(httpRequest).pipe(Effect.mapError(transportFailure));
+	if (response.status >= 400) {
+		const text = yield* response.text.pipe(Effect.orElseSucceed(() => ""));
+		return yield* httpStatusError(response.status, text);
+	}
+	const json = yield* response.json.pipe(Effect.mapError(transportFailure));
+	return yield* Schema.decodeUnknownEffect(WireResponse)(json).pipe(Effect.mapError(transportFailure));
+});
+const rerankImpl = (cfg) => (request) => Effect.map(postRerank(cfg, buildBody(request)), toResponse);
+/**
+* Build a `JinaRerankerService` value. For Layer-based setup, prefer
+* {@link layer}.
+*/
+const make = (cfg) => Effect.map(HttpClient.HttpClient, (client) => ({ rerank: (request) => rerankImpl(cfg)(request).pipe(Effect.provideService(HttpClient.HttpClient, client)) }));
+/**
+* Layer registering both the provider-typed `JinaReranker` tag and the generic
+* `Reranker` tag over one implementation. A `CommonRerankRequest` is
+* structurally a `JinaRerankRequest` with string documents, so the generic
+* registration forwards directly.
+*/
+const layer = (cfg) => {
+	const typed = Layer.effect(JinaReranker, make(cfg));
+	const generic = Layer.effect(Reranker, Effect.map(make(cfg), (s) => ({ rerank: (request) => s.rerank(request) })));
+	return Layer.merge(typed, generic);
+};
+//#endregion
+export { JinaReranker, layer, make, JinaReranker_exports as t };
+
+//# sourceMappingURL=JinaReranker.mjs.map

package/dist/JinaReranker.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"JinaReranker.mjs","names":[],"sources":["../src/JinaReranker.ts"],"sourcesContent":["import { Context, Effect, Encoding, Layer, Match, type Redacted, Schema } from \"effect\"\nimport { HttpClient, HttpClientRequest } from \"effect/unstable/http\"\nimport * as AiError from \"@effect-uai/core/AiError\"\nimport type { ImageSource } from \"@effect-uai/core/Image\"\nimport {\n  type CommonRerankRequest,\n  Reranker,\n  type RerankerService,\n  type RerankResponse,\n  type RerankUsage,\n} from \"@effect-uai/core/Reranker\"\nimport type { JinaRerankerModel } from \"./models.js\"\n\n// ---------------------------------------------------------------------------\n// Public types\n// ---------------------------------------------------------------------------\n\n/**\n * A candidate document. Text models take strings; `jina-reranker-m0` also\n * takes `{ text }` / `{ image }`. One entry is one modality, and image\n * *queries* are local-inference only, so `query` stays a string.\n */\nexport type JinaRerankDocument =\n  | string\n  | { readonly text: string }\n  | { readonly image: ImageSource }\n\n/** Widens {@link CommonRerankRequest} with Jina's multimodal documents. */\nexport type JinaRerankRequest = Omit<CommonRerankRequest, \"model\" | \"documents\"> & {\n  readonly model: JinaRerankerModel\n  readonly documents: ReadonlyArray<JinaRerankDocument>\n}\n\nexport type JinaRerankerService = {\n  readonly rerank: (request: JinaRerankRequest) => Effect.Effect<RerankResponse, AiError.AiError>\n}\n\n/**\n * Provider-typed service tag. Yield this for multimodal documents; yield the\n * generic `Reranker` tag for provider-portable code. Both are registered by\n * {@link layer}.\n */\nexport class JinaReranker extends Context.Service<JinaReranker, JinaRerankerService>()(\n  \"@betalyra/effect-uai/providers/jina/JinaReranker\",\n) {}\n\nexport type Config = {\n  readonly apiKey: Redacted.Redacted\n  readonly baseUrl?: string\n}\n\n// ---------------------------------------------------------------------------\n// Codec - request body\n// ---------------------------------------------------------------------------\n\ntype WireDocument = string | { readonly text: string } | { readonly image: string }\n\n// Jina detects URL vs base64 from the value itself, so both go in `image`.\nconst imageSourceToWire: (s: ImageSource) => WireDocument = Match.type<ImageSource>().pipe(\n  Match.tag(\"url\", (s): WireDocument => ({ image: s.url })),\n  Match.tag(\"base64\", (s): WireDocument => ({ image: s.base64 })),\n  Match.tag(\"bytes\", (s): WireDocument => ({ image: Encoding.encodeBase64(s.bytes) })),\n  Match.exhaustive,\n)\n\nconst documentToWire: (doc: JinaRerankDocument) => WireDocument =\n  Match.type<JinaRerankDocument>().pipe(\n    Match.when(Match.string, (s): WireDocument => s),\n    Match.when({ text: Match.string }, ({ text }): WireDocument => ({ text })),\n    Match.when({ image: Match.any }, ({ image }) => imageSourceToWire(image)),\n    Match.exhaustive,\n  )\n\ntype WireBody = {\n  readonly model: string\n  readonly query: string\n  readonly documents: ReadonlyArray<WireDocument>\n  readonly top_n?: number\n}\n\n// No `return_documents`: the caller holds `documents` and `results[].index`\n// points back into it.\nconst buildBody = (request: JinaRerankRequest): WireBody => ({\n  model: request.model,\n  query: request.query,\n  documents: request.documents.map(documentToWire),\n  ...(request.topN !== undefined && { top_n: request.topN }),\n})\n\n// ---------------------------------------------------------------------------\n// Codec - response\n// ---------------------------------------------------------------------------\n\nconst WireResult = Schema.Struct({\n  index: Schema.Number,\n  relevance_score: Schema.Number,\n})\n\nconst WireUsage = Schema.Struct({\n  total_tokens: Schema.optional(Schema.Number),\n})\ntype WireUsage = typeof WireUsage.Type\n\nconst WireResponse = Schema.Struct({\n  model: Schema.optional(Schema.String),\n  usage: Schema.optional(WireUsage),\n  results: Schema.Array(WireResult),\n})\ntype WireResponse = typeof WireResponse.Type\n\nconst usageOf = (usage: WireUsage | undefined): RerankUsage =>\n  usage?.total_tokens !== undefined ? { totalTokens: usage.total_tokens } : {}\n\n// Sort rather than trust wire order: the score contract promises descending.\nconst toResponse = (wire: WireResponse): RerankResponse => ({\n  results: [...wire.results]\n    .sort((a, b) => b.relevance_score - a.relevance_score)\n    .map((r) => ({ index: r.index, score: r.relevance_score })),\n  usage: usageOf(wire.usage),\n})\n\n// ---------------------------------------------------------------------------\n// HTTP\n// ---------------------------------------------------------------------------\n\nconst transportFailure = (cause: unknown): AiError.AiError =>\n  new AiError.Unavailable({ provider: \"jina\", raw: cause })\n\nconst httpStatusError = (status: number, body: string): AiError.AiError => {\n  const provider = \"jina\"\n  const raw = body\n  if (status === 429) return new AiError.RateLimited({ provider, raw })\n  if (status === 408 || status === 504) return new AiError.Timeout({ provider, raw })\n  if (status === 401) return new AiError.AuthFailed({ provider, subtype: \"auth\", raw })\n  if (status === 403) return new AiError.AuthFailed({ provider, subtype: \"permission\", raw })\n  if (status === 402) return new AiError.AuthFailed({ provider, subtype: \"billing\", raw })\n  if (status === 413) return new AiError.ContextLengthExceeded({ provider, raw })\n  if (status >= 500) return new AiError.Unavailable({ provider, status, raw })\n  return new AiError.InvalidRequest({ provider, raw })\n}\n\nconst baseUrl = (cfg: Config): string => cfg.baseUrl ?? \"https://api.jina.ai/v1\"\n\nconst postRerank = (\n  cfg: Config,\n  body: WireBody,\n): Effect.Effect<WireResponse, AiError.AiError, HttpClient.HttpClient> =>\n  Effect.gen(function* () {\n    const client = yield* HttpClient.HttpClient\n    const httpRequest = HttpClientRequest.post(`${baseUrl(cfg)}/rerank`).pipe(\n      HttpClientRequest.bearerToken(cfg.apiKey),\n      HttpClientRequest.bodyJsonUnsafe(body),\n    )\n    const response = yield* client.execute(httpRequest).pipe(Effect.mapError(transportFailure))\n    if (response.status >= 400) {\n      const text = yield* response.text.pipe(Effect.orElseSucceed(() => \"\"))\n      return yield* httpStatusError(response.status, text)\n    }\n    const json = yield* response.json.pipe(Effect.mapError(transportFailure))\n    return yield* Schema.decodeUnknownEffect(WireResponse)(json).pipe(\n      Effect.mapError(transportFailure),\n    )\n  })\n\nconst rerankImpl =\n  (cfg: Config) =>\n  (\n    request: JinaRerankRequest,\n  ): Effect.Effect<RerankResponse, AiError.AiError, HttpClient.HttpClient> =>\n    Effect.map(postRerank(cfg, buildBody(request)), toResponse)\n\n// ---------------------------------------------------------------------------\n// Constructors\n// ---------------------------------------------------------------------------\n\n/**\n * Build a `JinaRerankerService` value. For Layer-based setup, prefer\n * {@link layer}.\n */\nexport const make = (\n  cfg: Config,\n): Effect.Effect<JinaRerankerService, never, HttpClient.HttpClient> =>\n  Effect.map(HttpClient.HttpClient, (client) => ({\n    rerank: (request) =>\n      rerankImpl(cfg)(request).pipe(Effect.provideService(HttpClient.HttpClient, client)),\n  }))\n\n/**\n * Layer registering both the provider-typed `JinaReranker` tag and the generic\n * `Reranker` tag over one implementation. A `CommonRerankRequest` is\n * structurally a `JinaRerankRequest` with string documents, so the generic\n * registration forwards directly.\n */\nexport const layer = (\n  cfg: Config,\n): Layer.Layer<JinaReranker | Reranker, never, HttpClient.HttpClient> => {\n  const typed = Layer.effect(JinaReranker, make(cfg))\n  const generic = Layer.effect(\n    Reranker,\n    Effect.map(make(cfg), (s): RerankerService => ({ rerank: (request) => s.rerank(request) })),\n  )\n  return Layer.merge(typed, generic)\n}\n"],"mappings":";;;;;;;;;;;;;;;;AA0CA,IAAa,eAAb,cAAkC,QAAQ,QAA2C,CAAC,CACpF,kDACF,CAAC,CAAC,CAAC;AAcH,MAAM,oBAAsD,MAAM,KAAkB,CAAC,CAAC,KACpF,MAAM,IAAI,QAAQ,OAAqB,EAAE,OAAO,EAAE,IAAI,EAAE,GACxD,MAAM,IAAI,WAAW,OAAqB,EAAE,OAAO,EAAE,OAAO,EAAE,GAC9D,MAAM,IAAI,UAAU,OAAqB,EAAE,OAAO,SAAS,aAAa,EAAE,KAAK,EAAE,EAAE,GACnF,MAAM,UACR;AAEA,MAAM,iBACJ,MAAM,KAAyB,CAAC,CAAC,KAC/B,MAAM,KAAK,MAAM,SAAS,MAAoB,CAAC,GAC/C,MAAM,KAAK,EAAE,MAAM,MAAM,OAAO,IAAI,EAAE,YAA0B,EAAE,KAAK,EAAE,GACzE,MAAM,KAAK,EAAE,OAAO,MAAM,IAAI,IAAI,EAAE,YAAY,kBAAkB,KAAK,CAAC,GACxE,MAAM,UACR;AAWF,MAAM,aAAa,aAA0C;CAC3D,OAAO,QAAQ;CACf,OAAO,QAAQ;CACf,WAAW,QAAQ,UAAU,IAAI,cAAc;CAC/C,GAAI,QAAQ,SAAS,KAAA,KAAa,EAAE,OAAO,QAAQ,KAAK;AAC1D;AAMA,MAAM,aAAa,OAAO,OAAO;CAC/B,OAAO,OAAO;CACd,iBAAiB,OAAO;AAC1B,CAAC;AAED,MAAM,YAAY,OAAO,OAAO,EAC9B,cAAc,OAAO,SAAS,OAAO,MAAM,EAC7C,CAAC;AAGD,MAAM,eAAe,OAAO,OAAO;CACjC,OAAO,OAAO,SAAS,OAAO,MAAM;CACpC,OAAO,OAAO,SAAS,SAAS;CAChC,SAAS,OAAO,MAAM,UAAU;AAClC,CAAC;AAGD,MAAM,WAAW,UACf,OAAO,iBAAiB,KAAA,IAAY,EAAE,aAAa,MAAM,aAAa,IAAI,CAAC;AAG7E,MAAM,cAAc,UAAwC;CAC1D,SAAS,CAAC,GAAG,KAAK,OAAO,CAAC,CACvB,MAAM,GAAG,MAAM,EAAE,kBAAkB,EAAE,eAAe,CAAC,CACrD,KAAK,OAAO;EAAE,OAAO,EAAE;EAAO,OAAO,EAAE;CAAgB,EAAE;CAC5D,OAAO,QAAQ,KAAK,KAAK;AAC3B;AAMA,MAAM,oBAAoB,UACxB,IAAI,QAAQ,YAAY;CAAE,UAAU;CAAQ,KAAK;AAAM,CAAC;AAE1D,MAAM,mBAAmB,QAAgB,SAAkC;CACzE,MAAM,WAAW;CACjB,MAAM,MAAM;CACZ,IAAI,WAAW,KAAK,OAAO,IAAI,QAAQ,YAAY;EAAE;EAAU;CAAI,CAAC;CACpE,IAAI,WAAW,OAAO,WAAW,KAAK,OAAO,IAAI,QAAQ,QAAQ;EAAE;EAAU;CAAI,CAAC;CAClF,IAAI,WAAW,KAAK,OAAO,IAAI,QAAQ,WAAW;EAAE;EAAU,SAAS;EAAQ;CAAI,CAAC;CACpF,IAAI,WAAW,KAAK,OAAO,IAAI,QAAQ,WAAW;EAAE;EAAU,SAAS;EAAc;CAAI,CAAC;CAC1F,IAAI,WAAW,KAAK,OAAO,IAAI,QAAQ,WAAW;EAAE;EAAU,SAAS;EAAW;CAAI,CAAC;CACvF,IAAI,WAAW,KAAK,OAAO,IAAI,QAAQ,sBAAsB;EAAE;EAAU;CAAI,CAAC;CAC9E,IAAI,UAAU,KAAK,OAAO,IAAI,QAAQ,YAAY;EAAE;EAAU;EAAQ;CAAI,CAAC;CAC3E,OAAO,IAAI,QAAQ,eAAe;EAAE;EAAU;CAAI,CAAC;AACrD;AAEA,MAAM,WAAW,QAAwB,IAAI,WAAW;AAExD,MAAM,cACJ,KACA,SAEA,OAAO,IAAI,aAAa;CACtB,MAAM,SAAS,OAAO,WAAW;CACjC,MAAM,cAAc,kBAAkB,KAAK,GAAG,QAAQ,GAAG,EAAE,QAAQ,CAAC,CAAC,KACnE,kBAAkB,YAAY,IAAI,MAAM,GACxC,kBAAkB,eAAe,IAAI,CACvC;CACA,MAAM,WAAW,OAAO,OAAO,QAAQ,WAAW,CAAC,CAAC,KAAK,OAAO,SAAS,gBAAgB,CAAC;CAC1F,IAAI,SAAS,UAAU,KAAK;EAC1B,MAAM,OAAO,OAAO,SAAS,KAAK,KAAK,OAAO,oBAAoB,EAAE,CAAC;EACrE,OAAO,OAAO,gBAAgB,SAAS,QAAQ,IAAI;CACrD;CACA,MAAM,OAAO,OAAO,SAAS,KAAK,KAAK,OAAO,SAAS,gBAAgB,CAAC;CACxE,OAAO,OAAO,OAAO,oBAAoB,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,KAC3D,OAAO,SAAS,gBAAgB,CAClC;AACF,CAAC;AAEH,MAAM,cACH,SAEC,YAEA,OAAO,IAAI,WAAW,KAAK,UAAU,OAAO,CAAC,GAAG,UAAU;;;;;AAU9D,MAAa,QACX,QAEA,OAAO,IAAI,WAAW,aAAa,YAAY,EAC7C,SAAS,YACP,WAAW,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,OAAO,eAAe,WAAW,YAAY,MAAM,CAAC,EACtF,EAAE;;;;;;;AAQJ,MAAa,SACX,QACuE;CACvE,MAAM,QAAQ,MAAM,OAAO,cAAc,KAAK,GAAG,CAAC;CAClD,MAAM,UAAU,MAAM,OACpB,UACA,OAAO,IAAI,KAAK,GAAG,IAAI,OAAwB,EAAE,SAAS,YAAY,EAAE,OAAO,OAAO,EAAE,EAAE,CAC5F;CACA,OAAO,MAAM,MAAM,OAAO,OAAO;AACnC"}

package/dist/index.d.mts

@@ -1,4 +1,5 @@
-import { JinaEmbeddingModel, JinaEngine } from "./models.mjs";
+import { JinaEmbeddingModel, JinaEngine, JinaRerankerModel } from "./models.mjs";
 import { t as JinaEmbedding_d_exports } from "./JinaEmbedding.mjs";
 import { t as JinaReader_d_exports } from "./JinaReader.mjs";
-export { JinaEmbedding_d_exports as JinaEmbedding, JinaEmbeddingModel, JinaEngine, JinaReader_d_exports as JinaReader };
+import { t as JinaReranker_d_exports } from "./JinaReranker.mjs";
+export { JinaEmbedding_d_exports as JinaEmbedding, JinaEmbeddingModel, JinaEngine, JinaReader_d_exports as JinaReader, JinaReranker_d_exports as JinaReranker, JinaRerankerModel };

package/dist/index.mjs

@@ -1,4 +1,5 @@
 import { t as JinaEmbedding_exports } from "./JinaEmbedding.mjs";
 import { t as JinaReader_exports } from "./JinaReader.mjs";
+import { t as JinaReranker_exports } from "./JinaReranker.mjs";
 import "./models.mjs";
-export { JinaEmbedding_exports as JinaEmbedding, JinaReader_exports as JinaReader };
+export { JinaEmbedding_exports as JinaEmbedding, JinaReader_exports as JinaReader, JinaReranker_exports as JinaReranker };

package/dist/models.d.mts

@@ -19,6 +19,17 @@
  * Reference: https://jina.ai/embeddings/
  */
 type JinaEmbeddingModel = "jina-embeddings-v4" | "jina-embeddings-v5-text-small" | "jina-embeddings-v5-text-nano" | "jina-embeddings-v3" | "jina-clip-v2" | (string & {});
+/**
+ * Known Jina reranker model identifiers.
+ *
+ * - `jina-reranker-v3.5`: flagship listwise reranker, 131k context.
+ * - `jina-reranker-v3`: previous generation, schema-identical.
+ * - `jina-reranker-m0`: multimodal, takes `{ text }` / `{ image }`
+ *   documents (see `JinaRerankRequest`).
+ *
+ * Reference: https://jina.ai/reranker/
+ */
+type JinaRerankerModel = "jina-reranker-v3.5" | "jina-reranker-v3" | "jina-reranker-m0" | (string & {});
 /**
  * Reader fetching engine (`X-Engine`). `browser` renders JavaScript,
  * `curl` is a fast static fetch, `auto` picks per page. Provider-specific,
@@ -26,5 +37,5 @@ type JinaEmbeddingModel = "jina-embeddings-v4" | "jina-embeddings-v5-text-small"
  */
 type JinaEngine = "browser" | "curl" | "auto" | (string & {});
 //#endregion
-export { JinaEmbeddingModel, JinaEngine };
+export { JinaEmbeddingModel, JinaEngine, JinaRerankerModel };
 //# sourceMappingURL=models.d.mts.map

package/dist/models.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"models.d.mts","names":[],"sources":["../src/models.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;KAmBY;;;;;;KAcA"}
+{"version":3,"file":"models.d.mts","names":[],"sources":["../src/models.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;KAmBY;;;;;;;;;;;KAmBA;;;;;;KAYA"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@effect-uai/jina",
-  "version": "0.12.1",
+  "version": "0.14.0",
   "description": "Jina AI provider for @effect-uai/core (embeddings, rerank, web read).",
   "keywords": [
     "ai",
@@ -42,6 +42,10 @@
     "./JinaReader": {
       "types": "./dist/JinaReader.d.mts",
       "import": "./dist/JinaReader.mjs"
+    },
+    "./JinaReranker": {
+      "types": "./dist/JinaReranker.d.mts",
+      "import": "./dist/JinaReranker.mjs"
     }
   },
   "publishConfig": {
@@ -50,7 +54,7 @@
   "devDependencies": {
     "effect": "4.0.0-rc.111",
     "typescript": "^7.0.2",
-    "@effect-uai/core": "0.12.1"
+    "@effect-uai/core": "0.14.0"
   },
   "peerDependencies": {
     "@effect-uai/core": ">=0.2.0 <1",

package/src/JinaReranker.ts

@@ -0,0 +1,203 @@
+import { Context, Effect, Encoding, Layer, Match, type Redacted, Schema } from "effect"
+import { HttpClient, HttpClientRequest } from "effect/unstable/http"
+import * as AiError from "@effect-uai/core/AiError"
+import type { ImageSource } from "@effect-uai/core/Image"
+import {
+  type CommonRerankRequest,
+  Reranker,
+  type RerankerService,
+  type RerankResponse,
+  type RerankUsage,
+} from "@effect-uai/core/Reranker"
+import type { JinaRerankerModel } from "./models.js"
+
+// ---------------------------------------------------------------------------
+// Public types
+// ---------------------------------------------------------------------------
+
+/**
+ * A candidate document. Text models take strings; `jina-reranker-m0` also
+ * takes `{ text }` / `{ image }`. One entry is one modality, and image
+ * *queries* are local-inference only, so `query` stays a string.
+ */
+export type JinaRerankDocument =
+  | string
+  | { readonly text: string }
+  | { readonly image: ImageSource }
+
+/** Widens {@link CommonRerankRequest} with Jina's multimodal documents. */
+export type JinaRerankRequest = Omit<CommonRerankRequest, "model" | "documents"> & {
+  readonly model: JinaRerankerModel
+  readonly documents: ReadonlyArray<JinaRerankDocument>
+}
+
+export type JinaRerankerService = {
+  readonly rerank: (request: JinaRerankRequest) => Effect.Effect<RerankResponse, AiError.AiError>
+}
+
+/**
+ * Provider-typed service tag. Yield this for multimodal documents; yield the
+ * generic `Reranker` tag for provider-portable code. Both are registered by
+ * {@link layer}.
+ */
+export class JinaReranker extends Context.Service<JinaReranker, JinaRerankerService>()(
+  "@betalyra/effect-uai/providers/jina/JinaReranker",
+) {}
+
+export type Config = {
+  readonly apiKey: Redacted.Redacted
+  readonly baseUrl?: string
+}
+
+// ---------------------------------------------------------------------------
+// Codec - request body
+// ---------------------------------------------------------------------------
+
+type WireDocument = string | { readonly text: string } | { readonly image: string }
+
+// Jina detects URL vs base64 from the value itself, so both go in `image`.
+const imageSourceToWire: (s: ImageSource) => WireDocument = Match.type<ImageSource>().pipe(
+  Match.tag("url", (s): WireDocument => ({ image: s.url })),
+  Match.tag("base64", (s): WireDocument => ({ image: s.base64 })),
+  Match.tag("bytes", (s): WireDocument => ({ image: Encoding.encodeBase64(s.bytes) })),
+  Match.exhaustive,
+)
+
+const documentToWire: (doc: JinaRerankDocument) => WireDocument =
+  Match.type<JinaRerankDocument>().pipe(
+    Match.when(Match.string, (s): WireDocument => s),
+    Match.when({ text: Match.string }, ({ text }): WireDocument => ({ text })),
+    Match.when({ image: Match.any }, ({ image }) => imageSourceToWire(image)),
+    Match.exhaustive,
+  )
+
+type WireBody = {
+  readonly model: string
+  readonly query: string
+  readonly documents: ReadonlyArray<WireDocument>
+  readonly top_n?: number
+}
+
+// No `return_documents`: the caller holds `documents` and `results[].index`
+// points back into it.
+const buildBody = (request: JinaRerankRequest): WireBody => ({
+  model: request.model,
+  query: request.query,
+  documents: request.documents.map(documentToWire),
+  ...(request.topN !== undefined && { top_n: request.topN }),
+})
+
+// ---------------------------------------------------------------------------
+// Codec - response
+// ---------------------------------------------------------------------------
+
+const WireResult = Schema.Struct({
+  index: Schema.Number,
+  relevance_score: Schema.Number,
+})
+
+const WireUsage = Schema.Struct({
+  total_tokens: Schema.optional(Schema.Number),
+})
+type WireUsage = typeof WireUsage.Type
+
+const WireResponse = Schema.Struct({
+  model: Schema.optional(Schema.String),
+  usage: Schema.optional(WireUsage),
+  results: Schema.Array(WireResult),
+})
+type WireResponse = typeof WireResponse.Type
+
+const usageOf = (usage: WireUsage | undefined): RerankUsage =>
+  usage?.total_tokens !== undefined ? { totalTokens: usage.total_tokens } : {}
+
+// Sort rather than trust wire order: the score contract promises descending.
+const toResponse = (wire: WireResponse): RerankResponse => ({
+  results: [...wire.results]
+    .sort((a, b) => b.relevance_score - a.relevance_score)
+    .map((r) => ({ index: r.index, score: r.relevance_score })),
+  usage: usageOf(wire.usage),
+})
+
+// ---------------------------------------------------------------------------
+// HTTP
+// ---------------------------------------------------------------------------
+
+const transportFailure = (cause: unknown): AiError.AiError =>
+  new AiError.Unavailable({ provider: "jina", raw: cause })
+
+const httpStatusError = (status: number, body: string): AiError.AiError => {
+  const provider = "jina"
+  const raw = body
+  if (status === 429) return new AiError.RateLimited({ provider, raw })
+  if (status === 408 || status === 504) return new AiError.Timeout({ provider, raw })
+  if (status === 401) return new AiError.AuthFailed({ provider, subtype: "auth", raw })
+  if (status === 403) return new AiError.AuthFailed({ provider, subtype: "permission", raw })
+  if (status === 402) return new AiError.AuthFailed({ provider, subtype: "billing", raw })
+  if (status === 413) return new AiError.ContextLengthExceeded({ provider, raw })
+  if (status >= 500) return new AiError.Unavailable({ provider, status, raw })
+  return new AiError.InvalidRequest({ provider, raw })
+}
+
+const baseUrl = (cfg: Config): string => cfg.baseUrl ?? "https://api.jina.ai/v1"
+
+const postRerank = (
+  cfg: Config,
+  body: WireBody,
+): Effect.Effect<WireResponse, AiError.AiError, HttpClient.HttpClient> =>
+  Effect.gen(function* () {
+    const client = yield* HttpClient.HttpClient
+    const httpRequest = HttpClientRequest.post(`${baseUrl(cfg)}/rerank`).pipe(
+      HttpClientRequest.bearerToken(cfg.apiKey),
+      HttpClientRequest.bodyJsonUnsafe(body),
+    )
+    const response = yield* client.execute(httpRequest).pipe(Effect.mapError(transportFailure))
+    if (response.status >= 400) {
+      const text = yield* response.text.pipe(Effect.orElseSucceed(() => ""))
+      return yield* httpStatusError(response.status, text)
+    }
+    const json = yield* response.json.pipe(Effect.mapError(transportFailure))
+    return yield* Schema.decodeUnknownEffect(WireResponse)(json).pipe(
+      Effect.mapError(transportFailure),
+    )
+  })
+
+const rerankImpl =
+  (cfg: Config) =>
+  (
+    request: JinaRerankRequest,
+  ): Effect.Effect<RerankResponse, AiError.AiError, HttpClient.HttpClient> =>
+    Effect.map(postRerank(cfg, buildBody(request)), toResponse)
+
+// ---------------------------------------------------------------------------
+// Constructors
+// ---------------------------------------------------------------------------
+
+/**
+ * Build a `JinaRerankerService` value. For Layer-based setup, prefer
+ * {@link layer}.
+ */
+export const make = (
+  cfg: Config,
+): Effect.Effect<JinaRerankerService, never, HttpClient.HttpClient> =>
+  Effect.map(HttpClient.HttpClient, (client) => ({
+    rerank: (request) =>
+      rerankImpl(cfg)(request).pipe(Effect.provideService(HttpClient.HttpClient, client)),
+  }))
+
+/**
+ * Layer registering both the provider-typed `JinaReranker` tag and the generic
+ * `Reranker` tag over one implementation. A `CommonRerankRequest` is
+ * structurally a `JinaRerankRequest` with string documents, so the generic
+ * registration forwards directly.
+ */
+export const layer = (
+  cfg: Config,
+): Layer.Layer<JinaReranker | Reranker, never, HttpClient.HttpClient> => {
+  const typed = Layer.effect(JinaReranker, make(cfg))
+  const generic = Layer.effect(
+    Reranker,
+    Effect.map(make(cfg), (s): RerankerService => ({ rerank: (request) => s.rerank(request) })),
+  )
+  return Layer.merge(typed, generic)
+}

package/src/index.ts

@@ -1,3 +1,4 @@
 export * as JinaEmbedding from "./JinaEmbedding.js"
 export * as JinaReader from "./JinaReader.js"
+export * as JinaReranker from "./JinaReranker.js"
 export * from "./models.js"

package/src/models.ts

@@ -26,6 +26,23 @@ export type JinaEmbeddingModel =
   // eslint-disable-next-line @typescript-eslint/ban-types
   | (string & {})
 
+/**
+ * Known Jina reranker model identifiers.
+ *
+ * - `jina-reranker-v3.5`: flagship listwise reranker, 131k context.
+ * - `jina-reranker-v3`: previous generation, schema-identical.
+ * - `jina-reranker-m0`: multimodal, takes `{ text }` / `{ image }`
+ *   documents (see `JinaRerankRequest`).
+ *
+ * Reference: https://jina.ai/reranker/
+ */
+export type JinaRerankerModel =
+  | "jina-reranker-v3.5"
+  | "jina-reranker-v3"
+  | "jina-reranker-m0"
+  // eslint-disable-next-line @typescript-eslint/ban-types
+  | (string & {})
+
 /**
  * Reader fetching engine (`X-Engine`). `browser` renders JavaScript,
  * `curl` is a fast static fetch, `auto` picks per page. Provider-specific,