@homeflare/distilled-litellm 0.2.0

package/src/services/a2a.ts

@@ -0,0 +1,250 @@
+// AUTO-GENERATED by scripts/generate.ts from .generated-specs (pinned litellm-openapi @ 1.100.0 — see README). Do not edit.
+import * as S from "@distilled.cloud/core/schema";
+import * as API from "@distilled.cloud/core/api";
+import * as C from "@distilled.cloud/core/category";
+import * as T from "../traits.ts";
+import {
+  LitellmProtocol,
+  type LitellmOpError,
+  type LitellmOpContext,
+} from "../protocol.ts";
+import * as Retry from "../retry.ts";
+
+export type { LitellmOpError, LitellmOpContext };
+
+export class UnprocessableEntity
+  extends /*@__PURE__*/ T.applyErrorMatchers(
+    /*@__PURE__*/ S.TaggedError<UnprocessableEntity>()("UnprocessableEntity", {
+      code: S.Number,
+      message: S.String,
+    }).pipe(C.withBadRequestError),
+    [{ status: 422 }],
+  ) {}
+
+export interface GetAgentCardA2aAgentIdWellKnownAgentCardJsonGetRequest {
+  agent_id: string;
+}
+export const GetAgentCardA2aAgentIdWellKnownAgentCardJsonGetRequest =
+  /*@__PURE__*/ S.suspend(() =>
+    S.Struct({
+      agent_id: S.String.pipe(T.Label()),
+    }).pipe(
+      T.Http({
+        method: "GET",
+        uri: "/a2a/{agent_id}/.well-known/agent-card.json",
+        code: 200,
+      }),
+    ),
+  ).annotate({
+    identifier: "GetAgentCardA2aAgentIdWellKnownAgentCardJsonGetRequest",
+  }) as any as S.Schema<GetAgentCardA2aAgentIdWellKnownAgentCardJsonGetRequest>;
+
+export interface GetAgentCardA2aAgentIdWellKnownAgentCardJsonGetResponse {
+  body: unknown;
+}
+export const GetAgentCardA2aAgentIdWellKnownAgentCardJsonGetResponse =
+  /*@__PURE__*/ S.suspend(() =>
+    S.Struct({
+      body: S.Unknown,
+    }),
+  ).annotate({
+    identifier: "GetAgentCardA2aAgentIdWellKnownAgentCardJsonGetResponse",
+  }) as any as S.Schema<GetAgentCardA2aAgentIdWellKnownAgentCardJsonGetResponse>;
+
+export interface GetAgentCardA2aAgentIdWellKnownAgentJsonGetRequest {
+  agent_id: string;
+}
+export const GetAgentCardA2aAgentIdWellKnownAgentJsonGetRequest =
+  /*@__PURE__*/ S.suspend(() =>
+    S.Struct({
+      agent_id: S.String.pipe(T.Label()),
+    }).pipe(
+      T.Http({
+        method: "GET",
+        uri: "/a2a/{agent_id}/.well-known/agent.json",
+        code: 200,
+      }),
+    ),
+  ).annotate({
+    identifier: "GetAgentCardA2aAgentIdWellKnownAgentJsonGetRequest",
+  }) as any as S.Schema<GetAgentCardA2aAgentIdWellKnownAgentJsonGetRequest>;
+
+export interface GetAgentCardA2aAgentIdWellKnownAgentJsonGetResponse {
+  body: unknown;
+}
+export const GetAgentCardA2aAgentIdWellKnownAgentJsonGetResponse =
+  /*@__PURE__*/ S.suspend(() =>
+    S.Struct({
+      body: S.Unknown,
+    }),
+  ).annotate({
+    identifier: "GetAgentCardA2aAgentIdWellKnownAgentJsonGetResponse",
+  }) as any as S.Schema<GetAgentCardA2aAgentIdWellKnownAgentJsonGetResponse>;
+
+export interface InvokeAgentA2aA2aAgentIdMessageSendPostRequest {
+  agent_id: string;
+}
+export const InvokeAgentA2aA2aAgentIdMessageSendPostRequest =
+  /*@__PURE__*/ S.suspend(() =>
+    S.Struct({
+      agent_id: S.String.pipe(T.Label()),
+    }).pipe(
+      T.Http({
+        method: "POST",
+        uri: "/a2a/{agent_id}/message/send",
+        code: 200,
+      }),
+    ),
+  ).annotate({
+    identifier: "InvokeAgentA2aA2aAgentIdMessageSendPostRequest",
+  }) as any as S.Schema<InvokeAgentA2aA2aAgentIdMessageSendPostRequest>;
+
+export interface InvokeAgentA2aA2aAgentIdMessageSendPostResponse {
+  body: unknown;
+}
+export const InvokeAgentA2aA2aAgentIdMessageSendPostResponse =
+  /*@__PURE__*/ S.suspend(() =>
+    S.Struct({
+      body: S.Unknown,
+    }),
+  ).annotate({
+    identifier: "InvokeAgentA2aA2aAgentIdMessageSendPostResponse",
+  }) as any as S.Schema<InvokeAgentA2aA2aAgentIdMessageSendPostResponse>;
+
+export interface InvokeAgentA2aA2aAgentIdPostRequest {
+  agent_id: string;
+}
+export const InvokeAgentA2aA2aAgentIdPostRequest = /*@__PURE__*/ S.suspend(() =>
+  S.Struct({
+    agent_id: S.String.pipe(T.Label()),
+  }).pipe(T.Http({ method: "POST", uri: "/a2a/{agent_id}", code: 200 })),
+).annotate({
+  identifier: "InvokeAgentA2aA2aAgentIdPostRequest",
+}) as any as S.Schema<InvokeAgentA2aA2aAgentIdPostRequest>;
+
+export interface InvokeAgentA2aA2aAgentIdPostResponse {
+  body: unknown;
+}
+export const InvokeAgentA2aA2aAgentIdPostResponse = /*@__PURE__*/ S.suspend(
+  () =>
+    S.Struct({
+      body: S.Unknown,
+    }),
+).annotate({
+  identifier: "InvokeAgentA2aA2aAgentIdPostResponse",
+}) as any as S.Schema<InvokeAgentA2aA2aAgentIdPostResponse>;
+
+export interface InvokeAgentA2aV1A2aAgentIdMessageSendPostRequest {
+  agent_id: string;
+}
+export const InvokeAgentA2aV1A2aAgentIdMessageSendPostRequest =
+  /*@__PURE__*/ S.suspend(() =>
+    S.Struct({
+      agent_id: S.String.pipe(T.Label()),
+    }).pipe(
+      T.Http({
+        method: "POST",
+        uri: "/v1/a2a/{agent_id}/message/send",
+        code: 200,
+      }),
+    ),
+  ).annotate({
+    identifier: "InvokeAgentA2aV1A2aAgentIdMessageSendPostRequest",
+  }) as any as S.Schema<InvokeAgentA2aV1A2aAgentIdMessageSendPostRequest>;
+
+export interface InvokeAgentA2aV1A2aAgentIdMessageSendPostResponse {
+  body: unknown;
+}
+export const InvokeAgentA2aV1A2aAgentIdMessageSendPostResponse =
+  /*@__PURE__*/ S.suspend(() =>
+    S.Struct({
+      body: S.Unknown,
+    }),
+  ).annotate({
+    identifier: "InvokeAgentA2aV1A2aAgentIdMessageSendPostResponse",
+  }) as any as S.Schema<InvokeAgentA2aV1A2aAgentIdMessageSendPostResponse>;
+
+export type GetAgentCardA2aAgentIdWellKnownAgentCardJsonGetError =
+  | UnprocessableEntity
+  | LitellmOpError;
+/** Get Agent Card Get the agent card for an agent (A2A discovery endpoint). Supports both standard paths: - /.well-known/agent-card.json - /.well-known/agent.json The URL in the agent card is rewritten to point to the LiteLLM proxy, so all subsequent A2A calls go through LiteLLM for logging and cost tracking. */
+export const getAgentCardA2aAgentIdWellKnownAgentCardJsonGet: API.OperationMethod<
+  GetAgentCardA2aAgentIdWellKnownAgentCardJsonGetRequest,
+  GetAgentCardA2aAgentIdWellKnownAgentCardJsonGetResponse,
+  GetAgentCardA2aAgentIdWellKnownAgentCardJsonGetError,
+  LitellmOpContext
+> = /*@__PURE__*/ API.make(() => ({
+  input: GetAgentCardA2aAgentIdWellKnownAgentCardJsonGetRequest,
+  output: GetAgentCardA2aAgentIdWellKnownAgentCardJsonGetResponse,
+  errors: [UnprocessableEntity],
+  protocol: LitellmProtocol,
+  retry: Retry.Retry,
+}));
+
+export type GetAgentCardA2aAgentIdWellKnownAgentJsonGetError =
+  | UnprocessableEntity
+  | LitellmOpError;
+/** Get Agent Card Get the agent card for an agent (A2A discovery endpoint). Supports both standard paths: - /.well-known/agent-card.json - /.well-known/agent.json The URL in the agent card is rewritten to point to the LiteLLM proxy, so all subsequent A2A calls go through LiteLLM for logging and cost tracking. */
+export const getAgentCardA2aAgentIdWellKnownAgentJsonGet: API.OperationMethod<
+  GetAgentCardA2aAgentIdWellKnownAgentJsonGetRequest,
+  GetAgentCardA2aAgentIdWellKnownAgentJsonGetResponse,
+  GetAgentCardA2aAgentIdWellKnownAgentJsonGetError,
+  LitellmOpContext
+> = /*@__PURE__*/ API.make(() => ({
+  input: GetAgentCardA2aAgentIdWellKnownAgentJsonGetRequest,
+  output: GetAgentCardA2aAgentIdWellKnownAgentJsonGetResponse,
+  errors: [UnprocessableEntity],
+  protocol: LitellmProtocol,
+  retry: Retry.Retry,
+}));
+
+export type InvokeAgentA2aA2aAgentIdMessageSendPostError =
+  | UnprocessableEntity
+  | LitellmOpError;
+/** Invoke Agent A2A Invoke an agent using the A2A protocol (JSON-RPC 2.0). Supported methods: - message/send: Send a message and get a response - message/stream: Send a message and stream the response */
+export const invokeAgentA2aA2aAgentIdMessageSendPost: API.OperationMethod<
+  InvokeAgentA2aA2aAgentIdMessageSendPostRequest,
+  InvokeAgentA2aA2aAgentIdMessageSendPostResponse,
+  InvokeAgentA2aA2aAgentIdMessageSendPostError,
+  LitellmOpContext
+> = /*@__PURE__*/ API.make(() => ({
+  input: InvokeAgentA2aA2aAgentIdMessageSendPostRequest,
+  output: InvokeAgentA2aA2aAgentIdMessageSendPostResponse,
+  errors: [UnprocessableEntity],
+  protocol: LitellmProtocol,
+  retry: Retry.Retry,
+}));
+
+export type InvokeAgentA2aA2aAgentIdPostError =
+  | UnprocessableEntity
+  | LitellmOpError;
+/** Invoke Agent A2A Invoke an agent using the A2A protocol (JSON-RPC 2.0). Supported methods: - message/send: Send a message and get a response - message/stream: Send a message and stream the response */
+export const invokeAgentA2aA2aAgentIdPost: API.OperationMethod<
+  InvokeAgentA2aA2aAgentIdPostRequest,
+  InvokeAgentA2aA2aAgentIdPostResponse,
+  InvokeAgentA2aA2aAgentIdPostError,
+  LitellmOpContext
+> = /*@__PURE__*/ API.make(() => ({
+  input: InvokeAgentA2aA2aAgentIdPostRequest,
+  output: InvokeAgentA2aA2aAgentIdPostResponse,
+  errors: [UnprocessableEntity],
+  protocol: LitellmProtocol,
+  retry: Retry.Retry,
+}));
+
+export type InvokeAgentA2aV1A2aAgentIdMessageSendPostError =
+  | UnprocessableEntity
+  | LitellmOpError;
+/** Invoke Agent A2A Invoke an agent using the A2A protocol (JSON-RPC 2.0). Supported methods: - message/send: Send a message and get a response - message/stream: Send a message and stream the response */
+export const invokeAgentA2aV1A2aAgentIdMessageSendPost: API.OperationMethod<
+  InvokeAgentA2aV1A2aAgentIdMessageSendPostRequest,
+  InvokeAgentA2aV1A2aAgentIdMessageSendPostResponse,
+  InvokeAgentA2aV1A2aAgentIdMessageSendPostError,
+  LitellmOpContext
+> = /*@__PURE__*/ API.make(() => ({
+  input: InvokeAgentA2aV1A2aAgentIdMessageSendPostRequest,
+  output: InvokeAgentA2aV1A2aAgentIdMessageSendPostResponse,
+  errors: [UnprocessableEntity],
+  protocol: LitellmProtocol,
+  retry: Retry.Retry,
+}));

