Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
88 commits
Select commit Hold shift + click to select a range
4d717b1
Merge branch 'development' into release
karthikscale3 Apr 24, 2024
0233826
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Apr 28, 2024
7f4e951
Merge branch 'development' into release
karthikscale3 Apr 28, 2024
81a6ca0
Merge
karthikscale3 Jun 13, 2024
0c19f77
Merge branch 'development' into release
karthikscale3 Jun 13, 2024
c3a6ccf
remove logs
karthikscale3 Jun 13, 2024
a99cf10
remove requirements
karthikscale3 Jun 13, 2024
1379b27
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Jun 17, 2024
dae04e7
Merge branch 'development' into release
karthikscale3 Jun 17, 2024
129e927
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Jun 24, 2024
16e67f9
Merge branch 'development' into release
karthikscale3 Jun 24, 2024
e604e93
Bump version
karthikscale3 Jun 24, 2024
7e00473
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Jun 24, 2024
6ac71aa
Merge branch 'development' into release
karthikscale3 Jun 24, 2024
c39bf01
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Jun 24, 2024
f89e38c
Merge branch 'development' into release
karthikscale3 Jun 24, 2024
e95e743
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Jul 19, 2024
c62e803
Squash
karthikscale3 Jul 25, 2024
d7fd3fb
Merge
karthikscale3 Jul 25, 2024
c4ea507
Merge branch 'development' into release
karthikscale3 Jul 25, 2024
4c74fd8
Merge
karthikscale3 Jul 31, 2024
9a83e20
Merge branch 'development' into release
karthikscale3 Jul 31, 2024
09d5631
Merge
karthikscale3 Aug 3, 2024
508e72b
Merge
karthikscale3 Aug 3, 2024
ad44fa3
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Aug 13, 2024
ad168b3
Merge branch 'development' into release
karthikscale3 Aug 13, 2024
6876f92
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Aug 30, 2024
630169a
Merge branch 'development' into release
karthikscale3 Aug 30, 2024
0e1aae3
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Sep 1, 2024
c266698
Merge branch 'development' into release
karthikscale3 Sep 1, 2024
5b9895f
merge
karthikscale3 Sep 4, 2024
04fd825
Merge branch 'development' into release
karthikscale3 Sep 4, 2024
510e4b8
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Sep 6, 2024
e63bee2
Merge branch 'development' into release
karthikscale3 Sep 6, 2024
9741a3e
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Sep 8, 2024
4f7f3c4
Merge branch 'development' into release
karthikscale3 Sep 8, 2024
32fac05
merge
karthikscale3 Sep 11, 2024
4920395
Merge branch 'development' into release
karthikscale3 Sep 11, 2024
2e8956d
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Sep 11, 2024
372ae29
Merge branch 'development' into release
karthikscale3 Sep 11, 2024
da82f72
bump version
karthikscale3 Sep 26, 2024
ef0332f
Merge branch 'development' into release
karthikscale3 Sep 26, 2024
105749c
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Oct 8, 2024
0f99827
Merge branch 'development' into release
karthikscale3 Oct 8, 2024
b52fce7
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Oct 8, 2024
ddaf4b5
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Oct 10, 2024
54cddc8
Merge branch 'development' into release
karthikscale3 Oct 10, 2024
ced0a31
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Oct 16, 2024
4e22bde
Merge branch 'development' into release
karthikscale3 Oct 16, 2024
135bdc7
bump version
karthikscale3 Oct 18, 2024
6b0a460
Merge branch 'development' into release
karthikscale3 Oct 22, 2024
7824c39
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Oct 22, 2024
109990a
minor
karthikscale3 Oct 22, 2024
2e9befb
version
karthikscale3 Oct 31, 2024
21cdc08
Merge branch 'development' into release
karthikscale3 Oct 31, 2024
7959744
Merge branch 'development' into release
karthikscale3 Oct 31, 2024
aa3dc2f
switch to http exporter
karthikscale3 Oct 31, 2024
4e22e16
Include setuptools to deps
karthikscale3 Nov 15, 2024
a7b34ab
Merge branch 'development' into release
karthikscale3 Nov 15, 2024
736fb60
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Nov 19, 2024
41c58e8
Merge branch 'development' into release
karthikscale3 Nov 19, 2024
129f764
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Nov 25, 2024
f7188c5
Merge branch 'development' into release
karthikscale3 Nov 25, 2024
d43358a
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Dec 1, 2024
fa7b13b
Merge branch 'development' into release
karthikscale3 Dec 1, 2024
fab8619
merge
karthikscale3 Dec 10, 2024
5649f0a
Merge branch 'development' into release
karthikscale3 Dec 10, 2024
0274620
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Jan 20, 2025
0897cc7
Merge branch 'development' into release
karthikscale3 Jan 20, 2025
8eee1e1
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Jan 20, 2025
8cc7f7c
Merge branch 'development' into release
karthikscale3 Jan 20, 2025
d6b6531
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Jan 20, 2025
30b6e00
Merge branch 'development' into release
karthikscale3 Jan 20, 2025
1d7362a
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Jan 22, 2025
36ba737
Merge branch 'development' into release
karthikscale3 Jan 22, 2025
d9bcdb4
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Jan 27, 2025
a5f3da6
Merge branch 'development' into release
karthikscale3 Jan 27, 2025
b856e03
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Jan 27, 2025
5f14b58
Merge branch 'development' into release
karthikscale3 Jan 27, 2025
fd9baed
merge
karthikscale3 Feb 6, 2025
ece9c97
Merge branch 'development' into release
karthikscale3 Feb 6, 2025
cd1a481
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Feb 7, 2025
a3730af
Merge branch 'development' into release
karthikscale3 Feb 7, 2025
5453a92
merge
karthikscale3 Mar 5, 2025
5d65fb3
Merge branch 'development' into release
karthikscale3 Mar 5, 2025
858ca64
Add support for openai-agents (#495)
karthikscale3 Mar 12, 2025
0661c29
Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk in…
karthikscale3 Mar 12, 2025
a51cff1
Merge branch 'development' into release
karthikscale3 Mar 12, 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
3 changes: 2 additions & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,8 @@ requires-python = ">=3.9"

[project.optional-dependencies]
dev = [
"openai==1.60.0",
"openai>=1.60.0",
"openai-agents>=0.0.3",
"anthropic",
"chromadb",
"qdrant-client",
Expand Down
2 changes: 2 additions & 0 deletions src/langtrace_python_sdk/instrumentation/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
from .mistral import MistralInstrumentation
from .ollama import OllamaInstrumentor
from .openai import OpenAIInstrumentation
from .openai_agents import OpenAIAgentsInstrumentation
from .phidata import PhiDataInstrumentation
from .pinecone import PineconeInstrumentation
from .pymongo import PyMongoInstrumentation
Expand Down Expand Up @@ -64,4 +65,5 @@
"PhiDataInstrumentation",
"AgnoInstrumentation",
"CleanLabInstrumentation",
"OpenAIAgentsInstrumentation",
]
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
"""
Copyright (c) 2024 Scale3 Labs
Copyright (c) 2025 Scale3 Labs

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
Expand All @@ -14,13 +14,16 @@
limitations under the License.
"""

from typing import Collection

from importlib_metadata import version as v
from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
from opentelemetry.trace import get_tracer
from wrapt import wrap_function_wrapper as _W
from typing import Collection
from importlib_metadata import version as v

from .patch import patch_agent, patch_memory


class AgnoInstrumentation(BaseInstrumentor):
def instrumentation_dependencies(self) -> Collection[str]:
return ["agno >= 1.1.4"]
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
from .instrumentation import OpenAIAgentsInstrumentation

__all__ = [
"OpenAIAgentsInstrumentation",
]
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
"""
Copyright (c) 2025 Scale3 Labs
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""

import importlib.metadata
import logging
from typing import Any, Collection, Optional

from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
from opentelemetry.trace import TracerProvider, get_tracer
from wrapt import wrap_function_wrapper

from langtrace_python_sdk.instrumentation.openai_agents.patch import \
get_new_response

logging.basicConfig(level=logging.FATAL)


class OpenAIAgentsInstrumentation(BaseInstrumentor): # type: ignore

def instrumentation_dependencies(self) -> Collection[str]:
return ["openai-agents >= 0.0.3", "trace-attributes >= 4.0.5"]

def _instrument(self, **kwargs: Any) -> None:
tracer_provider: Optional[TracerProvider] = kwargs.get("tracer_provider")
tracer = get_tracer(__name__, "", tracer_provider)
version: str = importlib.metadata.version("openai")

# TODO(Karthik): This is adding a lot of noise to the trace.
# wrap_function_wrapper(
# "agents.run",
# "Runner._get_handoffs",
# get_handoffs(version, tracer),
# )

wrap_function_wrapper(
"agents.run",
"Runner._get_new_response",
get_new_response(version, tracer),
)

def _uninstrument(self, **kwargs: Any) -> None:
pass
Loading
Loading