Skip to content

Commit 78fa47f

Browse files
karthikscale3darshit-s3obinnaokaforobinnascale3rohit-kadhe
authored
Release (#187)
* feat: adding gemini support (#162) * feat: adding gemini support * chore: update package lock * docs: update changelog for 5.2.0 release * feat: adding vertexai instrumentation support (#165) * feat: adding vertexai instrumentation support * style: remove commented import * chore: create types for vertexai * docs: revert vercel ai py check * docs: add mistral check * style: remove new lines * docs: add vertexai to changelog (#166) * fix: allow manual patch for Gemini and VertexAI (#169) * star github link * feat: adding vertexai tools tracing support (#172) * add mistral instrumentation * support otel env vars * bump version adn add chat stream method * add examples * update readme * update package-lock * Add sentry integration * merge with dev * llama index enhacements * sentry to filter non langtrace exceptions * Fix ingestion url (#181) * Obinna/s3 en 2724 add bedrock (#184) * add aws bedrock converse implementation * bump version * cleanup and update readme * update service type * fixes --------- Co-authored-by: Obinna Okafor <[email protected]> Co-authored-by: Karthik Kalyanaraman <[email protected]> * fix (#186) --------- Co-authored-by: darshit-s3 <[email protected]> Co-authored-by: Darshit Suratwala <[email protected]> Co-authored-by: Obinna Okafor <[email protected]> Co-authored-by: obinnascale3 <[email protected]> Co-authored-by: Rohit Kadhe <[email protected]> Co-authored-by: Rohit Kadhe <[email protected]>
1 parent 6d92358 commit 78fa47f

File tree

2 files changed

+18
-7
lines changed

2 files changed

+18
-7
lines changed

package.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
{
22
"name": "@langtrase/typescript-sdk",
3-
"version": "6.1.0",
3+
"version": "6.1.1",
44
"description": "A typescript SDK for Langtrace",
55
"main": "dist/index.js",
66
"types": "dist/index.d.ts",

src/instrumentation/openai/patch.ts

Lines changed: 17 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -17,15 +17,12 @@
1717
import { LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY } from '@langtrace-constants/common'
1818
import { calculatePromptTokens, estimateTokens } from '@langtrace-utils/llm'
1919
import { addSpanEvent, createStreamProxy } from '@langtrace-utils/misc'
20-
import { APIS, LLMSpanAttributes, Event, Vendors } from '@langtrase/trace-attributes'
20+
import { APIS, Event, LLMSpanAttributes, Vendors } from '@langtrase/trace-attributes'
2121
import {
22-
Exception,
22+
context, Exception,
2323
Span,
2424
SpanKind,
25-
SpanStatusCode,
26-
Tracer,
27-
context,
28-
trace
25+
SpanStatusCode, trace, Tracer
2926
} from '@opentelemetry/api'
3027
import { LangtraceSdkError } from 'errors/sdk_error'
3128

@@ -43,6 +40,10 @@ export function imageEdit (
4340
let serviceProvider: string = Vendors.OPENAI
4441
if (originalContext?._client?.baseURL?.includes('azure') === true) {
4542
serviceProvider = 'azure'
43+
} else if (originalContext?._client?.baseURL?.includes('perplexity') === true) {
44+
serviceProvider = 'perplexity'
45+
} else if (originalContext?._client?.baseURL?.includes('x.ai') === true) {
46+
serviceProvider = 'XAI'
4647
}
4748
const attributes: LLMSpanAttributes = {
4849
'langtrace.sdk.name': '@langtrase/typescript-sdk',
@@ -101,6 +102,10 @@ export function imagesGenerate (
101102
let serviceProvider: string = Vendors.OPENAI
102103
if (originalContext?._client?.baseURL?.includes('azure') === true) {
103104
serviceProvider = 'azure'
105+
} else if (originalContext?._client?.baseURL?.includes('perplexity') === true) {
106+
serviceProvider = 'perplexity'
107+
} else if (originalContext?._client?.baseURL?.includes('x.ai') === true) {
108+
serviceProvider = 'XAI'
104109
}
105110
const customAttributes = context.active().getValue(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY) ?? {}
106111

@@ -159,6 +164,8 @@ export function chatCompletionCreate (
159164
serviceProvider = 'azure'
160165
} else if (originalContext?._client?.baseURL?.includes('perplexity') === true) {
161166
serviceProvider = 'perplexity'
167+
} else if (originalContext?._client?.baseURL?.includes('x.ai') === true) {
168+
serviceProvider = 'XAI'
162169
}
163170
const attributes: LLMSpanAttributes = {
164171
'langtrace.sdk.name': '@langtrase/typescript-sdk',
@@ -315,6 +322,10 @@ export function embeddingsCreate (
315322
let serviceProvider: string = Vendors.OPENAI
316323
if (originalContext?._client?.baseURL?.includes('azure') === true) {
317324
serviceProvider = 'azure'
325+
} else if (originalContext?._client?.baseURL?.includes('perplexity') === true) {
326+
serviceProvider = 'perplexity'
327+
} else if (originalContext?._client?.baseURL?.includes('x.ai') === true) {
328+
serviceProvider = 'XAI'
318329
}
319330
const attributes: LLMSpanAttributes = {
320331
'langtrace.sdk.name': '@langtrase/typescript-sdk',

0 commit comments

Comments
 (0)