fix(oss): align API naming with platform v3
- Rename `observation_date` param to `timestamp` (matches platform) - Change `rerank` default from True to False (matches platform) - Internal prompt builder still uses "Observation Date" section heading for LLM compatibility with ADDITIVE_EXTRACTION_PROMPT Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 4.6
parent
a33c557a25
commit
80749b92bf
@@ -1056,7 +1056,7 @@ def generate_additive_extraction_prompt(
|
|||||||
*,
|
*,
|
||||||
last_k_messages=None,
|
last_k_messages=None,
|
||||||
current_date=None,
|
current_date=None,
|
||||||
observation_date=None,
|
timestamp=None,
|
||||||
custom_instructions=None,
|
custom_instructions=None,
|
||||||
use_input_language=False,
|
use_input_language=False,
|
||||||
):
|
):
|
||||||
@@ -1065,7 +1065,7 @@ def generate_additive_extraction_prompt(
|
|||||||
Pairs with ADDITIVE_EXTRACTION_PROMPT system prompt.
|
Pairs with ADDITIVE_EXTRACTION_PROMPT system prompt.
|
||||||
The LLM will produce only ADD operations, with optional linked_memory_ids.
|
The LLM will produce only ADD operations, with optional linked_memory_ids.
|
||||||
"""
|
"""
|
||||||
current_date, observation_date = _resolve_dates(current_date, observation_date)
|
current_date, observation_date = _resolve_dates(current_date, timestamp)
|
||||||
|
|
||||||
sections = []
|
sections = []
|
||||||
sections.append(f"## Summary\n{_format_summary(summary)}")
|
sections.append(f"## Summary\n{_format_summary(summary)}")
|
||||||
|
|||||||
+6
-6
@@ -376,7 +376,7 @@ class Memory(MemoryBase):
|
|||||||
infer: bool = True,
|
infer: bool = True,
|
||||||
memory_type: Optional[str] = None,
|
memory_type: Optional[str] = None,
|
||||||
prompt: Optional[str] = None,
|
prompt: Optional[str] = None,
|
||||||
observation_date: Optional[str] = None,
|
timestamp: Optional[str] = None,
|
||||||
):
|
):
|
||||||
"""
|
"""
|
||||||
Create a new memory.
|
Create a new memory.
|
||||||
@@ -455,7 +455,7 @@ class Memory(MemoryBase):
|
|||||||
messages = parse_vision_messages(messages)
|
messages = parse_vision_messages(messages)
|
||||||
|
|
||||||
with concurrent.futures.ThreadPoolExecutor() as executor:
|
with concurrent.futures.ThreadPoolExecutor() as executor:
|
||||||
future1 = executor.submit(self._add_to_vector_store, messages, processed_metadata, effective_filters, infer, observation_date)
|
future1 = executor.submit(self._add_to_vector_store, messages, processed_metadata, effective_filters, infer, timestamp)
|
||||||
future2 = executor.submit(self._add_to_graph, messages, effective_filters)
|
future2 = executor.submit(self._add_to_graph, messages, effective_filters)
|
||||||
|
|
||||||
concurrent.futures.wait([future1, future2])
|
concurrent.futures.wait([future1, future2])
|
||||||
@@ -471,7 +471,7 @@ class Memory(MemoryBase):
|
|||||||
|
|
||||||
return {"results": vector_store_result}
|
return {"results": vector_store_result}
|
||||||
|
|
||||||
def _add_to_vector_store(self, messages, metadata, filters, infer, observation_date=None):
|
def _add_to_vector_store(self, messages, metadata, filters, infer, timestamp=None):
|
||||||
if not infer:
|
if not infer:
|
||||||
returned_memories = []
|
returned_memories = []
|
||||||
for message_dict in messages:
|
for message_dict in messages:
|
||||||
@@ -547,7 +547,7 @@ class Memory(MemoryBase):
|
|||||||
new_messages=parsed_messages,
|
new_messages=parsed_messages,
|
||||||
last_k_messages=last_messages,
|
last_k_messages=last_messages,
|
||||||
custom_instructions=custom_instr,
|
custom_instructions=custom_instr,
|
||||||
observation_date=observation_date,
|
timestamp=timestamp,
|
||||||
)
|
)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
@@ -871,7 +871,7 @@ class Memory(MemoryBase):
|
|||||||
limit: int = 100,
|
limit: int = 100,
|
||||||
filters: Optional[Dict[str, Any]] = None,
|
filters: Optional[Dict[str, Any]] = None,
|
||||||
threshold: float = 0.1,
|
threshold: float = 0.1,
|
||||||
rerank: bool = True,
|
rerank: bool = False,
|
||||||
):
|
):
|
||||||
"""
|
"""
|
||||||
Searches for memories based on a query
|
Searches for memories based on a query
|
||||||
@@ -2077,7 +2077,7 @@ class AsyncMemory(MemoryBase):
|
|||||||
filters: Optional[Dict[str, Any]] = None,
|
filters: Optional[Dict[str, Any]] = None,
|
||||||
threshold: float = 0.1,
|
threshold: float = 0.1,
|
||||||
metadata_filters: Optional[Dict[str, Any]] = None,
|
metadata_filters: Optional[Dict[str, Any]] = None,
|
||||||
rerank: bool = True,
|
rerank: bool = False,
|
||||||
):
|
):
|
||||||
"""
|
"""
|
||||||
Searches for memories based on a query
|
Searches for memories based on a query
|
||||||
|
|||||||
Reference in New Issue
Block a user