blob42/langchain
1
0
Fork 0
forked from Archives/langchain
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source 
These are being ignored
This commit is contained in:
Zander Chase 2023-06-07 13:09:29 -07:00 committed by GitHub
parent 8ef7274ee6
commit 3af36943e8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
langchain/callbacks/manager.py
View File

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