Handle Managed Motorhead Data Key (#6169)

softboyjimbo · web-flow · commit 0475d015fe0e · 2023-06-16T20:36:18.000-07:00
# 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.
diff --git a/langchain/memory/motorhead_memory.py b/langchain/memory/motorhead_memory.py
@@ -48,6 +48,8 @@ async def init(self) -> None:
             headers=self.__get_headers(),
         )
         res_data = res.json()
+        res_data = res_data.get("data", res_data)  # Handle Managed Version
+
         messages = res_data.get("messages", [])
         context = res_data.get("context", "NONE")
 

Original file line number	Diff line number	Diff line change
`@@ -48,6 +48,8 @@ async def init(self) -> None:`
`48`	`48`	`headers=self.__get_headers(),`
`49`	`49`	`)`
`50`	`50`	`res_data = res.json()`
	`51`	`+ res_data = res_data.get("data", res_data) # Handle Managed Version`
	`52`	`+`
`51`	`53`	`messages = res_data.get("messages", [])`
`52`	`54`	`context = res_data.get("context", "NONE")`
`53`	`55`