Skip to content

feat(core): Instrument LangChain AI #17955

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
RulaKhaled merged 13 commits into from
Oct 22, 2025
Merged

feat(core): Instrument LangChain AI #17955

Changes from 1 commit
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
a89edfc
feat(core): Instrument LangChain AI
RulaKhaled Oct 16, 2025
c8d9d4b
quick refactors
RulaKhaled Oct 17, 2025
ab4350e
fix ci failing
RulaKhaled Oct 17, 2025
e61c710
add missing imports
RulaKhaled Oct 17, 2025
630dfb2
more tests
RulaKhaled Oct 17, 2025
abe63cf
bump bundle size
RulaKhaled Oct 17, 2025
459ab35
nice catch cursor
RulaKhaled Oct 17, 2025
0ef3d71
add op to attributes
RulaKhaled Oct 17, 2025
0029220
update with better types
RulaKhaled Oct 21, 2025
10380b4
yarn lock fix
RulaKhaled Oct 21, 2025
3dc9a50
Merge branch 'develop' into instrument-langchain
RulaKhaled Oct 21, 2025
00cdaf6
resolve comments
RulaKhaled Oct 22, 2025
6cda4e0
check if span is recording before adding to it
RulaKhaled Oct 22, 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
Prev Previous commit
check if span is recording before adding to it
  • Loading branch information
@RulaKhaled
RulaKhaled committed Oct 22, 2025
commit 6cda4e0355eeeca2143137f6eeaa0bd0d35c8611
14 changes: 7 additions & 7 deletions packages/core/src/utils/langchain/index.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ export function createLangChainCallbackHandler(options: LangChainOptions = {}):
*/
const exitSpan = (runId: string): void => {
const span = spanMap.get(runId);
if (span) {
if (span?.isRecording()) {
span.end();
spanMap.delete(runId);
}
Expand Down Expand Up @@ -152,7 +152,7 @@ export function createLangChainCallbackHandler(options: LangChainOptions = {}):
_extraParams?: Record<string, unknown>,
) {
const span = spanMap.get(runId);
if (span) {
if (span?.isRecording()) {
const attributes = extractLlmResponseAttributes(output as LangChainLLMResult, recordOutputs);
if (attributes) {
span.setAttributes(attributes);
Expand All @@ -164,7 +164,7 @@ export function createLangChainCallbackHandler(options: LangChainOptions = {}):
// LLM Error Handler - note: handleLLMError with capital LLM
handleLLMError(error: Error, runId: string) {
const span = spanMap.get(runId);
if (span) {
if (span?.isRecording()) {
span.setStatus({ code: SPAN_STATUS_ERROR, message: 'llm_error' });
exitSpan(runId);
}
Expand Down Expand Up @@ -209,7 +209,7 @@ export function createLangChainCallbackHandler(options: LangChainOptions = {}):
// Chain End Handler
handleChainEnd(outputs: unknown, runId: string) {
const span = spanMap.get(runId);
if (span) {
if (span?.isRecording()) {
// Add outputs if recordOutputs is enabled
if (recordOutputs) {
span.setAttributes({
Expand All @@ -223,7 +223,7 @@ export function createLangChainCallbackHandler(options: LangChainOptions = {}):
// Chain Error Handler
handleChainError(error: Error, runId: string) {
const span = spanMap.get(runId);
if (span) {
if (span?.isRecording()) {
span.setStatus({ code: SPAN_STATUS_ERROR, message: 'chain_error' });
exitSpan(runId);
}
Expand Down Expand Up @@ -268,7 +268,7 @@ export function createLangChainCallbackHandler(options: LangChainOptions = {}):
// Tool End Handler
handleToolEnd(output: unknown, runId: string) {
const span = spanMap.get(runId);
if (span) {
if (span?.isRecording()) {
// Add output if recordOutputs is enabled
if (recordOutputs) {
span.setAttributes({
Expand All @@ -282,7 +282,7 @@ export function createLangChainCallbackHandler(options: LangChainOptions = {}):
// Tool Error Handler
handleToolError(error: Error, runId: string) {
const span = spanMap.get(runId);
if (span) {
if (span?.isRecording()) {
span.setStatus({ code: SPAN_STATUS_ERROR, message: 'tool_error' });
exitSpan(runId);
}
Expand Down
Loading