github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-06 07:04:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update ContextCallbackHandler Docstring & metadata key (#10732)

Browse Source 
- **Description:** Updating URL in Context Callback Docstrings and
update metadata key Context CallbackHandler uses to send model names.
- **Issue:** The URL in ContextCallbackHandler is out of date. Model
data being sent to Context should be under the "model" key and not
"llm_model". This allows Context to do more sophisticated analysis.
  - **Dependencies:** None

Tagging @agamble.
This commit is contained in:
BarberAlec 2023-09-19 06:04:13 +01:00 committed by GitHub
parent 54763a61f8
commit c898a4d7ba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/langchain/langchain/callbacks/context_callback.py
View File

@ -33,7 +33,7 @@ def import_context() -> Any:
class ContextCallbackHandler(BaseCallbackHandler): class ContextCallbackHandler(BaseCallbackHandler):
"""Callback Handler that records transcripts to the Context service. """Callback Handler that records transcripts to the Context service.
(https://getcontext.ai). (https://context.ai).
Keyword Args: Keyword Args:
token (optional): The token with which to authenticate requests to Context. token (optional): The token with which to authenticate requests to Context.
@ -122,7 +122,7 @@ class ContextCallbackHandler(BaseCallbackHandler):
"""Run when the chat model is started.""" """Run when the chat model is started."""
llm_model = kwargs.get("invocation_params", {}).get("model", None) llm_model = kwargs.get("invocation_params", {}).get("model", None)
if llm_model is not None: if llm_model is not None:
self.metadata["llm_model"] = llm_model self.metadata["model"] = llm_model
if len(messages) == 0: if len(messages) == 0:
return return