letta-nightly 0.6.1.dev20241206104246__py3-none-any.whl → 0.6.1.dev20241207104149__py3-none-any.whl

letta/services/organization_manager.py

@@ -30,19 +30,16 @@ class OrganizationManager:
     def get_organization_by_id(self, org_id: str) -> Optional[PydanticOrganization]:
         """Fetch an organization by ID."""
         with self.session_maker() as session:
-            try:
-                organization = OrganizationModel.read(db_session=session, identifier=org_id)
-                return organization.to_pydantic()
-            except NoResultFound:
-                return None
+            organization = OrganizationModel.read(db_session=session, identifier=org_id)
+            return organization.to_pydantic()
 
     @enforce_types
     def create_organization(self, pydantic_org: PydanticOrganization) -> PydanticOrganization:
-        """Create a new organization. If a name is provided, it is used, otherwise, a random one is generated."""
-        org = self.get_organization_by_id(pydantic_org.id)
-        if org:
+        """Create a new organization."""
+        try:
+            org = self.get_organization_by_id(pydantic_org.id)
             return org
-        else:
+        except NoResultFound:
             return self._create_organization(pydantic_org=pydantic_org)
 
     @enforce_types

letta/services/source_manager.py

@@ -141,5 +141,5 @@ class SourceManager:
         """Delete a file by its ID."""
         with self.session_maker() as session:
             file = FileMetadataModel.read(db_session=session, identifier=file_id)
-            file.delete(db_session=session, actor=actor)
+            file.hard_delete(db_session=session, actor=actor)
             return file.to_pydantic()

letta/services/tool_manager.py

@@ -122,7 +122,7 @@ class ToolManager:
                 tool.json_schema = new_schema
 
             # Save the updated tool to the database
-            return tool.update(db_session=session, actor=actor)
+            return tool.update(db_session=session, actor=actor).to_pydantic()
 
     @enforce_types
     def delete_tool_by_id(self, tool_id: str, actor: PydanticUser) -> None:

letta/services/user_manager.py

@@ -71,7 +71,7 @@ class UserManager:
         with self.session_maker() as session:
             # Delete from user table
             user = UserModel.read(db_session=session, identifier=user_id)
-            user.delete(session)
+            user.hard_delete(session)
 
             # TODO: Integrate this via the ORM models for the Agent, Source, and AgentSourceMapping
             # Cascade delete for related models: Agent, Source, AgentSourceMapping

