@@ -44,6 +44,8 @@ export function imageEdit (
4444 serviceProvider = 'perplexity'
4545 } else if ( originalContext ?. _client ?. baseURL ?. includes ( 'x.ai' ) === true ) {
4646 serviceProvider = 'XAI'
47+ } else if ( originalContext ?. _client ?. baseURL ?. includes ( ':12000' ) === true || originalContext ?. _client ?. baseURL ?. includes ( ':10000' ) === true ) {
48+ serviceProvider = 'arch'
4749 }
4850 const attributes : LLMSpanAttributes = {
4951 'langtrace.sdk.name' : '@langtrase/typescript-sdk' ,
@@ -106,6 +108,8 @@ export function imagesGenerate (
106108 serviceProvider = 'perplexity'
107109 } else if ( originalContext ?. _client ?. baseURL ?. includes ( 'x.ai' ) === true ) {
108110 serviceProvider = 'XAI'
111+ } else if ( originalContext ?. _client ?. baseURL ?. includes ( ':12000' ) === true || originalContext ?. _client ?. baseURL ?. includes ( ':10000' ) === true ) {
112+ serviceProvider = 'arch'
109113 }
110114 const customAttributes = context . active ( ) . getValue ( LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY ) ?? { }
111115
@@ -166,6 +170,8 @@ export function chatCompletionCreate (
166170 serviceProvider = 'perplexity'
167171 } else if ( originalContext ?. _client ?. baseURL ?. includes ( 'x.ai' ) === true ) {
168172 serviceProvider = 'XAI'
173+ } else if ( originalContext ?. _client ?. baseURL ?. includes ( ':12000' ) === true || originalContext ?. _client ?. baseURL ?. includes ( ':10000' ) === true ) {
174+ serviceProvider = 'arch'
169175 }
170176 const attributes : LLMSpanAttributes = {
171177 'langtrace.sdk.name' : '@langtrase/typescript-sdk' ,
@@ -326,6 +332,8 @@ export function embeddingsCreate (
326332 serviceProvider = 'perplexity'
327333 } else if ( originalContext ?. _client ?. baseURL ?. includes ( 'x.ai' ) === true ) {
328334 serviceProvider = 'XAI'
335+ } else if ( originalContext ?. _client ?. baseURL ?. includes ( ':12000' ) === true || originalContext ?. _client ?. baseURL ?. includes ( ':10000' ) === true ) {
336+ serviceProvider = 'arch'
329337 }
330338 const attributes : LLMSpanAttributes = {
331339 'langtrace.sdk.name' : '@langtrase/typescript-sdk' ,
0 commit comments