langtrace-python-sdk 1.2.21__py3-none-any.whl → 1.2.25__py3-none-any.whl

examples/perplexity_example/basic.py

@@ -0,0 +1,21 @@
+from openai import OpenAI
+
+from langtrace_python_sdk import langtrace
+from langtrace_python_sdk.utils.with_root_span import (
+    with_additional_attributes, with_langtrace_root_span)
+
+# _ = load_dotenv(find_dotenv())
+
+langtrace.init(batch=False, write_to_langtrace_cloud=False, debug_log_to_console=True)
+client = OpenAI(base_url="https://api.perplexity.ai", api_key="PPLX_API_KEY")
+
+
+@with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
+def basic():
+    response = client.chat.completions.create(
+        model="pplx-70b-online",
+        messages=[{"role": "user", "content": "What is the capital of France?"}],
+        stream=False,
+    )
+    print(response)
+    return response

langtrace_python_sdk/constants/exporter/langtrace_exporter.py

@@ -1 +1 @@
-LANGTRACE_REMOTE_URL = "https://langtrace.ai/api/trace"
+LANGTRACE_REMOTE_URL = "https://langtrace.ai"

langtrace_python_sdk/constants/instrumentation/common.py

@@ -16,6 +16,7 @@ SERVICE_PROVIDERS = {
     "LLAMAINDEX": "LlamaIndex",
     "OPENAI": "OpenAI",
     "PINECONE": "Pinecone",
+    "PPLX": "Perplexity",
 }
 
 LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY = "langtrace_additional_attributes"

langtrace_python_sdk/extensions/langtrace_exporter.py

@@ -2,13 +2,12 @@ import json
 import os
 import typing
 
-from langtrace_python_sdk.constants.exporter.langtrace_exporter import (
-    LANGTRACE_REMOTE_URL,
-)
 import requests
 from opentelemetry.sdk.trace.export import ReadableSpan, SpanExporter, SpanExportResult
 from opentelemetry.trace.span import format_trace_id
 
+from langtrace_python_sdk.constants.exporter.langtrace_exporter import LANGTRACE_REMOTE_URL
+
 
 class LangTraceExporter(SpanExporter):
     """
@@ -49,9 +48,10 @@ class LangTraceExporter(SpanExporter):
     api_key: str
     write_to_remote_url: bool
 
-    def __init__(self, api_key: str = None, write_to_remote_url: bool = False) -> None:
-        self.api_key = api_key if api_key else os.environ.get("LANGTRACE_API_KEY")
+    def __init__(self, api_key: str = None, write_to_remote_url: bool = False, api_host: typing.Optional[str] = None) -> None:
+        self.api_key = api_key or os.environ.get("LANGTRACE_API_KEY")
         self.write_to_remote_url = write_to_remote_url
+        self.api_host: str = api_host or LANGTRACE_REMOTE_URL
 
         if self.write_to_remote_url and not self.api_key:
             raise ValueError("No API key provided")
@@ -66,7 +66,6 @@ class LangTraceExporter(SpanExporter):
         Returns:
             The result of the export SUCCESS or FAILURE
         """
