Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-08 07:10:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle Managed Motorhead Data Key (#6169)

Browse Source 
# Handle Managed Motorhead Data Key
Managed motorhead will return a payload with a `data` key. we need to
handle this to properly access messages from the server.
This commit is contained in:
James O'Dwyer 2023-06-16 23:36:18 -04:00 committed by GitHub
parent 364f8e7b5d
commit 0475d015fe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
langchain/memory/motorhead_memory.py
View File

@ -48,6 +48,8 @@ class MotorheadMemory(BaseChatMemory):
headers=self.__get_headers(), headers=self.__get_headers(),
) )
res_data = res.json() res_data = res.json()
res_data = res_data.get("data", res_data) # Handle Managed Version
messages = res_data.get("messages", []) messages = res_data.get("messages", [])
context = res_data.get("context", "NONE") context = res_data.get("context", "NONE")