diff --git a/src/langtrace_python_sdk/constants/instrumentation/common.py b/src/langtrace_python_sdk/constants/instrumentation/common.py index dfede210..19195655 100644 --- a/src/langtrace_python_sdk/constants/instrumentation/common.py +++ b/src/langtrace_python_sdk/constants/instrumentation/common.py @@ -10,6 +10,7 @@ SERVICE_PROVIDERS = { "ANTHROPIC": "Anthropic", + "ARCH": "Arch", "AZURE": "Azure", "CHROMA": "Chroma", "CREWAI": "CrewAI", diff --git a/src/langtrace_python_sdk/instrumentation/openai/patch.py b/src/langtrace_python_sdk/instrumentation/openai/patch.py index 255a7eb7..6cab2057 100644 --- a/src/langtrace_python_sdk/instrumentation/openai/patch.py +++ b/src/langtrace_python_sdk/instrumentation/openai/patch.py @@ -254,6 +254,8 @@ def traced_method( service_provider = SERVICE_PROVIDERS["XAI"] elif "deepseek" in get_base_url(instance): service_provider = SERVICE_PROVIDERS["DEEPSEEK"] + elif ":12000" in get_base_url(instance) or ":10000" in get_base_url(instance): + service_provider = SERVICE_PROVIDERS["ARCH"] llm_prompts = [] for item in kwargs.get("messages", []): tools = get_tool_calls(item) @@ -348,6 +350,8 @@ async def traced_method( service_provider = SERVICE_PROVIDERS["XAI"] elif "deepseek" in get_base_url(instance): service_provider = SERVICE_PROVIDERS["DEEPSEEK"] + elif ":12000" in get_base_url(instance) or ":10000" in get_base_url(instance): + service_provider = SERVICE_PROVIDERS["ARCH"] llm_prompts = [] for item in kwargs.get("messages", []): tools = get_tool_calls(item) @@ -447,6 +451,8 @@ def traced_method( service_provider = SERVICE_PROVIDERS["XAI"] elif "deepseek" in base_url: service_provider = SERVICE_PROVIDERS["DEEPSEEK"] + elif ":12000" in base_url or ":10000" in base_url: + service_provider = SERVICE_PROVIDERS["ARCH"] span_attributes = { **get_langtrace_attributes(version, service_provider, vendor_type="llm"), @@ -533,6 +539,8 @@ async def traced_method( service_provider = SERVICE_PROVIDERS["XAI"] elif "deepseek" in base_url: service_provider = SERVICE_PROVIDERS["DEEPSEEK"] + elif ":12000" in base_url or ":10000" in base_url: + service_provider = SERVICE_PROVIDERS["ARCH"] span_attributes = { **get_langtrace_attributes(version, service_provider, vendor_type="llm"), diff --git a/src/langtrace_python_sdk/version.py b/src/langtrace_python_sdk/version.py index 943ca0bc..92a2c9aa 100644 --- a/src/langtrace_python_sdk/version.py +++ b/src/langtrace_python_sdk/version.py @@ -1 +1 @@ -__version__ = "3.3.24" +__version__ = "3.3.25"