@effect/ai-openai 0.11.4 → 0.12.0

package/src/OpenAiClient.ts

@@ -20,6 +20,7 @@ import * as Option from "effect/Option"
 import * as Redacted from "effect/Redacted"
 import * as Stream from "effect/Stream"
 import * as Generated from "./Generated.js"
+import { OpenAiConfig } from "./OpenAiConfig.js"
 
 /**
  * @since 1.0.0
@@ -79,7 +80,12 @@ export const make = (options: {
       options.transformClient ? options.transformClient : identity
     )
     const httpClientOk = HttpClient.filterStatusOk(httpClient)
-    const client = Generated.make(httpClient)
+    const client = Generated.make(httpClient, {
+      transformClient: (client) =>
+        OpenAiConfig.getOrUndefined.pipe(
+          Effect.map((config) => config?.transformClient ? config.transformClient(client) : client)
+        )
+    })
     const streamRequest = <A = unknown>(request: HttpClientRequest.HttpClientRequest) =>
       httpClientOk.execute(request).pipe(
         Effect.map((r) => r.stream),

package/src/OpenAiConfig.ts

@@ -1,8 +1,10 @@
 /**
  * @since 1.0.0
  */
+import type { HttpClient } from "@effect/platform/HttpClient"
 import * as Context from "effect/Context"
 import * as Effect from "effect/Effect"
+import { dual } from "effect/Function"
 import type { Simplify } from "effect/Types"
 import type * as Generated from "./Generated.js"
 
@@ -12,14 +14,7 @@ import type * as Generated from "./Generated.js"
  */
 export class OpenAiConfig extends Context.Tag("@effect/ai-openai/OpenAiConfig")<
   OpenAiConfig,
-  Simplify<
-    Partial<
-      Omit<
-        typeof Generated.CreateChatCompletionRequest.Encoded,
-        "messages" | "tools" | "tool_choice" | "stream" | "stream_options" | "functions"
-      >
-    >
-  >
+  OpenAiConfig.Service
 >() {
   /**
    * @since 1.0.0
@@ -29,3 +24,53 @@ export class OpenAiConfig extends Context.Tag("@effect/ai-openai/OpenAiConfig")<
     (context) => context.unsafeMap.get(OpenAiConfig.key)
   )
 }
+
+/**
+ * @since 1.0.0
+ * @category models
+ */
+export declare namespace OpenAiConfig {
+  /**
+   * @since 1.0.0
+   * @category models
+   */
+  export interface Service extends
+    Simplify<
+      Partial<
+        Omit<
+          typeof Generated.CreateChatCompletionRequest.Encoded,
+          "messages" | "tools" | "tool_choice" | "stream" | "stream_options" | "functions"
+        >
+      >
+    >
+  {
+    readonly transformClient?: (client: HttpClient) => HttpClient
+  }
+}
+
+/**
+ * @since 1.0.0
+ * @category configuration
+ */
+export const withClientTransform = dual<
+  /**
+   * @since 1.0.0
+   * @category configuration
+   */
+  (transform: (client: HttpClient) => HttpClient) => <A, E, R>(self: Effect.Effect<A, E, R>) => Effect.Effect<A, E, R>,
+  /**
+   * @since 1.0.0
+   * @category configuration
+   */
+  <A, E, R>(
+    self: Effect.Effect<A, E, R>,
+    transform: (client: HttpClient) => HttpClient
+  ) => Effect.Effect<A, E, R>
+>(
+  2,
+  (self, transformClient) =>
+    Effect.flatMap(
+      OpenAiConfig.getOrUndefined,
+      (config) => Effect.provideService(self, OpenAiConfig, { ...config, transformClient })
+    )
+)