package/src/services/a2a_registration.ts

@@ -0,0 +1,94 @@
+// AUTO-GENERATED by scripts/generate.ts from .generated-specs (pinned litellm-openapi @ 1.100.0 — see README). Do not edit.
+import * as S from "@distilled.cloud/core/schema";
+import * as API from "@distilled.cloud/core/api";
+import * as C from "@distilled.cloud/core/category";
+import * as T from "../traits.ts";
+import {
+  LitellmProtocol,
+  type LitellmOpError,
+  type LitellmOpContext,
+} from "../protocol.ts";
+import * as Retry from "../retry.ts";
+
+export type { LitellmOpError, LitellmOpContext };
+
+export class UnprocessableEntity
+  extends /*@__PURE__*/ T.applyErrorMatchers(
+    /*@__PURE__*/ S.TaggedError<UnprocessableEntity>()("UnprocessableEntity", {
+      code: S.Number,
+      message: S.String,
+    }).pipe(C.withBadRequestError),
+    [{ status: 422 }],
+  ) {}
+
+/** How to locate the upstream agent card. String-valued so it serializes cleanly over JSON / Pydantic. */
+export type DiscoveryMode = "well_known_fallback" | "langgraph_platform";
+export const DiscoveryMode = S.String;
+
+export type PostDiscoverAgentCardV1A2aDiscoverRequestParamsMap = {
+  [key: string]: unknown | undefined;
+};
+export const PostDiscoverAgentCardV1A2aDiscoverRequestParamsMap =
+  /*@__PURE__*/ S.Record(
+    S.String,
+    S.Unknown,
+  ) as any as S.Schema<PostDiscoverAgentCardV1A2aDiscoverRequestParamsMap>;
+
+export interface PostDiscoverAgentCardV1A2aDiscoverRequest {
+  /** How to locate the upstream card. ``well_known_fallback`` for pure A2A agents (try standard paths); ``langgraph_platform`` for LangGraph Platform deployments where the card is shared across assistants and disambiguated by a query parameter. */
+  discovery_mode?: DiscoveryMode | (string & {});
+  /** Mode-specific parameters. ``langgraph_platform`` requires ``{'assistant_id': <id>}``. ``well_known_fallback`` ignores this. */
+  params?: PostDiscoverAgentCardV1A2aDiscoverRequestParamsMap | null;
+  /** Base URL of the upstream agent. Behavior depends on ``discovery_mode``: ``well_known_fallback`` (default) tries /.well-known/agent-card.json, /.well-known/agent.json, /agent.json under this URL in order; ``langgraph_platform`` hits ``/.well-known/agent-card.json?assistant_id=<id>`` instead. */
+  url: string;
+}
+export const PostDiscoverAgentCardV1A2aDiscoverRequest =
+  /*@__PURE__*/ S.suspend(() =>
+    S.Struct({
+      discovery_mode: S.optional(DiscoveryMode),
+      params: S.optional(
+        S.NullOr(PostDiscoverAgentCardV1A2aDiscoverRequestParamsMap),
+      ),
+      url: S.String,
+    }).pipe(T.Http({ method: "POST", uri: "/v1/a2a/discover", code: 200 })),
+  ).annotate({
+    identifier: "PostDiscoverAgentCardV1A2aDiscoverRequest",
+  }) as any as S.Schema<PostDiscoverAgentCardV1A2aDiscoverRequest>;
+
+export type DiscoverAgentResponseAgentCardMap = {
+  [key: string]: unknown | undefined;
+};
+export const DiscoverAgentResponseAgentCardMap = /*@__PURE__*/ S.Record(
+  S.String,
+  S.Unknown,
+) as any as S.Schema<DiscoverAgentResponseAgentCardMap>;
+
+export interface DiscoverAgentResponse {
+  agent_card: DiscoverAgentResponseAgentCardMap;
+  url: string;
+}
+export const DiscoverAgentResponse = /*@__PURE__*/ S.suspend(() =>
+  S.Struct({
+    agent_card: DiscoverAgentResponseAgentCardMap,
+    url: S.String,
+  }),
+).annotate({
+  identifier: "DiscoverAgentResponse",
+}) as any as S.Schema<DiscoverAgentResponse>;
+
+export type PostDiscoverAgentCardV1A2aDiscoverError =
+  | UnprocessableEntity
+  | LitellmOpError;
+/** Discover Agent Card Fetch the upstream agent's well-known card so the UI can show the admin which skills/capabilities the agent exposes. Only proxy admins can call this — the UI uses it during agent registration, and we don't want arbitrary keys probing internal URLs. Example: ```bash curl -X POST "http://localhost:4000/v1/a2a/discover" \ -H "Authorization: Bearer <admin_key>" \ -H "Content-Type: application/json" \ -d '{"url": "https://upstream-agent.example.com"}' ``` */
+export const postDiscoverAgentCardV1A2aDiscover: API.OperationMethod<
+  PostDiscoverAgentCardV1A2aDiscoverRequest,
+  DiscoverAgentResponse,
+  PostDiscoverAgentCardV1A2aDiscoverError,
+  LitellmOpContext
+> = /*@__PURE__*/ API.make(() => ({
+  input: PostDiscoverAgentCardV1A2aDiscoverRequest,
+  output: DiscoverAgentResponse,
+  errors: [UnprocessableEntity],
+  protocol: LitellmProtocol,
+  retry: Retry.Retry,
+}));