Skip to content

Commit c17607d

Browse files
committed
fixed error
1 parent 6e34620 commit c17607d

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

instrumentation-genai/opentelemetry-instrumentation-langchain/src/opentelemetry/instrumentation/langchain/callback_handler.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,8 +77,9 @@ def on_chain_start(
7777
self._invocation_manager.add_invocation_state(run_id, None, wf)
7878
return
7979
else:
80+
# TODO: For agent invocation
8081
self._invocation_manager.add_invocation_state(
81-
run_id, parent_run_id
82+
run_id, parent_run_id,
8283
)
8384

8485
def on_chain_end(

instrumentation-genai/opentelemetry-instrumentation-langchain/src/opentelemetry/instrumentation/langchain/invocation_manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ def add_invocation_state(
3939
self,
4040
run_id: UUID,
4141
parent_run_id: Optional[UUID],
42-
invocation: GenAIInvocation,
42+
invocation: GenAIInvocation = None,
4343
):
4444
invocation_state = _InvocationState(invocation=invocation)
4545
self._invocations[run_id] = invocation_state

0 commit comments

Comments
 (0)