chore: remove legacy embeddings (#3846)

This commit is contained in:
Sarah Wooders
2025-08-12 15:11:09 -07:00
committed by GitHub
parent ad52c7001c
commit f4740b1388
14 changed files with 210 additions and 533 deletions

View File

@@ -63,7 +63,7 @@ def conversation_search(self: "Agent", query: str, page: Optional[int] = 0) -> O
return results_str
def archival_memory_insert(self: "Agent", content: str) -> Optional[str]:
async def archival_memory_insert(self: "Agent", content: str) -> Optional[str]:
"""
Add to archival memory. Make sure to phrase the memory contents such that it can be easily queried later.
@@ -73,7 +73,7 @@ def archival_memory_insert(self: "Agent", content: str) -> Optional[str]:
Returns:
Optional[str]: None is always returned as this function does not produce a response.
"""
self.passage_manager.insert_passage(
await self.passage_manager.insert_passage(
agent_state=self.agent_state,
text=content,
actor=self.user,
@@ -82,7 +82,7 @@ def archival_memory_insert(self: "Agent", content: str) -> Optional[str]:
return None
def archival_memory_search(self: "Agent", query: str, page: Optional[int] = 0, start: Optional[int] = 0) -> Optional[str]:
async def archival_memory_search(self: "Agent", query: str, page: Optional[int] = 0, start: Optional[int] = 0) -> Optional[str]:
"""
Search archival memory using semantic (embedding-based) search.
@@ -107,7 +107,7 @@ def archival_memory_search(self: "Agent", query: str, page: Optional[int] = 0, s
try:
# Get results using passage manager
all_results = self.agent_manager.list_passages(
all_results = await self.agent_manager.list_passages_async(
actor=self.user,
agent_id=self.agent_state.id,
query_text=query,