Skip to content

Commit aba5960

Browse files
committed
add arch as a service provider
1 parent 4445df2 commit aba5960

File tree

2 files changed

+9
-0
lines changed
  • src/langtrace_python_sdk

2 files changed

+9
-0
lines changed

src/langtrace_python_sdk/constants/instrumentation/common.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010

1111
SERVICE_PROVIDERS = {
1212
"ANTHROPIC": "Anthropic",
13+
"ARCH": "Arch",
1314
"AZURE": "Azure",
1415
"CHROMA": "Chroma",
1516
"CREWAI": "CrewAI",

src/langtrace_python_sdk/instrumentation/openai/patch.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -254,6 +254,8 @@ def traced_method(
254254
service_provider = SERVICE_PROVIDERS["XAI"]
255255
elif "deepseek" in get_base_url(instance):
256256
service_provider = SERVICE_PROVIDERS["DEEPSEEK"]
257+
elif ":12000" in get_base_url(instance) or ":10000" in get_base_url(instance):
258+
service_provider = SERVICE_PROVIDERS["ARCH"]
257259
llm_prompts = []
258260
for item in kwargs.get("messages", []):
259261
tools = get_tool_calls(item)
@@ -348,6 +350,8 @@ async def traced_method(
348350
service_provider = SERVICE_PROVIDERS["XAI"]
349351
elif "deepseek" in get_base_url(instance):
350352
service_provider = SERVICE_PROVIDERS["DEEPSEEK"]
353+
elif ":12000" in get_base_url(instance) or ":10000" in get_base_url(instance):
354+
service_provider = SERVICE_PROVIDERS["ARCH"]
351355
llm_prompts = []
352356
for item in kwargs.get("messages", []):
353357
tools = get_tool_calls(item)
@@ -447,6 +451,8 @@ def traced_method(
447451
service_provider = SERVICE_PROVIDERS["XAI"]
448452
elif "deepseek" in base_url:
449453
service_provider = SERVICE_PROVIDERS["DEEPSEEK"]
454+
elif ":12000" in get_base_url(instance) or ":10000" in get_base_url(instance):
455+
service_provider = SERVICE_PROVIDERS["ARCH"]
450456

451457
span_attributes = {
452458
**get_langtrace_attributes(version, service_provider, vendor_type="llm"),
@@ -533,6 +539,8 @@ async def traced_method(
533539
service_provider = SERVICE_PROVIDERS["XAI"]
534540
elif "deepseek" in base_url:
535541
service_provider = SERVICE_PROVIDERS["DEEPSEEK"]
542+
elif ":12000" in get_base_url(instance) or ":10000" in get_base_url(instance):
543+
service_provider = SERVICE_PROVIDERS["ARCH"]
536544

537545
span_attributes = {
538546
**get_langtrace_attributes(version, service_provider, vendor_type="llm"),

0 commit comments

Comments
 (0)