fix(manual-tracing): add missing llm.request.type attribute (#269)

Author: galkleinman

packages/traceloop-sdk/src/lib/tracing/manual.ts

@@ -163,6 +163,7 @@ export function withLLMCall<
   F extends ({ span }: { span: LLMSpan }) => ReturnType<F>,
 >({ vendor, type }: LLMCallConfig, fn: F, thisArg?: ThisParameterType<F>) {
   const span = getTracer().startSpan(`${vendor}.${type}`, {}, context.active());
+  span.setAttribute(SpanAttributes.LLM_REQUEST_TYPE, type);
   trace.setSpan(context.active(), span);
 
   const res = fn.apply(thisArg, [{ span: new LLMSpan(span) }]);

packages/traceloop-sdk/test/decorators.test.ts

@@ -415,6 +415,10 @@ describe("Test SDK Decorators", () => {
       workflowSpan.spanContext().spanId,
     );
 
+    assert.strictEqual(
+      completionSpan.attributes[`${SpanAttributes.LLM_REQUEST_TYPE}`],
+      "chat",
+    );
     assert.strictEqual(
       completionSpan.attributes[`${SpanAttributes.LLM_REQUEST_MODEL}`],
       "gpt-3.5-turbo",