{letta_nightly-0.6.1.dev20241206104246.dist-info → letta_nightly-0.6.1.dev20241207104149.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: letta-nightly
-Version: 0.6.1.dev20241206104246
+Version: 0.6.1.dev20241207104149
 Summary: Create LLM agents with long-term memory and custom tools
 License: Apache License
 Author: Letta Team

{letta_nightly-0.6.1.dev20241206104246.dist-info → letta_nightly-0.6.1.dev20241207104149.dist-info}/RECORD

@@ -1,31 +1,30 @@
 letta/__init__.py,sha256=1hJidwYqfS-iX2V3ZLC2ulm-PYX74k2EMIyzC4Z0pWc,1035
 letta/__main__.py,sha256=6Hs2PV7EYc5Tid4g4OtcLXhqVHiNYTGzSBdoOnW2HXA,29
-letta/agent.py,sha256=r80lyWFX7wjPM5gPcUOxifa4ulO_OfW3M1p0vQbmxYk,78428
+letta/agent.py,sha256=Kr55vl085tf-FvSWBPD-1SZmOn2xaCd_tLtoV8SvDuU,79198
 letta/agent_store/chroma.py,sha256=-kCEMBFKmqCyFeIETYf7RN-khGddsip2FAhSzNqaC7U,12537
-letta/agent_store/db.py,sha256=n15t8qhHfqhtFDxSQg_9uwvMntpWml8Jz_Y-ofL0loQ,23467
-letta/agent_store/lancedb.py,sha256=i63d4VZwj9UIOTNs5f0JZ_r5yZD-jKWz4FAH4RMpXOE,5104
+letta/agent_store/db.py,sha256=0AdN1-LpvIUhmiayTwzhKPaXTdlyOILtHGZ7zmS7pPs,20314
 letta/agent_store/milvus.py,sha256=xUu-D9a6N10MuGJ-R-QWR2IHX77ueqAp88tV4gg9B4M,8470
 letta/agent_store/qdrant.py,sha256=6_33V-FEDpT9LG5zmr6-3y9slw1YFLswxpahiyMkvHA,7880
-letta/agent_store/storage.py,sha256=4gKvMRYBGm9cwyaDOzljxDKgqr4MxGXcC4yGhAdKcAA,6693
+letta/agent_store/storage.py,sha256=YDQTrfgPBa0cMiMuQatk9hcadIUMvA4zKZHx2VKjlak,6475
 letta/benchmark/benchmark.py,sha256=ebvnwfp3yezaXOQyGXkYCDYpsmre-b9hvNtnyx4xkG0,3701
 letta/benchmark/constants.py,sha256=aXc5gdpMGJT327VuxsT5FngbCK2J41PQYeICBO7g_RE,536
 letta/chat_only_agent.py,sha256=606FQybmeN9s04rIFQlDkWOHirrT0p48_3pMKY8d5Ts,4740
-letta/cli/cli.py,sha256=fpcpBKEAzKU6o9gSS5pe6YRTkiybIik5CC9mCAVl_bA,16928
+letta/cli/cli.py,sha256=7DI6fHLCTgotP4jr6x4D07EaOjayi1vH-SZj9KiCOQQ,16834
 letta/cli/cli_config.py,sha256=tB0Wgz3O9j6KiCsU1HWfsKmhNM9RqLsAxzxEDFQFGnM,8565
 letta/cli/cli_load.py,sha256=x4L8s15GwIW13xrhKYFWHo_y-IVGtoPDHWWKcHDRP10,4587
 letta/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-letta/client/client.py,sha256=hQiTmkmddoGONuSlNG0Yrbo1Y2k9VRTRDFAOtHRbOPI,124143
+letta/client/client.py,sha256=Oetmz1FwIs0n9bCRmzlPqxIjQK3zKrraQ2t3gq_pNg4,124015
 letta/client/streaming.py,sha256=Hh5pjlyrdCuO2V75ZCxSSOCPd3BmHdKFGaIUJC6fBp0,4775
 letta/client/utils.py,sha256=OJlAKWrldc4I6M1WpcTWNtPJ4wfxlzlZqWLfCozkFtI,2872
 letta/config.py,sha256=AF4XY6grcu87OLjrWXh1ufnyKWsCL0qER-_9jQCAlU0,18947
-letta/constants.py,sha256=C85Bktu1fhj83LYG6fRcWlw-9DtidTEkbSFyaJ4leOU,6838
+letta/constants.py,sha256=l2IyZpVG-d_raFFX5bN6mCbprnnc1IA-z-oCKPoyPSM,6902
 letta/credentials.py,sha256=D9mlcPsdDWlIIXQQD8wSPE9M_QvsRrb0p3LB5i9OF5Q,5806
 letta/data_sources/connectors.py,sha256=5VKxfeV-QyUlK1wexLlpgar99dGm6PHxFaEbSeByo_U,9923
 letta/data_sources/connectors_helper.py,sha256=2TQjCt74fCgT5sw1AP8PalDEk06jPBbhrPG4HVr-WLs,3371
 letta/embeddings.py,sha256=3vJaQ8RMLLp6yiYZGsthq6Xsu4keb9gp7DYN_2GPBFU,8459
 letta/errors.py,sha256=mFeTpZP37otDMr68s9hyGOnafJPrWeblQOI79cgP4nQ,3209
 letta/functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-letta/functions/function_sets/base.py,sha256=pTLpCdJSTts7i9QV6tBphyBonhDDEbGRy692PzjqO2g,8104
+letta/functions/function_sets/base.py,sha256=anK6XJQlmTu9eC33700Eh7NWxQ0NTLUC2Vxpn6ihvnY,9034
 letta/functions/function_sets/extras.py,sha256=Jik3UiDqYTm4Lam1XPTvuVjvgUHwIAhopsnbmVhGMBg,4732
 letta/functions/functions.py,sha256=evH6GKnIJwVVre1Xre2gaSIqREv4eNM4DiWOhn8PMqg,3299
 letta/functions/helpers.py,sha256=K84kqAN1RXZIhjb7-btS0C2p-SInYNv6FvSfo-16Y6g,8578
@@ -84,32 +83,32 @@ letta/local_llm/webui/legacy_api.py,sha256=k3H3y4qp2Fs-XmP24iSIEyvq6wjWFWBzklY3-
 letta/local_llm/webui/legacy_settings.py,sha256=BLmd3TSx5StnY3ibjwaxYATPt_Lvq-o1rlcc_-Q1JcU,538
 letta/local_llm/webui/settings.py,sha256=gmLHfiOl1u4JmlAZU2d2O8YKF9lafdakyjwR_ftVPh8,552
 letta/log.py,sha256=FxkAk2f8Bl-u9dfImSj1DYnjAsmV6PL3tjTSnEiNP48,2218
-letta/main.py,sha256=5guUzYyxID3FDlegk3dNUev7vjPMglcIw-xqdyHdhKI,19175
-letta/memory.py,sha256=YupXOvzVJXH59RW4XWBrd7qMNEYaMbtWXCheKeWZwpU,17873
-letta/metadata.py,sha256=jv_s2Y6M8aE3bpOOTr-xwBBu58zZHx9tYtAekVa_Dz4,16233
-letta/o1_agent.py,sha256=jTMlP7LxR4iUDWaGHMy8SiZtlzn6_RqP0H1HaEWXydQ,3078
-letta/offline_memory_agent.py,sha256=loCrbMAvngqVbyng7Qu1Bg-k_RtQL-Lea1QKbeK0vm8,7724
+letta/main.py,sha256=M78MU-12bj-6qyoWz-yNIRsj8qMmwCZB4Ed8LTFfKrc,19135
+letta/memory.py,sha256=R4Z_MYD5ve04FBe2yohux1pzp8MTjOSI1AZ5oLCdGRA,14568
+letta/metadata.py,sha256=zwIe0Sr1Q2g5L2V6Yc5D_t_NG8nhdgleAGQEdaAi2Lw,15165
+letta/o1_agent.py,sha256=aNuv70NmmtKQcfOcd0L4VxHzg7o7xoiU81j1LWok4bk,3164
+letta/offline_memory_agent.py,sha256=369jMxqKMoex9JI269kpaO1Q_JPWLwn-BaiZ6YHc-Fw,7898
 letta/openai_backcompat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 letta/openai_backcompat/openai_object.py,sha256=Y1ZS1sATP60qxJiOsjOP3NbwSzuzvkNAvb3DeuhM5Uk,13490
 letta/orm/__all__.py,sha256=2gh2MZTkA3Hw67VWVKK3JIStJOqTeLdpCvYSVYNeEDA,692
-letta/orm/__init__.py,sha256=bV55IErRhxwPvHHRkIx8DJ7K-ckVBnwavygYNMc4sMQ,459
+letta/orm/__init__.py,sha256=HyH2suMD8-92bZ5paW8AC1o-639AQqwAc_HN8S7mjxI,542
 letta/orm/agents_tags.py,sha256=Qa7Yt9imL8xbGP57fflccAMy7Z32CQiU_7eZKSSPngc,1119
 letta/orm/base.py,sha256=K_LpNUURbsj44ycHbzvNXG_n8pBOjf1YvDaikIPDpQA,2716
 letta/orm/block.py,sha256=xymYeCTJJFkzADW6wjfP2LXNZZN9yg4mCSybbvEEMMM,2356
 letta/orm/blocks_agents.py,sha256=o6cfblODja7so4444npW0vusqKcvDPp8YJdsWsOePus,1164
 letta/orm/enums.py,sha256=KfHcFt_fR6GUmSlmfsa-TetvmuRxGESNve8MStRYW64,145
 letta/orm/errors.py,sha256=nv1HnF3z4-u9m_g7SO5TO5u2nmJN677_n8F0iIjluUI,460
-letta/orm/file.py,sha256=FtfZlJLXfac4ntaw3kC0N9VRoD255m8EK4p-pC2lcHk,1519
+letta/orm/file.py,sha256=JafpJhooBtMrmkRG68jT8rFK9mRIQSlYKYjq3VXhTcc,1525
 letta/orm/job.py,sha256=If-qSTJW4t5h-6Jolw3tS3-xMZEaPIbXe3S0GMf_FXI,1102
-letta/orm/mixins.py,sha256=LfwePamGyOwCtAEUm-sZpIBJjODIMe4MnA_JTUcppLs,1155
-letta/orm/organization.py,sha256=mliw4q7-SRfRcGIG8paNfCNn6ITTjR7nFalZzlRszqU,2272
+letta/orm/message.py,sha256=8x6zDFlnYqTL27JDaVcaHtIOkvTGVg_FxesSmOvH0kI,2677
+letta/orm/mixins.py,sha256=TpP5dFAcGu2VvHZQSt8ymc0UjQhcVp5-Fc2C4v_nUTM,1508
+letta/orm/organization.py,sha256=ZoHow3Tpdh9TXDvB4VZbArbUCgbErFTSQMriYAzulqA,2397
 letta/orm/sandbox_config.py,sha256=PCMHE-eJPzBT-90OYtXjEMRF4f9JB8AJIGETE7bu-f0,2870
 letta/orm/source.py,sha256=Ib0XHCMt345RjBSC30A398rZ21W5mA4PXX00XNXyd24,2021
-letta/orm/sqlalchemy_base.py,sha256=CmCwVCHVyhzDAd-xIUFh72bzSX-bA0mwvMUD7FrIlUk,10672
+letta/orm/sqlalchemy_base.py,sha256=_uix_6bCC_rZYbTKL_OpldjTgrEv-9l6FL3ysKsgUZU,13877
 letta/orm/tool.py,sha256=R1I3FGjt8xlW7DPkwV9zlliOcAB3vpphAZTQTsvDBXs,2982
 letta/orm/tools_agents.py,sha256=mBMGQsTEx_ckfhZb2-2nqbxHBEMhvDXim6w6tFHHWBQ,1195
 letta/orm/user.py,sha256=bUZzyBQXfR0w7mZAkThPAQE6kKx6W--Rw6uAiPEUW3s,1133
-letta/persistence_manager.py,sha256=sEnNdNisK7StqIzG8eIY0YMag6S9hZFbkDfmY7L2Ikc,5268
 letta/personas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 letta/personas/examples/anna_pa.txt,sha256=zgiNdSNhy1HQy58cF_6RFPzcg2i37F9v38YuL1CW40A,1849
 letta/personas/examples/google_search_persona.txt,sha256=RyObU80MIk2oeJJDWOK1aX5pHOtbHSSjIrbUpxov240,1194
@@ -148,13 +147,13 @@ letta/schemas/enums.py,sha256=F396hXs57up4Jqj1vwWVknMpoVo7MkccVBALvKGHPpE,1032
 letta/schemas/file.py,sha256=ChN2CWzLI2TT9WLItcfElEH0E8b7kzPylF2OQBr6Beg,1550
 letta/schemas/health.py,sha256=zT6mYovvD17iJRuu2rcaQQzbEEYrkwvAE9TB7iU824c,139
 letta/schemas/job.py,sha256=vRVVpMCHTxot9uaalLS8RARnqzJWvcLB1XP5XRBioPc,1398
-letta/schemas/letta_base.py,sha256=QlCY5BBSjEPNpUvSHDl_P0TXQIPlr7xfhK_2SqKqadQ,3567
+letta/schemas/letta_base.py,sha256=v3OvpVFVUfcuK1lIyTjM0x4fmWeWQw1DdlhKXHBUCz8,3608
 letta/schemas/letta_message.py,sha256=RuVVtwFbi85yP3dXQxowofQ6cI2cO_CdGtgpHGQzgHc,6563
 letta/schemas/letta_request.py,sha256=E6OwKiceNffpdGdQMI1qc0jEfpL_e7O9BTzklOkbt6Y,1019
 letta/schemas/letta_response.py,sha256=vQV5uqe-kq9fc4wCo-sVB_PJt5yUk8DB2zOgHsrmN-k,6189
 letta/schemas/llm_config.py,sha256=RbgnCaqYd_yl-Xs7t-DEI1NhpKD8WiVWjxcwq5mZd5M,4467
 letta/schemas/memory.py,sha256=80Y7gqWQQndhNkJ-0j38d2m619gTlfes_qJNA6_ant8,10040
-letta/schemas/message.py,sha256=oInRiw5x8gsxX5bylO_g29v8f-yBC-O1oQurluvpJHs,33691
+letta/schemas/message.py,sha256=KER04veCt-7FiZj3GU1x_QW38xnBC7jnnBDkRJuW3ak,33485
 letta/schemas/openai/chat_completion_request.py,sha256=AOIwgbN3CZKVqkuXeMHeSa53u4h0wVq69t3T_LJ0vIE,3389
 letta/schemas/openai/chat_completion_response.py,sha256=ub-oVSyLpuJd-5_yzCSIRR8tD3GM83IeDO1c1uAATa4,3970
 letta/schemas/openai/chat_completions.py,sha256=V0ZPIIk-ds3O6MAkNHMz8zh1hqMFSPrTcYr88WDYzWE,3588
@@ -187,8 +186,8 @@ letta/server/rest_api/routers/openai/assistants/threads.py,sha256=g8iu98__tQEMY9
 letta/server/rest_api/routers/openai/chat_completions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 letta/server/rest_api/routers/openai/chat_completions/chat_completions.py,sha256=qFMpxfYIlJ-PW08IQt09RW44u6hwkssdsUT-h_GuOvE,4836
 letta/server/rest_api/routers/v1/__init__.py,sha256=RZc0fIHNN4BGretjU6_TGK7q49RyV4jfYNudoiK_sUo,762
-letta/server/rest_api/routers/v1/agents.py,sha256=Tj7QyHjem_tOgzDzTyEJREDH2rzDgpE6S5azBDrhY_o,24884
-letta/server/rest_api/routers/v1/blocks.py,sha256=UCVfMbb8hzOXI6a8OYWKuXyOropIxw6PYKZkwwAh1v0,4880
+letta/server/rest_api/routers/v1/agents.py,sha256=h6i_HJZ0kcM4p2cTZeh0Mi65PhBw08uzDDxsQkFRLt4,24891
+letta/server/rest_api/routers/v1/blocks.py,sha256=hLGwm56xCA09KdEi8P0V3s__pD_yyl07p17cdSfDj8g,4878
 letta/server/rest_api/routers/v1/health.py,sha256=pKCuVESlVOhGIb4VC4K-H82eZqfghmT6kvj2iOkkKuc,401
 letta/server/rest_api/routers/v1/jobs.py,sha256=gnu__rjcd9SpKdQkSD_sci-xJYH0fw828PuHMcYwsCw,2678
 letta/server/rest_api/routers/v1/llms.py,sha256=TcyvSx6MEM3je5F4DysL7ligmssL_pFlJaaO4uL95VY,877
@@ -199,12 +198,12 @@ letta/server/rest_api/routers/v1/tools.py,sha256=TP16cpuTF2HYLFZVmabExw9gziB-Ptk
 letta/server/rest_api/routers/v1/users.py,sha256=M1wEr2IyHzuRwINYxLXTkrbAH3osLe_cWjzrWrzR1aw,3729
 letta/server/rest_api/static_files.py,sha256=NG8sN4Z5EJ8JVQdj19tkFa9iQ1kBPTab9f_CUxd_u4Q,3143
 letta/server/rest_api/utils.py,sha256=6c5a_-ZFTlwZ1IuzpRQtqxSG1eD56nNhKhWlrdgBYWk,3103
-letta/server/server.py,sha256=6arPhzclk47FL-pRkutnIwhr2U1i-gQOLBUV5p6jVJ4,81783
+letta/server/server.py,sha256=jX1KQsSoLY9gtKFSwskxZjx7lduuQ0YLTovriKnM-9Q,81561
 letta/server/startup.sh,sha256=wTOQOJJZw_Iec57WIu0UW0AVflk0ZMWYZWg8D3T_gSQ,698
-letta/server/static_files/assets/index-1b5d1a41.js,sha256=bUkyBXaNVT-Re4gNnQVGRQuxoXXtwizbJ75OryRAdtY,1816165
-letta/server/static_files/assets/index-56a3f8c6.css,sha256=VqP4xqJcLheEdnyJFM_gdBTqh4WtLriD4gvnwCrN43g,54450
+letta/server/static_files/assets/index-43ab4d62.css,sha256=Q6tNYu7XNRgkTZau1dVwCQTT4YpLGLF7VgTNzLcMaQI,7613
+letta/server/static_files/assets/index-4848e3d7.js,sha256=XkoX-NW0x1Hb7uTWslfpuPP73MnNDthOdJLHLID9EhM,146806
 letta/server/static_files/favicon.ico,sha256=DezhLdFSbM8o81wCOZcV3riq7tFUOGQD4h6-vr-HuU0,342
-letta/server/static_files/index.html,sha256=rwxD6lUXCAGI3Hdh5tNv1j7Vq_llUlgAUyC_CRZ3m9U,1198
+letta/server/static_files/index.html,sha256=RF-Kt4CbTPJbsHtK30NaWIsdUiBpmJgQsKmE6WwqsBk,1198
 letta/server/static_files/memgpt_logo_transparent.png,sha256=7l6niNb4MlUILxLlUZPxIE1TEHj_Z9f9XDxoST3d7Vw,85383
 letta/server/utils.py,sha256=rRvW6L1lzau4u9boamiyZH54lf5tQ91ypXzUW9cfSPA,1667
 letta/server/ws_api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -214,25 +213,26 @@ letta/server/ws_api/protocol.py,sha256=M_-gM5iuDBwa1cuN2IGNCG5GxMJwU2d3XW93XALv9
 letta/server/ws_api/server.py,sha256=C2Kv48PCwl46DQFb0ZP30s86KJLQ6dZk2AhWQEZn9pY,6004
 letta/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 letta/services/agents_tags_manager.py,sha256=zNqeXDpaf4dQ77jrRHiQfITdk4FawBzcND-9tWrj8gw,3127
-letta/services/block_manager.py,sha256=TrbStwHAREwnybA6jZSkNPe-EYUa5rdiuliPR2PTV-M,5426
+letta/services/block_manager.py,sha256=BmuK0k3yl9byT703syGtZNfjUdM2FkOwlE6bQYIQtjI,5448
 letta/services/blocks_agents_manager.py,sha256=mfO3EMW9os_E1_r4SRlC2wmBFFLpt8p-yhdOH_Iotaw,5627
 letta/services/job_manager.py,sha256=FrkSXloke48CZKuzlYdysxM5gKWoTu7FRigPrs_YW4A,3645
-letta/services/organization_manager.py,sha256=OfE2_NMmhqXURX4sg7hCOiFQVQpV5ZiPu7J3sboCSYc,3555
+letta/services/message_manager.py,sha256=pvRaV2GoW5EzRas5j2NSF75O_sTaHVUWwmPzTbHcfHw,6874
+letta/services/organization_manager.py,sha256=B7BgHkZcAOP1fzbg2fFF8eTfKmqOiGvoojQ8ys7JVY4,3412
 letta/services/per_agent_lock_manager.py,sha256=porM0cKKANQ1FvcGXOO_qM7ARk5Fgi1HVEAhXsAg9-4,546
 letta/services/sandbox_config_manager.py,sha256=9BCu59nHR4nIMFXgFyEMOY2UTmZvBMS3GlDBWWCHB4I,12648
-letta/services/source_manager.py,sha256=StX5Wfd7XSCKJet8qExIu3GMoI-eMIbEarAeTv2gq0s,6555
+letta/services/source_manager.py,sha256=8TILbVvJx1bopgDbIJCiIguZBw7uyYDFkasZxVos_G8,6560
 letta/services/tool_execution_sandbox.py,sha256=GTWdfAKIMIuODEFbmReyEYkOnE62uzDF-3FHWee1s3A,21295
-letta/services/tool_manager.py,sha256=7c6naDlI8GEvDs4n1LXMVeBn9QdYe4Wy31O7jv_f5Pc,7840
+letta/services/tool_manager.py,sha256=BFwD9J6WOIXnYV6QmJFeCKRIiAiD0vO0haNfshkn2jI,7854
 letta/services/tool_sandbox_env/.gitkeep,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 letta/services/tools_agents_manager.py,sha256=kF6yIsUO154_Q3l-cEeSU56QW-VUr_M6B-9csqhHBkg,4847
-letta/services/user_manager.py,sha256=UJa0hqCjz0yXtvrCR8OVBqlSR5lC_Ejn-uG__58zLds,4398
+letta/services/user_manager.py,sha256=cfXgnVlnfqPCk7c-wYRsU3RegHs-4stksk186RW89Do,4403
 letta/settings.py,sha256=ZcUcwvl7hStawZ0JOA0133jNk3j5qBd7qlFAAaIPsU8,3608
 letta/streaming_interface.py,sha256=_FPUWy58j50evHcpXyd7zB1wWqeCc71NCFeWh_TBvnw,15736
 letta/streaming_utils.py,sha256=329fsvj1ZN0r0LpQtmMPZ2vSxkDBIUUwvGHZFkjm2I8,11745
 letta/system.py,sha256=buKYPqG5n2x41hVmWpu6JUpyd7vTWED9Km2_M7dLrvk,6960
 letta/utils.py,sha256=7FGZYp8JCEtP0PBSfV03Zj4kW3pwV7qpNki4AOU3a94,32913
-letta_nightly-0.6.1.dev20241206104246.dist-info/LICENSE,sha256=mExtuZ_GYJgDEI38GWdiEYZizZS4KkVt2SF1g_GPNhI,10759
-letta_nightly-0.6.1.dev20241206104246.dist-info/METADATA,sha256=LZrJS9Y9Ewunwnq8ZX0OA4ZC1FPeJdfXMXGVX3Msl7g,11459
-letta_nightly-0.6.1.dev20241206104246.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-letta_nightly-0.6.1.dev20241206104246.dist-info/entry_points.txt,sha256=2zdiyGNEZGV5oYBuS-y2nAAgjDgcC9yM_mHJBFSRt5U,40
-letta_nightly-0.6.1.dev20241206104246.dist-info/RECORD,,
+letta_nightly-0.6.1.dev20241207104149.dist-info/LICENSE,sha256=mExtuZ_GYJgDEI38GWdiEYZizZS4KkVt2SF1g_GPNhI,10759
+letta_nightly-0.6.1.dev20241207104149.dist-info/METADATA,sha256=65Tw5JljqWDW7dJ2Qm-ZS-4hfPopKW9WAmstmTFOv3I,11459
+letta_nightly-0.6.1.dev20241207104149.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+letta_nightly-0.6.1.dev20241207104149.dist-info/entry_points.txt,sha256=2zdiyGNEZGV5oYBuS-y2nAAgjDgcC9yM_mHJBFSRt5U,40
+letta_nightly-0.6.1.dev20241207104149.dist-info/RECORD,,

letta/agent_store/lancedb.py

@@ -1,177 +0,0 @@
-# type: ignore
-
-import uuid
-from datetime import datetime
-from typing import Dict, Iterator, List, Optional
-
-from lancedb.pydantic import LanceModel, Vector
-
-from letta.agent_store.storage import StorageConnector, TableType
-from letta.config import AgentConfig, LettaConfig
-from letta.schemas.message import Message, Passage, Record
-
-""" Initial implementation - not complete """
-
-
-def get_db_model(table_name: str, table_type: TableType):
-    config = LettaConfig.load()
-
-    if table_type == TableType.ARCHIVAL_MEMORY or table_type == TableType.PASSAGES:
-        # create schema for archival memory
-        class PassageModel(LanceModel):
-            """Defines data model for storing Passages (consisting of text, embedding)"""
-
-            id: uuid.UUID
-            user_id: str
-            text: str
-            file_id: str
-            agent_id: str
-            data_source: str
-            embedding: Vector(config.default_embedding_config.embedding_dim)
-            metadata_: Dict
-
-            def __repr__(self):
-                return f"<Passage(passage_id='{self.id}', text='{self.text}', embedding='{self.embedding})>"
-
-            def to_record(self):
-                return Passage(
-                    text=self.text,
-                    embedding=self.embedding,
-                    file_id=self.file_id,
-                    user_id=self.user_id,
-                    id=self.id,
-                    data_source=self.data_source,
-                    agent_id=self.agent_id,
-                    metadata=self.metadata_,
-                )
-
-        return PassageModel
-    elif table_type == TableType.RECALL_MEMORY:
-
-        class MessageModel(LanceModel):
-            """Defines data model for storing Message objects"""
-
-            __abstract__ = True  # this line is necessary
-
-            # Assuming message_id is the primary key
-            id: uuid.UUID
-            user_id: str
-            agent_id: str
-
-            # openai info
-            role: str
-            name: str
-            text: str
-            model: str
-            user: str
-
-            # function info
-            function_name: str
-            function_args: str
-            function_response: str
-
-            embedding = Vector(config.default_embedding_config.embedding_dim)
-
-            # Add a datetime column, with default value as the current time
-            created_at = datetime
-
-            def __repr__(self):
-                return f"<Message(message_id='{self.id}', text='{self.text}', embedding='{self.embedding})>"
-
-            def to_record(self):
-                return Message(
-                    user_id=self.user_id,
-                    agent_id=self.agent_id,
-                    role=self.role,
-                    name=self.name,
-                    text=self.text,
-                    model=self.model,
-                    function_name=self.function_name,
-                    function_args=self.function_args,
-                    function_response=self.function_response,
-                    embedding=self.embedding,
-                    created_at=self.created_at,
-                    id=self.id,
-                )
-
-        """Create database model for table_name"""
-        return MessageModel
-
-    else:
-        raise ValueError(f"Table type {table_type} not implemented")
-
-
-class LanceDBConnector(StorageConnector):
-    """Storage via LanceDB"""
-
-    # TODO: this should probably eventually be moved into a parent DB class
-
-    def __init__(self, name: Optional[str] = None, agent_config: Optional[AgentConfig] = None):
-        # TODO
-        pass
-
-    def generate_where_filter(self, filters: Dict) -> str:
-        where_filters = []
-        for key, value in filters.items():
-            where_filters.append(f"{key}={value}")
-        return where_filters.join(" AND ")
-
-    @abstractmethod
-    def get_all_paginated(self, filters: Optional[Dict] = {}, page_size: Optional[int] = 1000) -> Iterator[List[Record]]:
-        # TODO
-        pass
-
-    @abstractmethod
-    def get_all(self, filters: Optional[Dict] = {}, limit=10) -> List[Record]:
-        # TODO
-        pass
-
-    @abstractmethod
-    def get(self, id: uuid.UUID) -> Optional[Record]:
-        # TODO
-        pass
-
-    @abstractmethod
-    def size(self, filters: Optional[Dict] = {}) -> int:
-        # TODO
-        pass
-
-    @abstractmethod
-    def insert(self, record: Record):
-        # TODO
-        pass
-
-    @abstractmethod
-    def insert_many(self, records: List[Record], show_progress=False):
-        # TODO
-        pass
-
-    @abstractmethod
-    def query(self, query: str, query_vec: List[float], top_k: int = 10, filters: Optional[Dict] = {}) -> List[Record]:
-        # TODO
-        pass
-
-    @abstractmethod
-    def query_date(self, start_date, end_date):
-        # TODO
-        pass
-
-    @abstractmethod
-    def query_text(self, query):
-        # TODO
-        pass
-
-    @abstractmethod
-    def delete_table(self):
-        # TODO
-        pass
-
-    @abstractmethod
-    def delete(self, filters: Optional[Dict] = {}):
-        # TODO
-        pass
-
-    @abstractmethod
-    def save(self):
-        # TODO
-        pass

letta/persistence_manager.py

@@ -1,149 +0,0 @@
-from abc import ABC, abstractmethod
-from datetime import datetime
-from typing import List
-
-from letta.memory import BaseRecallMemory, EmbeddingArchivalMemory
-from letta.schemas.agent import AgentState
-from letta.schemas.memory import Memory
-from letta.schemas.message import Message
-from letta.utils import printd
-
-
-def parse_formatted_time(formatted_time: str):
-    # parse times returned by letta.utils.get_formatted_time()
-    try:
-        return datetime.strptime(formatted_time.strip(), "%Y-%m-%d %I:%M:%S %p %Z%z")
-    except:
-        return datetime.strptime(formatted_time.strip(), "%Y-%m-%d %I:%M:%S %p")
-
-
-class PersistenceManager(ABC):
-    @abstractmethod
-    def trim_messages(self, num):
-        pass
-
-    @abstractmethod
-    def prepend_to_messages(self, added_messages):
-        pass
-
-    @abstractmethod
-    def append_to_messages(self, added_messages):
-        pass
-
-    @abstractmethod
-    def swap_system_message(self, new_system_message):
-        pass
-
-    @abstractmethod
-    def update_memory(self, new_memory):
-        pass
-
-
-class LocalStateManager(PersistenceManager):
-    """In-memory state manager has nothing to manage, all agents are held in-memory"""
-
-    recall_memory_cls = BaseRecallMemory
-    archival_memory_cls = EmbeddingArchivalMemory
-
-    def __init__(self, agent_state: AgentState):
-        # Memory held in-state useful for debugging stateful versions
-        self.memory = agent_state.memory
-        # self.messages = []  # current in-context messages
-        # self.all_messages = [] # all messages seen in current session (needed if lazily synchronizing state with DB)
-        self.archival_memory = EmbeddingArchivalMemory(agent_state)
-        self.recall_memory = BaseRecallMemory(agent_state)
-        # self.agent_state = agent_state
-
-    def save(self):
-        """Ensure storage connectors save data"""
-        self.archival_memory.save()
-        self.recall_memory.save()
-
-    '''
-    def json_to_message(self, message_json) -> Message:
-        """Convert agent message JSON into Message object"""
-
-        # get message
-        if "message" in message_json:
-            message = message_json["message"]
-        else:
-            message = message_json
-
-        # get timestamp
-        if "timestamp" in message_json:
-            timestamp = parse_formatted_time(message_json["timestamp"])
-        else:
-            timestamp = get_local_time()
-
-        # TODO: change this when we fully migrate to tool calls API
-        if "function_call" in message:
-            tool_calls = [
-                ToolCall(
-                    id=message["tool_call_id"],
-                    tool_call_type="function",
-                    function={
-                        "name": message["function_call"]["name"],
-                        "arguments": message["function_call"]["arguments"],
-                    },
-                )
-            ]
-            printd(f"Saving tool calls {[vars(tc) for tc in tool_calls]}")
-        else:
-            tool_calls = None
-
-        # if message["role"] == "function":
-        # message["role"] = "tool"
-
-        return Message(
-            user_id=self.agent_state.user_id,
-            agent_id=self.agent_state.id,
-            role=message["role"],
-            text=message["content"],
-            name=message["name"] if "name" in message else None,
-            model=self.agent_state.llm_config.model,
-            created_at=timestamp,
-            tool_calls=tool_calls,
-            tool_call_id=message["tool_call_id"] if "tool_call_id" in message else None,
-            id=message["id"] if "id" in message else None,
-        )
-    '''
-
-    def trim_messages(self, num):
-        # printd(f"InMemoryStateManager.trim_messages")
-        # self.messages = [self.messages[0]] + self.messages[num:]
-        pass
-
-    def prepend_to_messages(self, added_messages: List[Message]):
-        # first tag with timestamps
-        # added_messages = [{"timestamp": get_local_time(), "message": msg} for msg in added_messages]
-
-        printd(f"{self.__class__.__name__}.prepend_to_message")
-        # self.messages = [self.messages[0]] + added_messages + self.messages[1:]
-
-        # add to recall memory
-        self.recall_memory.insert_many([m for m in added_messages])
-
-    def append_to_messages(self, added_messages: List[Message]):
-        # first tag with timestamps
-        # added_messages = [{"timestamp": get_local_time(), "message": msg} for msg in added_messages]
-
-        printd(f"{self.__class__.__name__}.append_to_messages")
-        # self.messages = self.messages + added_messages
-
-        # add to recall memory
-        self.recall_memory.insert_many([m for m in added_messages])
-
-    def swap_system_message(self, new_system_message: Message):
-        # first tag with timestamps
-        # new_system_message = {"timestamp": get_local_time(), "message": new_system_message}
-
-        printd(f"{self.__class__.__name__}.swap_system_message")
-        # self.messages[0] = new_system_message
-
-        # add to recall memory
-        self.recall_memory.insert(new_system_message)
-
-    def update_memory(self, new_memory: Memory):
-        printd(f"{self.__class__.__name__}.update_memory")
-        assert isinstance(new_memory, Memory), type(new_memory)
-        self.memory = new_memory