feat: Rework wrapOpenAI (#114)

* Update wrapOpenAI

* Remove

* Fixes

* nits

* Feedback

* Cache

* Remove comment

* Fix

Author: jacoblee93

langsmith-java-core/build.gradle.kts

@@ -52,8 +52,8 @@ dependencies {
     api("io.opentelemetry:opentelemetry-exporter-otlp:1.32.0")
     api("io.opentelemetry.semconv:opentelemetry-semconv:1.23.1-alpha")
 
-    // OpenAI SDK (for OpenTelemetry wrappers)
-    api("com.openai:openai-java:4.6.1")
+    // OpenAI SDK (for wrapOpenAI tracing wrapper)
+    api("com.openai:openai-java:4.30.0")
 
     // Mustache template engine (for prompt template formatting)
     implementation("com.samskivert:jmustache:1.16")

langsmith-java-core/src/main/kotlin/com/langchain/smith/tracing/TraceProcessIO.kt

@@ -8,9 +8,13 @@ import java.util.function.Function
  * Typed processors for transforming inputs and outputs before they are recorded on a traced run.
  *
  * The type parameters describe what the processors receive:
- * - [PI] — the type passed to [inputs]. For 1-arg traced functions this is the raw input type; for
- *   0/2/3-arg functions this is `Map<String, Any?>` (the packed representation).
- * - [PO] — the type passed to [outputs]. This is always the raw output type of the traced function.
+ * - [PI] — the type passed to `processInputs`. Depends on the arity of the traced function:
+ *     - **1-arg** ([traceFunction]): the raw input type (e.g. `String`)
+ *     - **0-arg** ([traceSupplier]): `Map<String, Any?>` (empty map)
+ *     - **2-arg** ([traceBiFunction]): `Pair<I1, I2>`
+ *     - **3-arg** ([traceTriFunction]): `Triple<I1, I2, I3>`
+ * - [PO] — the type passed to `processOutputs`. This is always the raw output type of the traced
+ *   function.
  *
  * When [inputs] is set, it replaces the default input serialization entirely. Same for [outputs].
  *

langsmith-java-core/src/main/kotlin/com/langchain/smith/tracing/Traceable.kt

@@ -464,8 +464,9 @@ fun <I, O> traceable(block: Function<I, O>, config: TraceConfig): Function<I, O>
 fun <I1, I2, O> traceable(block: (I1, I2) -> O, config: TraceConfig): (I1, I2) -> O {
     val resolvedConfig = resolveConfig(config, block)
     return { i1, i2 ->
-        val packed = mapOf<String, Any?>("args" to listOf(i1, i2))
-        val serializedInputs = applyProcessInputs(resolvedConfig, packed) ?: packed
+        val rawArgs = Pair(i1, i2)
+        val serializedInputs =
+            applyProcessInputs(resolvedConfig, rawArgs) ?: mapOf("args" to listOf(i1, i2))
         executeTraced(resolvedConfig, serializedInputs) { block(i1, i2) }
     }
 }
@@ -484,8 +485,9 @@ fun <I1, I2, O> traceable(
 fun <I1, I2, I3, O> traceable(block: (I1, I2, I3) -> O, config: TraceConfig): (I1, I2, I3) -> O {
     val resolvedConfig = resolveConfig(config, block)
     return { i1, i2, i3 ->
-        val packed = mapOf<String, Any?>("args" to listOf(i1, i2, i3))
-        val serializedInputs = applyProcessInputs(resolvedConfig, packed) ?: packed
+        val rawArgs = Triple(i1, i2, i3)
+        val serializedInputs =
+            applyProcessInputs(resolvedConfig, rawArgs) ?: mapOf("args" to listOf(i1, i2, i3))
         executeTraced(resolvedConfig, serializedInputs) { block(i1, i2, i3) }
     }
 }