NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-13 15:14:50 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: Remove excessive db call to update state table.

Browse Source 
PiperOrigin-RevId: 764375522
This commit is contained in:
Shangjie Chen 2025-05-28 12:45:41 -07:00 committed by Copybara-Service
parent 3611e9fe09
commit d83ee1a6f2
1 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/google/adk/sessions/database_session_service.py
View File

@ -11,6 +11,8 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from __future__ import annotations
import copy
from datetime import datetime
import json
@ -512,15 +514,16 @@ class DatabaseSessionService(BaseSessionService):
_extract_state_delta(event.actions.state_delta)
)
# Merge state
app_state.update(app_state_delta)
user_state.update(user_state_delta)
session_state.update(session_state_delta)
# Update storage
storage_app_state.state = app_state
storage_user_state.state = user_state
storage_session.state = session_state
# Merge state and update storage
if app_state_delta:
app_state.update(app_state_delta)
storage_app_state.state = app_state
if user_state_delta:
user_state.update(user_state_delta)
storage_user_state.state = user_state
if session_state_delta:
session_state.update(session_state_delta)
storage_session.state = session_state
storage_event = StorageEvent(
id=event.id,