langtrace-python-sdk 1.0.11__tar.gz → 1.0.12__tar.gz

{langtrace-python-sdk-1.0.11/src/langtrace_python_sdk.egg-info → langtrace-python-sdk-1.0.12}/PKG-INFO

@@ -1,10 +1,12 @@
 Metadata-Version: 2.1
 Name: langtrace-python-sdk
-Version: 1.0.11
-Summary: LangTrace - Python SDK
+Version: 1.0.12
+Summary: Python SDK for LangTrace
 Home-page: https://github.com/Scale3-Labs/langtrace-python-sdk
-Author: Ali Waleed
+Author: Scale3 Labs
 Author-email: ali@scale3labs.com
+Maintainer: ['Karthik Kalyanaraman', 'Ali Waleed', 'Rohit Kadhe']
+License: AGPL-3.0-or-later
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/setup.py

@@ -4,10 +4,12 @@ with open('requirements.txt') as f:
 
 setup(
     name='langtrace-python-sdk',  # Choose a unique name for PyPI
-    version='1.0.11',
-    author='Ali Waleed',
+    version='1.0.12',
+    author='Scale3 Labs',
+    license= "AGPL-3.0-or-later",
     author_email='ali@scale3labs.com',
-    description='LangTrace - Python SDK',
+    maintainer=['Karthik Kalyanaraman', 'Ali Waleed', 'Rohit Kadhe'],
+    description='Python SDK for LangTrace',
     long_description="LangTrace - Python SDK",
     long_description_content_type='text/markdown',
     url='https://github.com/Scale3-Labs/langtrace-python-sdk',  # Project home page

langtrace-python-sdk-1.0.11/src/instrumentation/constants.py → langtrace-python-sdk-1.0.12/src/constants/instrumentation/common.py

@@ -1,6 +1,11 @@
-"""
-This file contains the constants used in the project.
-"""
+TIKTOKEN_MODEL_MAPPING = {
+    "gpt-4": "cl100k_base",
+    "gpt-4-32k": "cl100k_base",
+    "gpt-4-0125-preview": "cl100k_base",
+    "gpt-4-1106-preview": "cl100k_base",
+    "gpt-4-1106-vision-preview": "cl100k_base",
+}
+
 SERVICE_PROVIDERS = {
     "OPENAI": "OpenAI",
     "AZURE": "Azure",
@@ -10,4 +15,4 @@ SERVICE_PROVIDERS = {
     "PINECONE": "Pinecone",
     "LLAMAINDEX": "LlamaIndex",
     "CHROMA": "Chroma",
-}
+}

langtrace-python-sdk-1.0.11/src/instrumentation/openai/constants.py → langtrace-python-sdk-1.0.12/src/constants/instrumentation/openai.py

@@ -1,5 +1,4 @@
-"""
-Constants for OpenAI API"""
+from langtrace.trace_attributes import OpenAIMethods
 
 OPENAI_COST_TABLE = {
     "gpt-4-0125-preview": {
@@ -32,12 +31,19 @@ OPENAI_COST_TABLE = {
     },
 }
 
-# TODO: Add more models
-# https://github.com/dqbd/tiktoken/blob/74c147e19584a3a1acea0c8e0da4d39415cd33e0/wasm/src/lib.rs#L328
-TIKTOKEN_MODEL_MAPPING = {
-    "gpt-4": "cl100k_base",
-    "gpt-4-32k": "cl100k_base",
-    "gpt-4-0125-preview": "cl100k_base",
-    "gpt-4-1106-preview": "cl100k_base",
-    "gpt-4-1106-vision-preview": "cl100k_base",
+APIS = {
+    "CHAT_COMPLETION": {
+        "METHOD": OpenAIMethods.CHAT_COMPLETION.value,
+        "ENDPOINT": "/chat/completions",
+    },
+    "IMAGES_GENERATION": {
+        "METHOD": OpenAIMethods.IMAGES_GENERATION.value,
+        "ENDPOINT": "/images/generations",
+    },
+    "EMBEDDINGS_CREATE": {
+        "METHOD": OpenAIMethods.EMBEDDINGS_CREATE.value,
+        "ENDPOINT": "/embeddings",
+    },
 }
+
+

langtrace-python-sdk-1.0.11/src/instrumentation/pinecone/apis.py → langtrace-python-sdk-1.0.12/src/constants/instrumentation/pinecone.py

@@ -12,7 +12,7 @@ APIS = {
         "OPERATION": "query",
     },
     "DELETE": {
-        "METHOD": PineconeMethods.DELETE.value,
+        "METHOD": PineconeMethods.DELETE,
         "ENDPOINT": "/vectors/delete",
         "OPERATION": "delete",
     }

langtrace-python-sdk-1.0.12/src/examples/chroma_example/__init__.py

@@ -0,0 +1 @@
+from .basic import basic

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/examples/chroma_example/basic.py

@@ -2,12 +2,11 @@ import chromadb
 from chromadb.utils import embedding_functions
 from dotenv import find_dotenv, load_dotenv
 
-from examples.setup import setup_instrumentation
-from instrumentation.with_root_span import with_langtrace_root_span
-
+from utils.with_root_span import with_langtrace_root_span
+from init import init
 _ = load_dotenv(find_dotenv())
 
-setup_instrumentation()
+init()
 
 
 @with_langtrace_root_span()

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/examples/langchain_example/basic.py

@@ -1,4 +1,5 @@
 from dotenv import find_dotenv, load_dotenv
+from init.init import init
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_community.document_loaders import PyPDFLoader
 from langchain_community.vectorstores.faiss import FAISS
@@ -6,14 +7,12 @@ from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts.chat import ChatPromptTemplate
 from langchain_core.runnables import RunnablePassthrough
 from langchain_openai import ChatOpenAI, OpenAIEmbeddings
+from utils.with_root_span import with_langtrace_root_span
 
-from examples.setup import setup_instrumentation
-from instrumentation.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-setup_instrumentation()
-
+init()
 
 @with_langtrace_root_span()
 def basic():

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/examples/langchain_example/tool.py

@@ -1,17 +1,17 @@
 from dotenv import find_dotenv, load_dotenv
+from init import init
 from langchain import hub
 from langchain.agents import AgentExecutor, create_openai_functions_agent
 from langchain.chains import LLMMathChain
 from langchain_core.pydantic_v1 import BaseModel, Field
 from langchain_core.tools import Tool
 from langchain_openai import ChatOpenAI
+from utils.with_root_span import with_langtrace_root_span
 
-from examples.setup import setup_instrumentation
-from instrumentation.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-setup_instrumentation()
+init()
 
 
 llm = ChatOpenAI(temperature=0, model="gpt-4")

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/examples/llamaindex_example/basic.py

@@ -1,12 +1,13 @@
 from dotenv import find_dotenv, load_dotenv
+from init import init
 from llama_index.core import SimpleDirectoryReader, VectorStoreIndex
+from utils.with_root_span import with_langtrace_root_span
 
-from examples.setup import setup_instrumentation
-from instrumentation.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-setup_instrumentation()
+
+init()
 
 
 @with_langtrace_root_span()

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/examples/openai/chat_completion.py

@@ -1,12 +1,13 @@
 from dotenv import find_dotenv, load_dotenv
 from openai import OpenAI
 
-from examples.setup import setup_instrumentation
-from instrumentation.with_root_span import with_langtrace_root_span
+from init.init import init
+from utils.with_root_span import with_langtrace_root_span
+
 
 _ = load_dotenv(find_dotenv())
 
-setup_instrumentation()
+init()
 
 client = OpenAI()
 

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/examples/openai/embeddings_create.py

@@ -1,13 +1,13 @@
 from dotenv import find_dotenv, load_dotenv
+from init import init
 from openai import OpenAI
 
-from examples.setup import setup_instrumentation
-from instrumentation.with_root_span import with_langtrace_root_span
+from utils.with_root_span import with_langtrace_root_span
 
-_ = load_dotenv(find_dotenv())
 
-setup_instrumentation()
+_ = load_dotenv(find_dotenv())
 
+init()
 client = OpenAI()
 
 

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/examples/openai/function_calling.py

@@ -1,14 +1,15 @@
 import json
 
 from dotenv import find_dotenv, load_dotenv
+from init import init
 from openai import OpenAI
 
-from examples.setup import setup_instrumentation
-from instrumentation.with_root_span import with_langtrace_root_span
+from utils.with_root_span import with_langtrace_root_span
+
 
 _ = load_dotenv(find_dotenv())
 
-setup_instrumentation()
+init()
 
 client = OpenAI()
 

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/examples/openai/images_generate.py

@@ -1,12 +1,12 @@
 from dotenv import find_dotenv, load_dotenv
+from init import init
 from openai import OpenAI
+from utils.with_root_span import with_langtrace_root_span
 
-from examples.setup import setup_instrumentation
-from instrumentation.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-setup_instrumentation()
+init()
 
 client = OpenAI()
 

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/examples/pinecone_example/basic.py

@@ -1,13 +1,13 @@
 from dotenv import find_dotenv, load_dotenv
+from init import init
 from openai import OpenAI
 from pinecone import Pinecone
+from utils.with_root_span import with_langtrace_root_span
 
-from examples.setup import setup_instrumentation
-from instrumentation.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-setup_instrumentation()
+init()
 
 client = OpenAI()
 pinecone = Pinecone()

langtrace-python-sdk-1.0.12/src/extensions/langtrace_exporter.py

@@ -0,0 +1,75 @@
+import json
+from opentelemetry.sdk.trace.export import (SpanExporter, ReadableSpan, SpanExportResult)
+import os
+import typing
+import requests
+
+
+
+
+class LangTraceExporter(SpanExporter):
+    api_key: str
+    url: str
+    write_to_remote_url: bool
+
+    def __init__(self, api_key: str = None, url: str = None, write_to_remote_url: bool = False) -> None:
+        self.api_key = api_key if api_key else os.environ.get('LANGTRACE_API_KEY')
+        self.url = url if url else os.environ.get('LANGTRACE_URL')
+        self.write_to_remote_url = write_to_remote_url
+
+        if not self.api_key:
+            raise ValueError('No API key provided')
+
+        if not self.url and self.write_to_remote_url:
+            raise ValueError('No URL provided')
+
+    def export(self, spans: typing.Sequence[ReadableSpan]) -> SpanExportResult:
+
+      """
+      Exports a batch of telemetry data.
+
+      Args:
+          spans: The list of `opentelemetry.trace.Span` objects to be exported
+
+      Returns:
+          The result of the export SUCCESS or FAILURE
+      """
+      
+
+      data = [
+        {
+          'traceId': span.get_span_context().trace_id,
+          'instrumentationLibrary': span.instrumentation_info.__repr__(),
+          'droppedEventsCount': span.dropped_events,
+          'droppedAttributesCount': span.dropped_attributes,
+          'droppedLinksCount': span.dropped_links,
+          'ended': span.status.is_ok,
+          'duration': span.end_time - span.start_time,
+           **json.loads(span.to_json()),
+        } for span in spans]
+
+
+      if not self.write_to_remote_url:
+        return
+
+      # Send data to remote URL
+      try:
+        requests.post(
+          url=self.url,
+          data=json.dumps(data),
+          headers={
+            'Content-Type': 'application/json',
+            'x-api-key': self.api_key
+          }
+        )
+        print(f"Sent To {self.url} with Body {data} ") 
+        return SpanExportResult.SUCCESS
+      except Exception as e:
+        print("Error sending data to remote URL", e)
+        return SpanExportResult.FAILURE
+
+
+
+      
+    def shutdown(self) -> None:
+       pass

langtrace-python-sdk-1.0.12/src/init/__init__.py

@@ -0,0 +1 @@
+from init.init import init

langtrace-python-sdk-1.0.12/src/init/init.py

@@ -0,0 +1,72 @@
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import ConsoleSpanExporter, SimpleSpanProcessor, BatchSpanProcessor
+from extensions.langtrace_exporter import LangTraceExporter
+
+from instrumentation.chroma.instrumentation import ChromaInstrumentation
+from instrumentation.langchain.instrumentation import LangchainInstrumentation
+from instrumentation.langchain_community.instrumentation import \
+    LangchainCommunityInstrumentation
+from instrumentation.langchain_core.instrumentation import \
+    LangchainCoreInstrumentation
+from instrumentation.llamaindex.instrumentation import \
+    LlamaindexInstrumentation
+from instrumentation.openai.instrumentation import OpenAIInstrumentation
+from instrumentation.pinecone.instrumentation import PineconeInstrumentation
+from opentelemetry import trace
+
+def init(
+    api_key: str = None,
+    remote_url: str = None,
+    batch: bool = False,
+    log_spans_to_console: bool = False,
+    write_to_remote_url: bool = True
+  ):
+  
+  provider = TracerProvider()
+  remote_write_exporter = LangTraceExporter(api_key, remote_url, write_to_remote_url)
+  console_exporter = ConsoleSpanExporter()
+  batch_processor_remote = BatchSpanProcessor(remote_write_exporter)
+  simple_processor_remote = SimpleSpanProcessor(remote_write_exporter)
+  batch_processor_console = BatchSpanProcessor(console_exporter)
+  simple_processor_console = SimpleSpanProcessor(console_exporter)
+
+  if log_spans_to_console:
+    if batch:
+      provider.add_span_processor(batch_processor_console)
+    else:
+      provider.add_span_processor(simple_processor_console)
+  
+  if write_to_remote_url:
+    if batch:
+      provider.add_span_processor(batch_processor_remote)
+    else:
+      provider.add_span_processor(simple_processor_remote)
+
+
+ # Initialize tracer
+  trace.set_tracer_provider(provider)
+
+
+  openai_instrumentation = OpenAIInstrumentation()
+  pinecone_instrumentation = PineconeInstrumentation()
+  llamaindex_instrumentation = LlamaindexInstrumentation()
+  chroma_instrumentation = ChromaInstrumentation()
+  langchain_instrumentation = LangchainInstrumentation()
+  langchain_core_instrumentation = LangchainCoreInstrumentation()
+  langchain_community_instrumentation = LangchainCommunityInstrumentation()
+
+
+  # Call the instrument method with some arguments
+  openai_instrumentation.instrument()
+  pinecone_instrumentation.instrument()
+  llamaindex_instrumentation.instrument()
+  chroma_instrumentation.instrument()
+  langchain_instrumentation.instrument()
+  langchain_core_instrumentation.instrument()
+  langchain_community_instrumentation.instrument()
+
+
+
+
+
+

langtrace-python-sdk-1.0.12/src/instrumentation/chroma/apis.py

@@ -0,0 +1,40 @@
+from langtrace.trace_attributes import ChromaDBMethods
+
+APIS = {
+    "ADD": {
+        "METHOD": ChromaDBMethods.ADD.value,
+        "OPERATION": "add",
+    },
+    "GET": {
+        "METHOD": ChromaDBMethods.GET.value,
+        "OPERATION": "get",
+    },
+    "QUERY": {
+        "METHOD": ChromaDBMethods.QUERY.value,
+        "OPERATION": "query",
+    },
+    "DELETE": {
+        "METHOD": ChromaDBMethods.DELETE.value,
+        "OPERATION": "delete",
+    },
+    "PEEK": {
+        "METHOD": ChromaDBMethods.PEEK.value,
+        "OPERATION": "peek",
+    },
+    "UPDATE": {
+        "METHOD": ChromaDBMethods.UPDATE.value,
+        "OPERATION": "update",
+    },
+    "UPSERT": {
+        "METHOD": ChromaDBMethods.UPSERT.value,
+        "OPERATION": "upsert",
+    },
+    "MODIFY": {
+        "METHOD": ChromaDBMethods.MODIFY.value,
+        "OPERATION": "modify",
+    },
+    "COUNT": {
+        "METHOD": ChromaDBMethods.COUNT.value,
+        "OPERATION": "count",
+    },
+}

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/instrumentation/chroma/instrumentation.py

@@ -8,7 +8,7 @@ from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.trace import get_tracer
 from wrapt import wrap_function_wrapper
 
-from instrumentation.chroma.apis import APIS
+from constants.instrumentation.chroma import APIS
 from instrumentation.chroma.patch import collection_patch
 
 

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/instrumentation/chroma/patch.py

@@ -5,8 +5,8 @@ from langtrace.trace_attributes import DatabaseSpanAttributes
 from opentelemetry.trace import SpanKind
 from opentelemetry.trace.status import Status, StatusCode
 
-from instrumentation.chroma.apis import APIS
-from instrumentation.constants import SERVICE_PROVIDERS
+from constants.instrumentation.chroma import APIS
+from constants.instrumentation.common import SERVICE_PROVIDERS
 
 
 def collection_patch(method, version, tracer):

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/instrumentation/langchain/patch.py

@@ -7,7 +7,7 @@ from langtrace.trace_attributes import FrameworkSpanAttributes
 from opentelemetry.trace import SpanKind, StatusCode
 from opentelemetry.trace.status import Status
 
-from instrumentation.constants import SERVICE_PROVIDERS
+from constants.instrumentation.common import SERVICE_PROVIDERS
 
 
 def generic_patch(method_name, task, tracer, version, trace_output=True, trace_input=True):

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/instrumentation/langchain_community/patch.py

@@ -4,7 +4,8 @@ from langtrace.trace_attributes import FrameworkSpanAttributes
 from opentelemetry.trace import SpanKind, StatusCode
 from opentelemetry.trace.status import Status, StatusCode
 
-from instrumentation.constants import SERVICE_PROVIDERS
+from constants.instrumentation.common import SERVICE_PROVIDERS
+
 
 
 def generic_patch(method_name, task, tracer, version, trace_output=True, trace_input=True):

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/instrumentation/langchain_core/patch.py

@@ -7,7 +7,8 @@ from langtrace.trace_attributes import FrameworkSpanAttributes
 from opentelemetry.trace import SpanKind, StatusCode
 from opentelemetry.trace.status import Status
 
-from instrumentation.constants import SERVICE_PROVIDERS
+from constants.instrumentation.common import SERVICE_PROVIDERS
+
 
 
 def generic_patch(method_name, task, tracer, version, trace_output=True, trace_input=True):

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/instrumentation/llamaindex/patch.py

@@ -5,7 +5,8 @@ from langtrace.trace_attributes import FrameworkSpanAttributes
 from opentelemetry.trace import SpanKind
 from opentelemetry.trace.status import Status, StatusCode
 
-from instrumentation.constants import SERVICE_PROVIDERS
+from constants.instrumentation.common import SERVICE_PROVIDERS
+
 
 
 def generic_patch(method, task, tracer, version):

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/instrumentation/openai/patch.py

@@ -6,8 +6,8 @@ from langtrace.trace_attributes import Event, LLMSpanAttributes
 from opentelemetry.trace import SpanKind
 from opentelemetry.trace.status import Status, StatusCode
 
-from instrumentation.constants import SERVICE_PROVIDERS
-from instrumentation.openai.apis import APIS
+from constants.instrumentation.common import SERVICE_PROVIDERS
+from constants.instrumentation.openai import APIS
 from instrumentation.openai.token_estimation import (calculate_prompt_tokens,
                                                      estimate_tokens)
 
@@ -94,9 +94,9 @@ def chat_completions_create(original_method, version, tracer):
             attributes.llm_top_p = kwargs.get('top_p')
         if kwargs.get('user') is not None:
             attributes.llm_user = kwargs.get('user')
-        if kwargs.get('functions') is not None:
-            attributes.llm_function_prompts = json.dumps(
-                kwargs.get('functions'))
+        # if kwargs.get('functions') is not None:
+        #     attributes.llm_function_prompts = json.dumps(
+        #         kwargs.get('functions'))
 
         # TODO(Karthik): Gotta figure out how to handle streaming with context
         # with tracer.start_as_current_span(APIS["CHAT_COMPLETION"]["METHOD"],

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/instrumentation/openai/token_estimation.py

@@ -4,9 +4,8 @@ to calculate the price of a model based on its usage.
 """
 
 from tiktoken import get_encoding
-
-from instrumentation.openai.constants import (OPENAI_COST_TABLE,
-                                              TIKTOKEN_MODEL_MAPPING)
+from constants.instrumentation.openai import OPENAI_COST_TABLE
+from constants.instrumentation.common import TIKTOKEN_MODEL_MAPPING
 
 
 def estimate_tokens(prompt):

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/instrumentation/pinecone/instrumentation.py

@@ -11,7 +11,7 @@ from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.trace import get_tracer
 from wrapt import wrap_function_wrapper
 
-from instrumentation.pinecone.apis import APIS
+from constants.instrumentation.pinecone import APIS
 from instrumentation.pinecone.patch import generic_patch
 
 

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/instrumentation/pinecone/patch.py

@@ -4,8 +4,8 @@ from langtrace.trace_attributes import DatabaseSpanAttributes
 from opentelemetry.trace import SpanKind
 from opentelemetry.trace.status import Status, StatusCode
 
-from instrumentation.constants import SERVICE_PROVIDERS
-from instrumentation.pinecone.apis import APIS
+from constants.instrumentation.common import SERVICE_PROVIDERS
+from constants.instrumentation.pinecone import APIS
 
 
 def generic_patch(original_method, method, version, tracer):

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12/src/langtrace_python_sdk.egg-info}/PKG-INFO

@@ -1,10 +1,12 @@
 Metadata-Version: 2.1
 Name: langtrace-python-sdk
-Version: 1.0.11
-Summary: LangTrace - Python SDK
+Version: 1.0.12
+Summary: Python SDK for LangTrace
 Home-page: https://github.com/Scale3-Labs/langtrace-python-sdk
-Author: Ali Waleed
+Author: Scale3 Labs
 Author-email: ali@scale3labs.com
+Maintainer: ['Karthik Kalyanaraman', 'Ali Waleed', 'Rohit Kadhe']
+License: AGPL-3.0-or-later
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent

{langtrace-python-sdk-1.0.11 → langtrace-python-sdk-1.0.12}/src/langtrace_python_sdk.egg-info/SOURCES.txt

@@ -1,8 +1,13 @@
 LICENSE
 README.md
 setup.py
+src/constants/__init__.py
+src/constants/instrumentation/__init__.py
+src/constants/instrumentation/chroma.py
+src/constants/instrumentation/common.py
+src/constants/instrumentation/openai.py
+src/constants/instrumentation/pinecone.py
 src/examples/__init__.py
-src/examples/setup.py
 src/examples/chroma_example/__init__.py
 src/examples/chroma_example/basic.py
 src/examples/langchain_example/__init__.py
@@ -17,14 +22,15 @@ src/examples/openai/function_calling.py
 src/examples/openai/images_generate.py
 src/examples/pinecone_example/__init__.py
 src/examples/pinecone_example/basic.py
+src/extensions/__init__.py
+src/extensions/langtrace_exporter.py
+src/init/__init__.py
+src/init/init.py
 src/instrumentation/__init__.py
-src/instrumentation/constants.py
-src/instrumentation/with_root_span.py
 src/instrumentation/chroma/__init__.py
 src/instrumentation/chroma/apis.py
 src/instrumentation/chroma/instrumentation.py
 src/instrumentation/chroma/patch.py
-src/instrumentation/chroma/lib/__init__.py
 src/instrumentation/langchain/__init__.py
 src/instrumentation/langchain/instrumentation.py
 src/instrumentation/langchain/patch.py
@@ -38,19 +44,16 @@ src/instrumentation/llamaindex/__init__.py
 src/instrumentation/llamaindex/instrumentation.py
 src/instrumentation/llamaindex/patch.py
 src/instrumentation/openai/__init__.py
-src/instrumentation/openai/apis.py
-src/instrumentation/openai/constants.py
 src/instrumentation/openai/instrumentation.py
 src/instrumentation/openai/patch.py
 src/instrumentation/openai/token_estimation.py
-src/instrumentation/openai/lib/__init__.py
 src/instrumentation/pinecone/__init__.py
-src/instrumentation/pinecone/apis.py
 src/instrumentation/pinecone/instrumentation.py
 src/instrumentation/pinecone/patch.py
-src/instrumentation/pinecone/lib/__init__.py
 src/langtrace_python_sdk.egg-info/PKG-INFO
 src/langtrace_python_sdk.egg-info/SOURCES.txt
 src/langtrace_python_sdk.egg-info/dependency_links.txt
 src/langtrace_python_sdk.egg-info/requires.txt
-src/langtrace_python_sdk.egg-info/top_level.txt
+src/langtrace_python_sdk.egg-info/top_level.txt
+src/utils/__init__.py
+src/utils/with_root_span.py

langtrace-python-sdk-1.0.12/src/langtrace_python_sdk.egg-info/top_level.txt

@@ -0,0 +1,6 @@
+constants
+examples
+extensions
+init
+instrumentation
+utils

langtrace-python-sdk-1.0.11/src/examples/setup.py

@@ -1,50 +0,0 @@
-
-from opentelemetry import trace
-from opentelemetry.sdk.trace import TracerProvider
-from opentelemetry.sdk.trace.export import (ConsoleSpanExporter,
-                                            SimpleSpanProcessor)
-
-from instrumentation.chroma.instrumentation import ChromaInstrumentation
-from instrumentation.langchain.instrumentation import LangchainInstrumentation
-from instrumentation.langchain_community.instrumentation import \
-    LangchainCommunityInstrumentation
-from instrumentation.langchain_core.instrumentation import \
-    LangchainCoreInstrumentation
-from instrumentation.llamaindex.instrumentation import \
-    LlamaindexInstrumentation
-from instrumentation.openai.instrumentation import OpenAIInstrumentation
-from instrumentation.pinecone.instrumentation import PineconeInstrumentation
-
-
-def setup_instrumentation():
-
-    # Set up OpenTelemetry tracing
-    tracer_provider = TracerProvider()
-
-    # Use the ConsoleSpanExporter to print traces to the console
-    console_exporter = ConsoleSpanExporter()
-    tracer_provider.add_span_processor(SimpleSpanProcessor(console_exporter))
-
-    # Initialize tracer
-    trace.set_tracer_provider(tracer_provider)
-
-    # Initialize and enable your custom OpenAI instrumentation
-    # Create an instance of OpenAIInstrumentation
-    openai_instrumentation = OpenAIInstrumentation()
-    pinecone_instrumentation = PineconeInstrumentation()
-    llamaindex_instrumentation = LlamaindexInstrumentation()
-    chroma_instrumentation = ChromaInstrumentation()
-    langchain_instrumentation = LangchainInstrumentation()
-    langchain_core_instrumentation = LangchainCoreInstrumentation()
-    langchain_community_instrumentation = LangchainCommunityInstrumentation()
-
-    # Call the instrument method with some arguments
-    openai_instrumentation.instrument()
-    pinecone_instrumentation.instrument()
-    llamaindex_instrumentation.instrument()
-    chroma_instrumentation.instrument()
-    langchain_instrumentation.instrument()
-    langchain_core_instrumentation.instrument()
-    langchain_community_instrumentation.instrument()
-
-    print("setup complete")

langtrace-python-sdk-1.0.11/src/instrumentation/openai/apis.py

@@ -1,19 +0,0 @@
-"""
-APIs to instrument OpenAI.
-"""
-from langtrace.trace_attributes import OpenAIMethods
-
-APIS = {
-    "CHAT_COMPLETION": {
-        "METHOD": OpenAIMethods.CHAT_COMPLETION.value,
-        "ENDPOINT": "/chat/completions",
-    },
-    "IMAGES_GENERATION": {
-        "METHOD": OpenAIMethods.IMAGES_GENERATION.value,
-        "ENDPOINT": "/images/generations",
-    },
-    "EMBEDDINGS_CREATE": {
-        "METHOD": OpenAIMethods.EMBEDDINGS_CREATE.value,
-        "ENDPOINT": "/embeddings",
-    },
-}

langtrace-python-sdk-1.0.11/src/langtrace_python_sdk.egg-info/top_level.txt

@@ -1,2 +0,0 @@
-examples
-instrumentation