small bug fixes for inline attachments

2025-10-04 12:07:34 +00:00 · 2024-08-24 23:51:27 -07:00 · 2024-08-24 23:51:27 -07:00 · 440d125ea0
commit 440d125ea0
parent 58e2feceb0
2 changed files with 32 additions and 11 deletions
--- a/llama_toolchain/agentic_system/meta_reference/agent_instance.py
+++ b/llama_toolchain/agentic_system/meta_reference/agent_instance.py
@ -580,6 +580,7 @@ class ChatAgent(ShieldRunnerMixin):
                name=f"memory_bank_{session.session_id}",
                config=VectorMemoryBankConfig(
                    embedding_model="sentence-transformer/all-MiniLM-L6-v2",
+                    chunk_size_in_tokens=512,
                ),
            )

@ -619,7 +620,7 @@ class ChatAgent(ShieldRunnerMixin):

            documents = [
                MemoryBankDocument(
-                    doc_id=str(uuid.uuid4()),
+                    document_id=str(uuid.uuid4()),
                    content=a.content,
                    mime_type=a.mime_type,
                    metadata={},