From aba5960d638ad9685801c7e9f95aa7c7833c4e9d Mon Sep 17 00:00:00 2001 From: Obinna Okafor Date: Tue, 14 Jan 2025 20:24:55 +0100 Subject: [PATCH 1/2] add arch as a service provider --- .../constants/instrumentation/common.py | 1 + src/langtrace_python_sdk/instrumentation/openai/patch.py | 8 ++++++++ 2 files changed, 9 insertions(+) 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..87b0c92a 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 get_base_url(instance) or ":10000" in get_base_url(instance): + 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 get_base_url(instance) or ":10000" in get_base_url(instance): + service_provider = SERVICE_PROVIDERS["ARCH"] span_attributes = { **get_langtrace_attributes(version, service_provider, vendor_type="llm"), From 8a35f73a07d3f7103926257ee1e949a84475694c Mon Sep 17 00:00:00 2001 From: Obinna Okafor Date: Wed, 15 Jan 2025 09:50:26 +0100 Subject: [PATCH 2/2] bump version --- src/langtrace_python_sdk/instrumentation/openai/patch.py | 4 ++-- src/langtrace_python_sdk/version.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/langtrace_python_sdk/instrumentation/openai/patch.py b/src/langtrace_python_sdk/instrumentation/openai/patch.py index 87b0c92a..6cab2057 100644 --- a/src/langtrace_python_sdk/instrumentation/openai/patch.py +++ b/src/langtrace_python_sdk/instrumentation/openai/patch.py @@ -451,7 +451,7 @@ def traced_method( service_provider = SERVICE_PROVIDERS["XAI"] elif "deepseek" in base_url: service_provider = SERVICE_PROVIDERS["DEEPSEEK"] - elif ":12000" in get_base_url(instance) or ":10000" in get_base_url(instance): + elif ":12000" in base_url or ":10000" in base_url: service_provider = SERVICE_PROVIDERS["ARCH"] span_attributes = { @@ -539,7 +539,7 @@ async def traced_method( service_provider = SERVICE_PROVIDERS["XAI"] elif "deepseek" in base_url: service_provider = SERVICE_PROVIDERS["DEEPSEEK"] - elif ":12000" in get_base_url(instance) or ":10000" in get_base_url(instance): + elif ":12000" in base_url or ":10000" in base_url: service_provider = SERVICE_PROVIDERS["ARCH"] span_attributes = { 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"