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

mend

Browse Source
This commit is contained in:
Mikolaj Mikuliszyn 2025-04-27 17:48:12 +01:00
parent 21c491a022
commit d6a032d8e7
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
libs/langchain/langchain/memory/persona.py
View File

@ -1,12 +1,14 @@
from typing import Any, Callable, Optional
from pydantic import BaseModel, Field
import re import re
from typing import Any, Callable, Optional
from pydantic import BaseModel, Field
from langchain_core.memory import BaseMemory from langchain_core.memory import BaseMemory
class EnrichedMessage(BaseModel): class EnrichedMessage(BaseModel):
"""A message enriched with persona traits and metadata.""" """
A message enriched with persona traits and metadata.
"""
id: str id: str
content: str content: str
@ -43,7 +45,8 @@ class PersonaMemory(BaseMemory):
def _detect_traits(self, text: str) -> dict[str, int]: def _detect_traits(self, text: str) -> dict[str, int]:
""" """
Detect persona traits using both a default method and optionally an external engine. Detect persona traits using both a default method and optionally
an external engine.
Always guarantees a usable result, even if external services fail. Always guarantees a usable result, even if external services fail.
""" """
@ -94,7 +97,10 @@ class PersonaMemory(BaseMemory):
def load_memory_variables( def load_memory_variables(
self, inputs: dict[str, Any], include_messages: bool = False self, inputs: dict[str, Any], include_messages: bool = False
) -> dict[str, Any]: ) -> dict[str, Any]:
"""Return the stored persona traits and optionally recent conversation messages.""" """
Return the stored persona traits and optionally recent
conversation messages.
"""
memory_data = {"traits": self.persona_traits.copy()} memory_data = {"traits": self.persona_traits.copy()}
if include_messages: if include_messages:
@ -105,7 +111,9 @@ class PersonaMemory(BaseMemory):
return {self.memory_key: memory_data} return {self.memory_key: memory_data}
def save_context(self, inputs: dict[str, Any], outputs: dict[str, str]) -> None: def save_context(self, inputs: dict[str, Any], outputs: dict[str, str]) -> None:
"""Analyze outputs and update the persona traits and recent messages.""" """
Analyze outputs and update the persona traits and recent messages.
"""
output_text = outputs.get(self.output_key, "") output_text = outputs.get(self.output_key, "")
traits_detected = self._detect_traits(output_text) traits_detected = self._detect_traits(output_text)