Rm extraneous args to the trace group helper (#5801)

These are being ignored
searx_updates
Zander Chase 11 months ago committed by GitHub
parent 8ef7274ee6
commit 3af36943e8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -133,15 +133,11 @@ def trace_as_chain_group(
*,
session_name: Optional[str] = None,
example_id: Optional[Union[str, UUID]] = None,
tenant_id: Optional[str] = None,
session_extra: Optional[Dict[str, Any]] = None,
) -> Generator[CallbackManager, None, None]:
"""Get a callback manager for a chain group in a context manager."""
cb = LangChainTracer(
tenant_id=tenant_id,
session_name=session_name,
example_id=example_id,
session_extra=session_extra,
)
cm = CallbackManager.configure(
inheritable_callbacks=[cb],
@ -158,15 +154,11 @@ async def atrace_as_chain_group(
*,
session_name: Optional[str] = None,
example_id: Optional[Union[str, UUID]] = None,
tenant_id: Optional[str] = None,
session_extra: Optional[Dict[str, Any]] = None,
) -> AsyncGenerator[AsyncCallbackManager, None]:
"""Get a callback manager for a chain group in a context manager."""
cb = LangChainTracer(
tenant_id=tenant_id,
session_name=session_name,
example_id=example_id,
session_extra=session_extra,
)
cm = AsyncCallbackManager.configure(
inheritable_callbacks=[cb],

Loading…
Cancel
Save