Skip to content

Commit

Permalink
feat: Drop metadata & Add tags to generations
Browse files Browse the repository at this point in the history
  • Loading branch information
Granipouss committed Apr 26, 2024
1 parent d9902ce commit e0d0300
Showing 1 changed file with 6 additions and 14 deletions.
20 changes: 6 additions & 14 deletions src/instrumentation/openai.ts
Original file line number Diff line number Diff line change
Expand Up @@ -286,7 +286,6 @@ export type OpenAIOutput =
| Stream<ChatCompletionChunk>;

export interface InstrumentOpenAIOptions {
metadata?: Maybe<Record<string, any>>;
tags?: Maybe<string[]>;
}

Expand All @@ -302,7 +301,8 @@ const instrumentOpenAI = async (
const baseGeneration = {
provider: 'openai',
model: inputs.model,
settings: getSettings(inputs)
settings: getSettings(inputs),
tags: options.tags
};

if (output instanceof Stream) {
Expand Down Expand Up @@ -340,9 +340,7 @@ const instrumentOpenAI = async (
generation,
output: messageCompletion,
startTime: new Date(start).toISOString(),
endTime: new Date(start + metrics.duration).toISOString(),
tags: options.tags,
metadata: options.metadata
endTime: new Date(start + metrics.duration).toISOString()
});
await step.send();
} else {
Expand All @@ -368,9 +366,7 @@ const instrumentOpenAI = async (
generation,
output: { content: completion },
startTime: new Date(start).toISOString(),
endTime: new Date(start + metrics.duration).toISOString(),
tags: options.tags,
metadata: options.metadata
endTime: new Date(start + metrics.duration).toISOString()
});
await step.send();
} else {
Expand Down Expand Up @@ -408,9 +404,7 @@ const instrumentOpenAI = async (
generation,
output: messageCompletion,
startTime: new Date(start).toISOString(),
endTime: new Date().toISOString(),
tags: options.tags,
metadata: options.metadata
endTime: new Date().toISOString()
});
await step.send();
} else {
Expand All @@ -434,9 +428,7 @@ const instrumentOpenAI = async (
generation,
output: { content: completion },
startTime: new Date(start).toISOString(),
endTime: new Date().toISOString(),
tags: options.tags,
metadata: options.metadata
endTime: new Date().toISOString()
});
await step.send();
} else {
Expand Down

0 comments on commit e0d0300

Please sign in to comment.