Merge pull request #456 from Scale3-Labs/obinna/S3EN-2894-add-arch

add arch as a service provider

Author: obinnascale3

src/langtrace_python_sdk/constants/instrumentation/common.py

@@ -10,6 +10,7 @@
 
 SERVICE_PROVIDERS = {
     "ANTHROPIC": "Anthropic",
+    "ARCH": "Arch",
     "AZURE": "Azure",
     "CHROMA": "Chroma",
     "CREWAI": "CrewAI",

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"),

src/langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "3.3.24"
+__version__ = "3.3.25"