Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
27 commits
Select commit Hold shift + click to select a range
2349adc
add instrumentation for neo4j
obinnaokafor Mar 20, 2025
9826364
update version
obinnaokafor Mar 20, 2025
ea6bcd7
update trace attributes version
obinnaokafor Mar 20, 2025
6379505
update imports
obinnaokafor Mar 24, 2025
04766a9
Merge pull request #507 from Scale3-Labs/obinna/S3EN-1111-support-neo…
obinnascale3 Mar 24, 2025
ed83a02
add neo4j result span attributes
obinnaokafor Mar 26, 2025
a3c4c2e
add neo4j result span attributes
obinnaokafor Mar 26, 2025
081e457
bump bersion
obinnaokafor Mar 26, 2025
565f3b9
fix db attributes error and add examples
obinnaokafor Mar 26, 2025
017df27
handle result transformer
obinnaokafor Mar 27, 2025
a76cb32
Merge pull request #510 from Scale3-Labs/obinna/S3EN-1111-update-neo4…
obinnascale3 Mar 27, 2025
75b0db2
fix genai and pinecone instrumentations (#512)
obinnascale3 Apr 2, 2025
7b0810d
Improve Agno instrumentation (#513)
karthikscale3 Apr 2, 2025
ae3598b
cast token count to int
obinnaokafor Apr 2, 2025
f83afda
bump version
obinnaokafor Apr 2, 2025
528c3d8
only redirect stdout if logging is disabled
obinnaokafor Apr 2, 2025
624cdc3
bump version
obinnaokafor Apr 2, 2025
56e78cb
fix openai agents attribute error
obinnaokafor Apr 2, 2025
519cfdb
bump version
obinnaokafor Apr 2, 2025
34d9396
Merge pull request #518 from Scale3-Labs/obinna/S3EN-3020-fix-openai-…
obinnascale3 Apr 2, 2025
cbaa8aa
Merge pull request #517 from Scale3-Labs/obinna/S3EN-2977-fix-sysout-…
obinnascale3 Apr 2, 2025
db4ca91
Merge pull request #516 from Scale3-Labs/obinna/S3EN-1111-fix-cohere-…
obinnascale3 Apr 2, 2025
3777dd1
fix openai responses api instrumentation (#519)
obinnascale3 Apr 2, 2025
2884436
fix JSON serializable error in neo4j (#521)
obinnascale3 Apr 8, 2025
748e02f
clean up warning messages
alizenhom Apr 14, 2025
89b5c9f
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
alizenhom Apr 14, 2025
1b99dc0
Merge pull request #523 from Scale3-Labs/ali/filter-warnings
alizenhom Apr 14, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
90 changes: 63 additions & 27 deletions src/langtrace_python_sdk/langtrace.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,47 +17,79 @@
import logging
import os
import sys
import warnings
from typing import Any, Dict, Optional

import sentry_sdk
from colorama import Fore
from opentelemetry import trace
from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import \
OTLPSpanExporter as GRPCExporter
from opentelemetry.exporter.otlp.proto.http.trace_exporter import \
OTLPSpanExporter as HTTPExporter
from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (
OTLPSpanExporter as GRPCExporter,
)
from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
OTLPSpanExporter as HTTPExporter,
)
from opentelemetry.instrumentation.sqlalchemy import SQLAlchemyInstrumentor
from opentelemetry.sdk.resources import SERVICE_NAME, Resource
from opentelemetry.sdk.trace import TracerProvider
from opentelemetry.sdk.trace.export import (BatchSpanProcessor,
ConsoleSpanExporter,
SimpleSpanProcessor)
from opentelemetry.sdk.trace.export import (
BatchSpanProcessor,
ConsoleSpanExporter,
SimpleSpanProcessor,
)
from opentelemetry.util.re import parse_env_headers
from sentry_sdk.types import Event, Hint

from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME, SENTRY_DSN
from langtrace_python_sdk.constants.exporter.langtrace_exporter import (
LANGTRACE_REMOTE_URL, LANGTRACE_SESSION_ID_HEADER)
from langtrace_python_sdk.extensions.langtrace_exporter import \
LangTraceExporter
LANGTRACE_REMOTE_URL,
LANGTRACE_SESSION_ID_HEADER,
)
from langtrace_python_sdk.extensions.langtrace_exporter import LangTraceExporter
from langtrace_python_sdk.instrumentation import (
AgnoInstrumentation, AnthropicInstrumentation, AutogenInstrumentation,
AWSBedrockInstrumentation, CerebrasInstrumentation, ChromaInstrumentation,
CleanLabInstrumentation, CohereInstrumentation, CrewAIInstrumentation,
CrewaiToolsInstrumentation, DspyInstrumentation, EmbedchainInstrumentation,
GeminiInstrumentation, GoogleGenaiInstrumentation, GraphlitInstrumentation,
GroqInstrumentation, LangchainCommunityInstrumentation,
LangchainCoreInstrumentation, LangchainInstrumentation, LanggraphInstrumentation,
LiteLLMInstrumentation, LlamaindexInstrumentation, MilvusInstrumentation,
MistralInstrumentation, Neo4jInstrumentation, Neo4jGraphRAGInstrumentation,
OllamaInstrumentor, OpenAIAgentsInstrumentation, OpenAIInstrumentation,
PhiDataInstrumentation, PineconeInstrumentation, PyMongoInstrumentation,
QdrantInstrumentation, VertexAIInstrumentation, WeaviateInstrumentation)
from langtrace_python_sdk.types import (DisableInstrumentations,
InstrumentationMethods)
from langtrace_python_sdk.utils import (check_if_sdk_is_outdated,
get_sdk_version, is_package_installed,
validate_instrumentations)
AgnoInstrumentation,
AnthropicInstrumentation,
AutogenInstrumentation,
AWSBedrockInstrumentation,
CerebrasInstrumentation,
ChromaInstrumentation,
CleanLabInstrumentation,
CohereInstrumentation,
CrewAIInstrumentation,
CrewaiToolsInstrumentation,
DspyInstrumentation,
EmbedchainInstrumentation,
GeminiInstrumentation,
GoogleGenaiInstrumentation,
GraphlitInstrumentation,
GroqInstrumentation,
LangchainCommunityInstrumentation,
LangchainCoreInstrumentation,
LangchainInstrumentation,
LanggraphInstrumentation,
LiteLLMInstrumentation,
LlamaindexInstrumentation,
MilvusInstrumentation,
MistralInstrumentation,
Neo4jInstrumentation,
Neo4jGraphRAGInstrumentation,
OllamaInstrumentor,
OpenAIAgentsInstrumentation,
OpenAIInstrumentation,
PhiDataInstrumentation,
PineconeInstrumentation,
PyMongoInstrumentation,
QdrantInstrumentation,
VertexAIInstrumentation,
WeaviateInstrumentation,
)
from langtrace_python_sdk.types import DisableInstrumentations, InstrumentationMethods
from langtrace_python_sdk.utils import (
check_if_sdk_is_outdated,
get_sdk_version,
is_package_installed,
validate_instrumentations,
)
from langtrace_python_sdk.utils.langtrace_sampler import LangtraceSampler


Expand Down Expand Up @@ -329,6 +361,8 @@ def init_instrumentations(
if is_package_installed(name):
try:
v.instrument()
warnings.filterwarnings("ignore", category=DeprecationWarning)
warnings.filterwarnings("ignore", category=UserWarning)
except Exception as e:
print(f"Skipping {name} due to error while instrumenting: {e}")

Expand All @@ -354,5 +388,7 @@ def init_instrumentations(
if is_package_installed(name):
try:
v.instrument()
warnings.filterwarnings("ignore", category=DeprecationWarning)
warnings.filterwarnings("ignore", category=UserWarning)
except Exception as e:
print(f"Skipping {name} due to error while instrumenting: {e}")
2 changes: 1 addition & 1 deletion src/langtrace_python_sdk/version.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
__version__ = "3.8.12"
__version__ = "3.8.13"
Loading