Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

feat: disable session reuse after close (#2547)

Browse Source
This commit is contained in:
cthomas
2025-05-30 12:51:19 -07:00
committed by GitHub
parent fa9b27b191
commit f5aa97f02b
2 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
letta/server/db.py
View File

@@ -131,7 +131,7 @@ class DatabaseRegistry:
# Create async session factory
self._async_engines["default"] = async_engine
self._async_session_factories["default"] = async_sessionmaker(
autocommit=False, autoflush=False, bind=self._async_engines["default"], class_=AsyncSession
close_resets_only=False, autocommit=False, autoflush=False, bind=self._async_engines["default"], class_=AsyncSession
)
self._initialized["async"] = True

26
letta/services/identity_manager.py
View File

@@ -93,19 +93,19 @@ class IdentityManager:
actor=actor,
)
if existing_identity is None:
return await self.create_identity_async(identity=IdentityCreate(**identity.model_dump()), actor=actor)
else:
identity_update = IdentityUpdate(
name=identity.name,
identifier_key=identity.identifier_key,
identity_type=identity.identity_type,
agent_ids=identity.agent_ids,
properties=identity.properties,
)
return await self._update_identity_async(
session=session, existing_identity=existing_identity, identity=identity_update, actor=actor, replace=True
)
if existing_identity is None:
return await self.create_identity_async(identity=IdentityCreate(**identity.model_dump()), actor=actor)
else:
identity_update = IdentityUpdate(
name=identity.name,
identifier_key=identity.identifier_key,
identity_type=identity.identity_type,
agent_ids=identity.agent_ids,
properties=identity.properties,
)
return await self._update_identity_async(
session=session, existing_identity=existing_identity, identity=identity_update, actor=actor, replace=True
)
@enforce_types
@trace_method