-
         data = [
             {
                 "traceId": format_trace_id(span.get_span_context().trace_id),
@@ -86,7 +85,7 @@ class LangTraceExporter(SpanExporter):
         # Send data to remote URL
         try:
             requests.post(
-                url=LANGTRACE_REMOTE_URL,
+                url=f"{self.api_host}/api/trace",
                 data=json.dumps(data),
                 headers={"Content-Type": "application/json", "x-api-key": self.api_key},
             )

langtrace_python_sdk/instrumentation/openai/patch.py

@@ -97,6 +97,12 @@ def chat_completions_create(original_method, version, tracer):
             else ""
         )
         service_provider = SERVICE_PROVIDERS["OPENAI"]
+        # If base url contains perplexity or azure, set the service provider accordingly
+        if "perplexity" in base_url:
+            service_provider = SERVICE_PROVIDERS["PPLX"]
+        elif "azure" in base_url:
+            service_provider = SERVICE_PROVIDERS["AZURE"]
+
         extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
 
         span_attributes = {

langtrace_python_sdk/langtrace.py

@@ -1,4 +1,4 @@
-
+from typing import Optional
 from opentelemetry import trace
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import (
@@ -40,12 +40,12 @@ def init(
     write_to_langtrace_cloud: bool = True,
     debug_log_to_console: bool = False,
     custom_remote_exporter=None,
+    api_host: Optional[str] = None,
 ):
-
     provider = TracerProvider()
 
     remote_write_exporter = (
-        LangTraceExporter(api_key, write_to_langtrace_cloud)
+        LangTraceExporter(api_key, write_to_langtrace_cloud, api_host=api_host)
         if custom_remote_exporter is None
         else custom_remote_exporter
     )

langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "1.2.21"
+__version__ = "1.2.25"

{langtrace_python_sdk-1.2.21.dist-info → langtrace_python_sdk-1.2.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 1.2.21
+Version: 1.2.25
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>
@@ -72,6 +72,68 @@ from langtrace_python_sdk import langtrace # Must precede any llm module imports
 langtrace.init() # LANGTRACE_API_KEY as an ENVIRONMENT variable
 ```
 
+## FastAPI Quick Start
+Initialize FastAPI project and add this inside the ```main.py``` file
+
+```python
+from fastapi import FastAPI
+from langtrace_python_sdk import langtrace
+from openai import OpenAI
+
+langtrace.init()
+app = FastAPI()
+client = OpenAI()
+
+@app.get("/")
+def root():
+    client.chat.completions.create(
+        model="gpt-4",
+        messages=[{"role": "user", "content": "Say this is a test three times"}],
+        stream=False,
+    )
+    return {"Hello": "World"}
+```
+
+## Django Quick Start
+Initialize django project and add this inside the ```__init.py__``` file
+```python
+from langtrace_python_sdk import langtrace
+from openai import OpenAI
+
+
+langtrace.init()
+client = OpenAI()
+
+client.chat.completions.create(
+    model="gpt-4",
+    messages=[{"role": "user", "content": "Say this is a test three times"}],
+    stream=False,
+)
+
+```
+
+## Flask Quick Start
+Initialize flask project and this inside ```app.py``` file
+```python
+from flask import Flask
+from langtrace_python_sdk import langtrace
+from openai import OpenAI
+
+langtrace.init()
+client = OpenAI()
+app = Flask(__name__)
+
+
+@app.route("/")
+def main():
+    client.chat.completions.create(
+        model="gpt-4",
+        messages=[{"role": "user", "content": "Say this is a test three times"}],
+        stream=False,
+    )
+    return "Hello, World!"
+```
+
 ## Langtrace Self Hosted
 
 Get started by adding simply two lines to your code and see traces being logged to the console!

{langtrace_python_sdk-1.2.21.dist-info → langtrace_python_sdk-1.2.25.dist-info}/RECORD

@@ -14,21 +14,22 @@ examples/openai/chat_completion.py,sha256=ynVWdvEwDxNPpkWu9fv4al7_1mm3rFilFvqkrc
 examples/openai/embeddings_create.py,sha256=AhDNAqg-WzRYLJAE_b2RKGjuVCh4aZSU7MxcZv2kCHQ,518
 examples/openai/function_calling.py,sha256=Zd1gkl1TU-g_cxXox3_8vmFRZVyZSyyLaboEJLc6N2A,2375
 examples/openai/images_generate.py,sha256=ZioxTuHKE_yYlhpESqXKVzdkiwdegkmLVB7N8T2LU00,506
+examples/perplexity_example/basic.py,sha256=jEqZnY9cmDezMAwMvKaZeRk7vW1H68Zc2de3gDnb6hg,704
 examples/pinecone_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 examples/pinecone_example/basic.py,sha256=hdV6-5Fmol9zeyFzDtdabD62vkqUJ4lCHG2YcVNpIpI,933
 langtrace_python_sdk/__init__.py,sha256=SlHg447-nQBbw8exRNJP_OyHUZ39Sldb7aaQ35hIRm8,262
-langtrace_python_sdk/langtrace.py,sha256=sWC7WqbOdOUB1sZ_hzLZVp0_Q8fBTRJRMUvOPyO6f20,3170
-langtrace_python_sdk/version.py,sha256=3uF8dV6igE_qWszFViblw_rtJOgCwoi-rVuWcQrSR0A,23
+langtrace_python_sdk/langtrace.py,sha256=0gT4eCn8-hSSfI-B5zxUmd7weHvuCsnS4B7N4EmFzHs,3251
+langtrace_python_sdk/version.py,sha256=9j001fhDZzLSWs6YCbGQFuK5ERX5EUWz3RKLN8l5JS8,23
 langtrace_python_sdk/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/constants/exporter/langtrace_exporter.py,sha256=OfgvphRBPENwsAizqzka9lFu9aDw-A7ErYJ0atl3bpI,56
+langtrace_python_sdk/constants/exporter/langtrace_exporter.py,sha256=5MNjnAOg-4am78J3gVMH6FSwq5N8TOj72ugkhsw4vi0,46
 langtrace_python_sdk/constants/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/constants/instrumentation/anthropic.py,sha256=YX3llt3zwDY6XrYk3CB8WEVqgrzRXEw_ffyk56JoF3k,126
 langtrace_python_sdk/constants/instrumentation/chroma.py,sha256=hiPGYdHS0Yj4Kh3eaYBbuCAl_swqIygu80yFqkOgdak,955
-langtrace_python_sdk/constants/instrumentation/common.py,sha256=KzNCWEl_44pkZ5DcBBLAInh-QNgSMPSBkbhIcC8rH0A,602
+langtrace_python_sdk/constants/instrumentation/common.py,sha256=cLJDG_wbTZnoy0sQRjTbuouWs6vuCUvhM_gRATeKY1o,628
 langtrace_python_sdk/constants/instrumentation/openai.py,sha256=9VF6ic9Ed3bpSvdp6iNmrpx2Ppo6DPav3hoUcqSQSv0,1048
 langtrace_python_sdk/constants/instrumentation/pinecone.py,sha256=Xaqqw-xBO0JJLGk75hiCUQGztNm0HiVaLQvjtYK7VJM,472
 langtrace_python_sdk/extensions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/extensions/langtrace_exporter.py,sha256=dEMSwkbqVMMToD6mH1m3x4i0Dk0g4I8-oNhOmoFZDBA,3980
+langtrace_python_sdk/extensions/langtrace_exporter.py,sha256=K-grSUEBkSGfo_LwyyZDr_iz3oDy1mHnx34FfD-qFSQ,4067
 langtrace_python_sdk/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/anthropic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/anthropic/instrumentation.py,sha256=DTvz6fNMMxYc2z6oqOveBX6mIXHoOFeOoosyXB2HJ6Y,1159
@@ -50,14 +51,14 @@ langtrace_python_sdk/instrumentation/llamaindex/instrumentation.py,sha256=wVSvxS
 langtrace_python_sdk/instrumentation/llamaindex/patch.py,sha256=NqBsPM4eE3yMu1aIev8p8yH5Thsv72JL68ABEZNe2ug,2043
 langtrace_python_sdk/instrumentation/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/openai/instrumentation.py,sha256=stlFopPggB8mgnn4QdVh41rmA8mlzH5x-C00XHfApCE,1408
-langtrace_python_sdk/instrumentation/openai/patch.py,sha256=iP4RFL5ZHqu3BlNNq5DmmstLMaqaXmk4VKRbGXts1EI,15341
+langtrace_python_sdk/instrumentation/openai/patch.py,sha256=0XPs8KNHxBZ-L_L_M4GThq0z3xZaE4C9NT1pcxc3k2o,15617
 langtrace_python_sdk/instrumentation/pinecone/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/pinecone/instrumentation.py,sha256=o0EUd5jvHaDKOUTj4NjnL5UfDHDHxyXkWGlTW4oeRDk,1784
 langtrace_python_sdk/instrumentation/pinecone/patch.py,sha256=5lF7hQmg2-U2EWtOC0w8_peRaNMysBomb0fjiNoS6eQ,2200
 langtrace_python_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/utils/llm.py,sha256=4z2e-md_ELXCEuOIRVWracR6qH2pmsOxCqpkuF9_3Nw,1589
 langtrace_python_sdk/utils/with_root_span.py,sha256=N7ONrcF0myZbHBy5gpQffDbX-Kf63Crsz9szG0i3m08,1889
-langtrace_python_sdk-1.2.21.dist-info/METADATA,sha256=9yDWY-fJ-Sgnc6-M4apfEza7U9X-j0TWn6OQhfWmOZA,7695
-langtrace_python_sdk-1.2.21.dist-info/WHEEL,sha256=as-1oFTWSeWBgyzh0O_qF439xqBe6AbBgt4MfYe5zwY,87
-langtrace_python_sdk-1.2.21.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-1.2.21.dist-info/RECORD,,
+langtrace_python_sdk-1.2.25.dist-info/METADATA,sha256=LJykh5FPkALqNHlY9yFiMtVEe7k8ReIqEh4cwth8BcY,9018
+langtrace_python_sdk-1.2.25.dist-info/WHEEL,sha256=as-1oFTWSeWBgyzh0O_qF439xqBe6AbBgt4MfYe5zwY,87
+langtrace_python_sdk-1.2.25.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-1.2.25.dist-info/RECORD,,