From b5a28346e11602eb36d2a5b014203c29a624032f Mon Sep 17 00:00:00 2001
From: Ali Waleed <134522290+alizenhom@users.noreply.github.com>
Date: Thu, 15 Aug 2024 19:37:59 +0300
Subject: [PATCH] Pass Service name as enviroment variable

---
 src/langtrace_python_sdk/langtrace.py | 13 +++++++++----
 src/langtrace_python_sdk/version.py   |  2 +-
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/langtrace_python_sdk/langtrace.py b/src/langtrace_python_sdk/langtrace.py
index a8bb4b7f..9be99de1 100644
--- a/src/langtrace_python_sdk/langtrace.py
+++ b/src/langtrace_python_sdk/langtrace.py
@@ -33,7 +33,8 @@
     SimpleSpanProcessor,
 )
 import sys
-from opentelemetry.sdk.resources import Resource
+from opentelemetry.sdk.resources import Resource, SERVICE_NAME
+
 
 from langtrace_python_sdk.extensions.langtrace_exporter import LangTraceExporter
 from langtrace_python_sdk.instrumentation import (
@@ -79,10 +80,14 @@ def init(
     check_if_sdk_is_outdated()
     print(Fore.GREEN + "Initializing Langtrace SDK.." + Fore.RESET)
     sampler = LangtraceSampler(disabled_methods=disable_tracing_for_functions)
-    provider = TracerProvider(
-        resource=Resource.create({"service.name": service_name or sys.argv[0]}),
-        sampler=sampler,
+    resource = Resource.create(
+        attributes={
+            SERVICE_NAME: os.environ.get("OTEL_SERVICE_NAME")
+            or service_name
+            or sys.argv[0]
+        }
     )
+    provider = TracerProvider(resource=resource, sampler=sampler)
 
     remote_write_exporter = (
         LangTraceExporter(api_key=api_key, api_host=host)
diff --git a/src/langtrace_python_sdk/version.py b/src/langtrace_python_sdk/version.py
index 47f9cd67..97eeff4a 100644
--- a/src/langtrace_python_sdk/version.py
+++ b/src/langtrace_python_sdk/version.py
@@ -1 +1 @@
-__version__ = "2.2.19"
+__version__ = "2.2.20"