@openrouter/sdk 0.0.0-beta.0

package/src/react-query/chatSend.ts

@@ -0,0 +1,87 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  MutationKey,
+  useMutation,
+  UseMutationResult,
+} from "@tanstack/react-query";
+import { OpenRouterCore } from "../core.js";
+import { chatSend, SendAcceptEnum } from "../funcs/chatSend.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.js";
+import * as models from "../models/index.js";
+import * as operations from "../models/operations/index.js";
+import { unwrapAsync } from "../types/fp.js";
+import { useOpenRouterContext } from "./_context.js";
+import { MutationHookOptions } from "./_types.js";
+
+export { SendAcceptEnum } from "../funcs/chatSend.js";
+
+export type ChatSendMutationVariables = {
+  request?: models.ChatGenerationParams | undefined;
+  options?: RequestOptions & { acceptHeaderOverride?: SendAcceptEnum };
+};
+
+export type ChatSendMutationData = operations.PostChatCompletionsResponse;
+
+/**
+ * Create a chat completion
+ *
+ * @remarks
+ * Sends a request for a model response for the given chat conversation. Supports both streaming and non-streaming modes.
+ */
+export function useChatSendMutation(
+  options?: MutationHookOptions<
+    ChatSendMutationData,
+    Error,
+    ChatSendMutationVariables
+  >,
+): UseMutationResult<ChatSendMutationData, Error, ChatSendMutationVariables> {
+  const client = useOpenRouterContext();
+  return useMutation({
+    ...buildChatSendMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyChatSend(): MutationKey {
+  return ["@openrouter/sdk", "chat", "send"];
+}
+
+export function buildChatSendMutation(
+  client$: OpenRouterCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: ChatSendMutationVariables,
+  ) => Promise<ChatSendMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyChatSend(),
+    mutationFn: function chatSendMutationFn({
+      request,
+      options,
+    }): Promise<ChatSendMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(chatSend(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

package/src/react-query/embeddingsPostEmbeddings.ts

@@ -0,0 +1,89 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  MutationKey,
+  useMutation,
+  UseMutationResult,
+} from "@tanstack/react-query";
+import { OpenRouterCore } from "../core.js";
+import { embeddingsPostEmbeddings } from "../funcs/embeddingsPostEmbeddings.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.js";
+import * as operations from "../models/operations/index.js";
+import { unwrapAsync } from "../types/fp.js";
+import { useOpenRouterContext } from "./_context.js";
+import { MutationHookOptions } from "./_types.js";
+
+export type EmbeddingsPostEmbeddingsMutationVariables = {
+  request?: operations.PostEmbeddingsRequest | undefined;
+  options?: RequestOptions;
+};
+
+export type EmbeddingsPostEmbeddingsMutationData =
+  operations.PostEmbeddingsResponse;
+
+/**
+ * Submit an embedding request
+ *
+ * @remarks
+ * Submits an embedding request to the embeddings router
+ */
+export function useEmbeddingsPostEmbeddingsMutation(
+  options?: MutationHookOptions<
+    EmbeddingsPostEmbeddingsMutationData,
+    Error,
+    EmbeddingsPostEmbeddingsMutationVariables
+  >,
+): UseMutationResult<
+  EmbeddingsPostEmbeddingsMutationData,
+  Error,
+  EmbeddingsPostEmbeddingsMutationVariables
+> {
+  const client = useOpenRouterContext();
+  return useMutation({
+    ...buildEmbeddingsPostEmbeddingsMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyEmbeddingsPostEmbeddings(): MutationKey {
+  return ["@openrouter/sdk", "Embeddings", "postEmbeddings"];
+}
+
+export function buildEmbeddingsPostEmbeddingsMutation(
+  client$: OpenRouterCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: EmbeddingsPostEmbeddingsMutationVariables,
+  ) => Promise<EmbeddingsPostEmbeddingsMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyEmbeddingsPostEmbeddings(),
+    mutationFn: function embeddingsPostEmbeddingsMutationFn({
+      request,
+      options,
+    }): Promise<EmbeddingsPostEmbeddingsMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(embeddingsPostEmbeddings(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

package/src/react-query/endpointsGetEndpointsZdr.ts

@@ -0,0 +1,119 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  InvalidateQueryFilters,
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+  useQuery,
+  UseQueryResult,
+  useSuspenseQuery,
+  UseSuspenseQueryResult,
+} from "@tanstack/react-query";
+import { OpenRouterCore } from "../core.js";
+import { endpointsGetEndpointsZdr } from "../funcs/endpointsGetEndpointsZdr.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.js";
+import * as operations from "../models/operations/index.js";
+import { unwrapAsync } from "../types/fp.js";
+import { useOpenRouterContext } from "./_context.js";
+import { QueryHookOptions, SuspenseQueryHookOptions } from "./_types.js";
+
+export type EndpointsGetEndpointsZdrQueryData =
+  operations.GetEndpointsZdrResponse;
+
+/**
+ * Preview the impact of ZDR on the available endpoints
+ */
+export function useEndpointsGetEndpointsZdr(
+  options?: QueryHookOptions<EndpointsGetEndpointsZdrQueryData>,
+): UseQueryResult<EndpointsGetEndpointsZdrQueryData, Error> {
+  const client = useOpenRouterContext();
+  return useQuery({
+    ...buildEndpointsGetEndpointsZdrQuery(
+      client,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * Preview the impact of ZDR on the available endpoints
+ */
+export function useEndpointsGetEndpointsZdrSuspense(
+  options?: SuspenseQueryHookOptions<EndpointsGetEndpointsZdrQueryData>,
+): UseSuspenseQueryResult<EndpointsGetEndpointsZdrQueryData, Error> {
+  const client = useOpenRouterContext();
+  return useSuspenseQuery({
+    ...buildEndpointsGetEndpointsZdrQuery(
+      client,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function prefetchEndpointsGetEndpointsZdr(
+  queryClient: QueryClient,
+  client$: OpenRouterCore,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildEndpointsGetEndpointsZdrQuery(
+      client$,
+    ),
+  });
+}
+
+export function setEndpointsGetEndpointsZdrData(
+  client: QueryClient,
+  data: EndpointsGetEndpointsZdrQueryData,
+): EndpointsGetEndpointsZdrQueryData | undefined {
+  const key = queryKeyEndpointsGetEndpointsZdr();
+
+  return client.setQueryData<EndpointsGetEndpointsZdrQueryData>(key, data);
+}
+
+export function invalidateAllEndpointsGetEndpointsZdr(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@openrouter/sdk", "Endpoints", "getEndpointsZdr"],
+  });
+}
+
+export function buildEndpointsGetEndpointsZdrQuery(
+  client$: OpenRouterCore,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<EndpointsGetEndpointsZdrQueryData>;
+} {
+  return {
+    queryKey: queryKeyEndpointsGetEndpointsZdr(),
+    queryFn: async function endpointsGetEndpointsZdrQueryFn(
+      ctx,
+    ): Promise<EndpointsGetEndpointsZdrQueryData> {
+      const sig = combineSignals(ctx.signal, options?.fetchOptions?.signal);
+      const mergedOptions = {
+        ...options,
+        fetchOptions: { ...options?.fetchOptions, signal: sig },
+      };
+
+      return unwrapAsync(endpointsGetEndpointsZdr(
+        client$,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyEndpointsGetEndpointsZdr(): QueryKey {
+  return ["@openrouter/sdk", "Endpoints", "getEndpointsZdr"];
+}

package/src/react-query/endpointsGetModelsAuthorSlugEndpoints.ts

@@ -0,0 +1,168 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  InvalidateQueryFilters,
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+  useQuery,
+  UseQueryResult,
+  useSuspenseQuery,
+  UseSuspenseQueryResult,
+} from "@tanstack/react-query";
+import { OpenRouterCore } from "../core.js";
+import { endpointsGetModelsAuthorSlugEndpoints } from "../funcs/endpointsGetModelsAuthorSlugEndpoints.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.js";
+import * as operations from "../models/operations/index.js";
+import { unwrapAsync } from "../types/fp.js";
+import { useOpenRouterContext } from "./_context.js";
+import {
+  QueryHookOptions,
+  SuspenseQueryHookOptions,
+  TupleToPrefixes,
+} from "./_types.js";
+
+export type EndpointsGetModelsAuthorSlugEndpointsQueryData =
+  operations.GetModelsAuthorSlugEndpointsResponse;
+
+/**
+ * List all endpoints for a model
+ */
+export function useEndpointsGetModelsAuthorSlugEndpoints(
+  request: operations.GetModelsAuthorSlugEndpointsRequestRequest,
+  options?: QueryHookOptions<EndpointsGetModelsAuthorSlugEndpointsQueryData>,
+): UseQueryResult<EndpointsGetModelsAuthorSlugEndpointsQueryData, Error> {
+  const client = useOpenRouterContext();
+  return useQuery({
+    ...buildEndpointsGetModelsAuthorSlugEndpointsQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * List all endpoints for a model
+ */
+export function useEndpointsGetModelsAuthorSlugEndpointsSuspense(
+  request: operations.GetModelsAuthorSlugEndpointsRequestRequest,
+  options?: SuspenseQueryHookOptions<
+    EndpointsGetModelsAuthorSlugEndpointsQueryData
+  >,
+): UseSuspenseQueryResult<
+  EndpointsGetModelsAuthorSlugEndpointsQueryData,
+  Error
+> {
+  const client = useOpenRouterContext();
+  return useSuspenseQuery({
+    ...buildEndpointsGetModelsAuthorSlugEndpointsQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function prefetchEndpointsGetModelsAuthorSlugEndpoints(
+  queryClient: QueryClient,
+  client$: OpenRouterCore,
+  request: operations.GetModelsAuthorSlugEndpointsRequestRequest,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildEndpointsGetModelsAuthorSlugEndpointsQuery(
+      client$,
+      request,
+    ),
+  });
+}
+
+export function setEndpointsGetModelsAuthorSlugEndpointsData(
+  client: QueryClient,
+  queryKeyBase: [author: string, slug: string],
+  data: EndpointsGetModelsAuthorSlugEndpointsQueryData,
+): EndpointsGetModelsAuthorSlugEndpointsQueryData | undefined {
+  const key = queryKeyEndpointsGetModelsAuthorSlugEndpoints(...queryKeyBase);
+
+  return client.setQueryData<EndpointsGetModelsAuthorSlugEndpointsQueryData>(
+    key,
+    data,
+  );
+}
+
+export function invalidateEndpointsGetModelsAuthorSlugEndpoints(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<[author: string, slug: string]>,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: [
+      "@openrouter/sdk",
+      "Endpoints",
+      "getModelsAuthorSlugEndpoints",
+      ...queryKeyBase,
+    ],
+  });
+}
+
+export function invalidateAllEndpointsGetModelsAuthorSlugEndpoints(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@openrouter/sdk", "Endpoints", "getModelsAuthorSlugEndpoints"],
+  });
+}
+
+export function buildEndpointsGetModelsAuthorSlugEndpointsQuery(
+  client$: OpenRouterCore,
+  request: operations.GetModelsAuthorSlugEndpointsRequestRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<EndpointsGetModelsAuthorSlugEndpointsQueryData>;
+} {
+  return {
+    queryKey: queryKeyEndpointsGetModelsAuthorSlugEndpoints(
+      request.author,
+      request.slug,
+    ),
+    queryFn: async function endpointsGetModelsAuthorSlugEndpointsQueryFn(
+      ctx,
+    ): Promise<EndpointsGetModelsAuthorSlugEndpointsQueryData> {
+      const sig = combineSignals(ctx.signal, options?.fetchOptions?.signal);
+      const mergedOptions = {
+        ...options,
+        fetchOptions: { ...options?.fetchOptions, signal: sig },
+      };
+
+      return unwrapAsync(endpointsGetModelsAuthorSlugEndpoints(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyEndpointsGetModelsAuthorSlugEndpoints(
+  author: string,
+  slug: string,
+): QueryKey {
+  return [
+    "@openrouter/sdk",
+    "Endpoints",
+    "getModelsAuthorSlugEndpoints",
+    author,
+    slug,
+  ];
+}

package/src/react-query/generationsGetGeneration.ts

@@ -0,0 +1,150 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  InvalidateQueryFilters,
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+  useQuery,
+  UseQueryResult,
+  useSuspenseQuery,
+  UseSuspenseQueryResult,
+} from "@tanstack/react-query";
+import { OpenRouterCore } from "../core.js";
+import { generationsGetGeneration } from "../funcs/generationsGetGeneration.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.js";
+import * as operations from "../models/operations/index.js";
+import { unwrapAsync } from "../types/fp.js";
+import { useOpenRouterContext } from "./_context.js";
+import {
+  QueryHookOptions,
+  SuspenseQueryHookOptions,
+  TupleToPrefixes,
+} from "./_types.js";
+
+export type GenerationsGetGenerationQueryData =
+  operations.GetGenerationResponse;
+
+/**
+ * Get request & usage metadata for a generation
+ */
+export function useGenerationsGetGeneration(
+  request: operations.GetGenerationRequest,
+  options?: QueryHookOptions<GenerationsGetGenerationQueryData>,
+): UseQueryResult<GenerationsGetGenerationQueryData, Error> {
+  const client = useOpenRouterContext();
+  return useQuery({
+    ...buildGenerationsGetGenerationQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+/**
+ * Get request & usage metadata for a generation
+ */
+export function useGenerationsGetGenerationSuspense(
+  request: operations.GetGenerationRequest,
+  options?: SuspenseQueryHookOptions<GenerationsGetGenerationQueryData>,
+): UseSuspenseQueryResult<GenerationsGetGenerationQueryData, Error> {
+  const client = useOpenRouterContext();
+  return useSuspenseQuery({
+    ...buildGenerationsGetGenerationQuery(
+      client,
+      request,
+      options,
+    ),
+    ...options,
+  });
+}
+
+export function prefetchGenerationsGetGeneration(
+  queryClient: QueryClient,
+  client$: OpenRouterCore,
+  request: operations.GetGenerationRequest,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildGenerationsGetGenerationQuery(
+      client$,
+      request,
+    ),
+  });
+}
+
+export function setGenerationsGetGenerationData(
+  client: QueryClient,
+  queryKeyBase: [parameters: { id: string }],
+  data: GenerationsGetGenerationQueryData,
+): GenerationsGetGenerationQueryData | undefined {
+  const key = queryKeyGenerationsGetGeneration(...queryKeyBase);
+
+  return client.setQueryData<GenerationsGetGenerationQueryData>(key, data);
+}
+
+export function invalidateGenerationsGetGeneration(
+  client: QueryClient,
+  queryKeyBase: TupleToPrefixes<[parameters: { id: string }]>,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: [
+      "@openrouter/sdk",
+      "Generations",
+      "getGeneration",
+      ...queryKeyBase,
+    ],
+  });
+}
+
+export function invalidateAllGenerationsGetGeneration(
+  client: QueryClient,
+  filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
+): Promise<void> {
+  return client.invalidateQueries({
+    ...filters,
+    queryKey: ["@openrouter/sdk", "Generations", "getGeneration"],
+  });
+}
+
+export function buildGenerationsGetGenerationQuery(
+  client$: OpenRouterCore,
+  request: operations.GetGenerationRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<GenerationsGetGenerationQueryData>;
+} {
+  return {
+    queryKey: queryKeyGenerationsGetGeneration({ id: request.id }),
+    queryFn: async function generationsGetGenerationQueryFn(
+      ctx,
+    ): Promise<GenerationsGetGenerationQueryData> {
+      const sig = combineSignals(ctx.signal, options?.fetchOptions?.signal);
+      const mergedOptions = {
+        ...options,
+        fetchOptions: { ...options?.fetchOptions, signal: sig },
+      };
+
+      return unwrapAsync(generationsGetGeneration(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyGenerationsGetGeneration(
+  parameters: { id: string },
+): QueryKey {
+  return ["@openrouter/sdk", "Generations", "getGeneration", parameters];
+}