From 564f0e5f936e202d8771adc4978ff5c2a06bc254 Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Thu, 27 Feb 2025 10:37:15 -0800
Subject: [PATCH] fix: Revert "chore: remove vector_db_id from
 AgentSessionInfo" (#1299)

Reverts meta-llama/llama-stack#1296

This change breaks test: `session_info.vector_db_id` is actually used
```
pytest -v tests/client-sdk/agents/test_agents.py::test_rag_and_code_agent --inference-model meta-llama/Llama-3.1-8B-Instruct
```
---
 .../providers/inline/agents/meta_reference/persistence.py       | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/llama_stack/providers/inline/agents/meta_reference/persistence.py b/llama_stack/providers/inline/agents/meta_reference/persistence.py
index 9abb7413c..9a2663567 100644
--- a/llama_stack/providers/inline/agents/meta_reference/persistence.py
+++ b/llama_stack/providers/inline/agents/meta_reference/persistence.py
@@ -21,6 +21,8 @@ log = logging.getLogger(__name__)
 class AgentSessionInfo(BaseModel):
     session_id: str
     session_name: str
+    # TODO: is this used anywhere?
+    vector_db_id: Optional[str] = None
     started_at: datetime