OpenAi: completion: capture array prompt (#44)

traceloop · Jan 29, 2024 · 792bbc2 · 792bbc2
1 parent a67e5d3
commit 792bbc2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/packages/instrumentation-openai/src/instrumentation.ts b/packages/instrumentation-openai/src/instrumentation.ts
@@ -255,9 +255,12 @@ export class OpenAIInstrumentation extends InstrumentationBase<any> {
           }
         });
       } else {
+        attributes[`${SpanAttributes.LLM_PROMPTS}.0.role`] = "user";
         if (typeof params.prompt === "string") {
-          attributes[`${SpanAttributes.LLM_PROMPTS}.0.role`] = "user";
           attributes[`${SpanAttributes.LLM_PROMPTS}.0.content`] = params.prompt;
+        } else {
+          attributes[`${SpanAttributes.LLM_PROMPTS}.0.content`] =
+            JSON.stringify(params.prompt);
         }
       }
     }

diff --git a/packages/sample-app/src/sample_with.ts b/packages/sample-app/src/sample_with.ts
@@ -24,7 +24,7 @@ async function chat() {
 async function completion() {
   return await traceloop.withWorkflow("sample_completion", {}, async () => {
     const completion = await openai.completions.create({
-      prompt: "Tell me a joke about TypeScript",
+      prompt: ["Tell me a joke about TypeScript"],
       model: "gpt-3.5-turbo-instruct",
     });