letta-nightly 0.5.3.dev20241120104224__py3-none-any.whl → 0.5.3.dev20241121022151__py3-none-any.whl

letta/client/client.py

@@ -79,6 +79,7 @@ class AbstractClient(object):
         include_base_tools: Optional[bool] = True,
         metadata: Optional[Dict] = {"human:": DEFAULT_HUMAN, "persona": DEFAULT_PERSONA},
         description: Optional[str] = None,
+        tags: Optional[List[str]] = None,
     ) -> AgentState:
         raise NotImplementedError
 
@@ -94,6 +95,7 @@ class AbstractClient(object):
         embedding_config: Optional[EmbeddingConfig] = None,
         message_ids: Optional[List[str]] = None,
         memory: Optional[Memory] = None,
+        tags: Optional[List[str]] = None,
     ):
         raise NotImplementedError
 
@@ -381,6 +383,7 @@ class RESTClient(AbstractClient):
         metadata: Optional[Dict] = {"human:": DEFAULT_HUMAN, "persona": DEFAULT_PERSONA},
         description: Optional[str] = None,
         initial_message_sequence: Optional[List[Message]] = None,
+        tags: Optional[List[str]] = None,
     ) -> AgentState:
         """Create an agent
 
@@ -394,6 +397,7 @@ class RESTClient(AbstractClient):
             include_base_tools (bool): Include base tools
             metadata (Dict): Metadata
             description (str): Description
+            tags (List[str]): Tags for filtering agents
 
         Returns:
             agent_state (AgentState): State of the created agent
@@ -435,6 +439,7 @@ class RESTClient(AbstractClient):
             llm_config=llm_config if llm_config else self._default_llm_config,
             embedding_config=embedding_config if embedding_config else self._default_embedding_config,
             initial_message_sequence=initial_message_sequence,
+            tags=tags,
         )
 
         # Use model_dump_json() instead of model_dump()
@@ -482,12 +487,12 @@ class RESTClient(AbstractClient):
         description: Optional[str] = None,
         system: Optional[str] = None,
         tools: Optional[List[str]] = None,
-        tags: Optional[List[str]] = None,
         metadata: Optional[Dict] = None,
         llm_config: Optional[LLMConfig] = None,
         embedding_config: Optional[EmbeddingConfig] = None,
         message_ids: Optional[List[str]] = None,
         memory: Optional[Memory] = None,
+        tags: Optional[List[str]] = None,
     ):
         """
         Update an existing agent
@@ -503,6 +508,7 @@ class RESTClient(AbstractClient):
             embedding_config (EmbeddingConfig): Embedding configuration
             message_ids (List[str]): List of message IDs
             memory (Memory): Memory configuration
+            tags (List[str]): Tags for filtering agents
 
         Returns:
             agent_state (AgentState): State of the updated agent
@@ -1669,6 +1675,7 @@ class LocalClient(AbstractClient):
         metadata: Optional[Dict] = {"human:": DEFAULT_HUMAN, "persona": DEFAULT_PERSONA},
         description: Optional[str] = None,
         initial_message_sequence: Optional[List[Message]] = None,
+        tags: Optional[List[str]] = None,
     ) -> AgentState:
         """Create an agent
 
@@ -1683,6 +1690,7 @@ class LocalClient(AbstractClient):
             include_base_tools (bool): Include base tools
             metadata (Dict): Metadata
             description (str): Description
+            tags (List[str]): Tags for filtering agents
 
         Returns:
             agent_state (AgentState): State of the created agent
@@ -1724,6 +1732,7 @@ class LocalClient(AbstractClient):
                 llm_config=llm_config if llm_config else self._default_llm_config,
                 embedding_config=embedding_config if embedding_config else self._default_embedding_config,
                 initial_message_sequence=initial_message_sequence,
+                tags=tags,
             ),
             actor=self.user,
         )
@@ -1780,6 +1789,7 @@ class LocalClient(AbstractClient):
             embedding_config (EmbeddingConfig): Embedding configuration
             message_ids (List[str]): List of message IDs
             memory (Memory): Memory configuration
+            tags (List[str]): Tags for filtering agents
 
         Returns:
             agent_state (AgentState): State of the updated agent

letta/embeddings.py

@@ -3,18 +3,7 @@ import uuid
 from typing import Any, List, Optional
 
 import numpy as np
-
-# from llama_index.core.base.embeddings import BaseEmbedding
-# from llama_index.core.embeddings import BaseEmbedding
-# from llama_index.core.base.embeddings.base import BaseEmbedding
-# from llama_index.bridge.pydantic import PrivateAttr
-# from llama_index.embeddings.base import BaseEmbedding
-# from llama_index.embeddings.huggingface_utils import format_text
 import tiktoken
-from llama_index.core import Document as LlamaIndexDocument
-
-# from llama_index.core.base.embeddings import BaseEmbedding
-from llama_index.core.node_parser import SentenceSplitter
 
 from letta.constants import (
     EMBEDDING_TO_TOKENIZER_DEFAULT,
@@ -26,6 +15,9 @@ from letta.utils import is_valid_url, printd
 
 
 def parse_and_chunk_text(text: str, chunk_size: int) -> List[str]:
+    from llama_index.core import Document as LlamaIndexDocument
+    from llama_index.core.node_parser import SentenceSplitter
+
     parser = SentenceSplitter(chunk_size=chunk_size)
     llama_index_docs = [LlamaIndexDocument(text=text)]
     nodes = parser.get_nodes_from_documents(llama_index_docs)

letta/server/server.py

@@ -860,6 +860,7 @@ class SyncServer(Server):
                 memory=request.memory,
                 description=request.description,
                 metadata_=request.metadata_,
+                tags=request.tags,
             )
             if request.agent_type == AgentType.memgpt_agent:
                 agent = Agent(
@@ -904,8 +905,15 @@ class SyncServer(Server):
         save_agent(agent, self.ms)
         logger.debug(f"Created new agent from config: {agent}")
 
+        # TODO: move this into save_agent. save_agent should be moved to server.py
+        if request.tags:
+            for tag in request.tags:
+                self.agents_tags_manager.add_tag_to_agent(agent_id=agent.agent_state.id, tag=tag, actor=actor)
+
         assert isinstance(agent.agent_state.memory, Memory), f"Invalid memory type: {type(agent_state.memory)}"
-        # return AgentState
+
+        # TODO: remove (hacky)
+        agent.agent_state.tags = self.agents_tags_manager.get_tags_for_agent(agent_id=agent.agent_state.id, actor=actor)
 
         return agent.agent_state
 

{letta_nightly-0.5.3.dev20241120104224.dist-info → letta_nightly-0.5.3.dev20241121022151.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: letta-nightly
-Version: 0.5.3.dev20241120104224
+Version: 0.5.3.dev20241121022151
 Summary: Create LLM agents with long-term memory and custom tools
 License: Apache License
 Author: Letta Team
@@ -56,7 +56,7 @@ Requires-Dist: prettytable (>=3.9.0,<4.0.0)
 Requires-Dist: psycopg2 (>=2.9.10,<3.0.0) ; extra == "postgres"
 Requires-Dist: psycopg2-binary (>=2.9.10,<3.0.0) ; extra == "postgres"
 Requires-Dist: pyautogen (==0.2.22) ; extra == "autogen"
-Requires-Dist: pydantic (>=2.7.4,<3.0.0)
+Requires-Dist: pydantic (>=2.7.4,<2.10.0)
 Requires-Dist: pydantic-settings (>=2.2.1,<3.0.0)
 Requires-Dist: pyhumps (>=3.8.0,<4.0.0)
 Requires-Dist: pymilvus (>=2.4.3,<3.0.0) ; extra == "milvus"

{letta_nightly-0.5.3.dev20241120104224.dist-info → letta_nightly-0.5.3.dev20241121022151.dist-info}/RECORD

@@ -13,7 +13,7 @@ letta/cli/cli.py,sha256=1dJIZ8DIGM8mg0G0UGLKMTa3fwgHzrN8Hkxd5Uxx7X4,16946
 letta/cli/cli_config.py,sha256=D-CpSZI1cDvdSQr3-zhGuDrJnZo1Ko7bi_wuxcBxxqo,8555
 letta/cli/cli_load.py,sha256=x4L8s15GwIW13xrhKYFWHo_y-IVGtoPDHWWKcHDRP10,4587
 letta/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-letta/client/client.py,sha256=rodNUpHgZjMRWgbxIylG1kR_k8QnJ0Vrat4Ms3OXPzg,98275
+letta/client/client.py,sha256=1eLqzAUKoO63Ee_p4LjJjrlqmv4phUZc2MPMXhPpEw0,98717
 letta/client/streaming.py,sha256=Hh5pjlyrdCuO2V75ZCxSSOCPd3BmHdKFGaIUJC6fBp0,4775
 letta/client/utils.py,sha256=OJlAKWrldc4I6M1WpcTWNtPJ4wfxlzlZqWLfCozkFtI,2872
 letta/config.py,sha256=eK-ip06ELHNYriInkgfidDvJxQ2tD1u49I-VLXB87nE,18929
@@ -21,7 +21,7 @@ letta/constants.py,sha256=gg9bI-dqfejJem73iTBVoCygnbKoxCX-PCrnX6kw1YQ,6653
 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=qPt8kB-wmuRIg1py7DHnQGJpw3DmQHJ505FJvc0K6Yk,8873
+letta/embeddings.py,sha256=3vJaQ8RMLLp6yiYZGsthq6Xsu4keb9gp7DYN_2GPBFU,8459
 letta/errors.py,sha256=cDOo4cSYL-LA0w0b0GdsxXd5k2I1LLOY8nhtXk9YqYs,2875
 letta/functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 letta/functions/function_sets/base.py,sha256=N4QmOjL6gDEyOg67ocF6zVKM-NquTo-yXG_T8r18buA,6440
@@ -185,7 +185,7 @@ letta/server/rest_api/routers/v1/tools.py,sha256=Bkb9oKswOycj5S3fBeim7LpDrZf37Sy
 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=GdHYCzXtbM5VCAYDPR0z5gnNZpRhwPld2BGZV7xT6cU,2924
-letta/server/server.py,sha256=IGYtbQur5Ry7zDyCfH7dM3GgmMBCJ5qrkwsuyww2A4g,76870
+letta/server/server.py,sha256=hYlHYb93kjgtBvnQ6RoRstA80wTRfsKpSECVQVkgR7s,77287
 letta/server/startup.sh,sha256=wTOQOJJZw_Iec57WIu0UW0AVflk0ZMWYZWg8D3T_gSQ,698
 letta/server/static_files/assets/index-3ab03d5b.css,sha256=OrA9W4iKJ5h2Wlr7GwdAT4wow0CM8hVit1yOxEL49Qw,54295
 letta/server/static_files/assets/index-9fa459a2.js,sha256=j2oMcDJO9dWJaH5e-tsflbVpWK20gLWpZKJk4-Kuy6A,1815592
@@ -210,8 +210,8 @@ letta/streaming_interface.py,sha256=_FPUWy58j50evHcpXyd7zB1wWqeCc71NCFeWh_TBvnw,
 letta/streaming_utils.py,sha256=329fsvj1ZN0r0LpQtmMPZ2vSxkDBIUUwvGHZFkjm2I8,11745
 letta/system.py,sha256=buKYPqG5n2x41hVmWpu6JUpyd7vTWED9Km2_M7dLrvk,6960
 letta/utils.py,sha256=COwQLAt02eEM9tjp6p5kN8YeTqGXr714l5BvffLVCLU,32376
-letta_nightly-0.5.3.dev20241120104224.dist-info/LICENSE,sha256=mExtuZ_GYJgDEI38GWdiEYZizZS4KkVt2SF1g_GPNhI,10759
-letta_nightly-0.5.3.dev20241120104224.dist-info/METADATA,sha256=Et7R4CD6nbcmKAlwkH-F1NPjNXXKcLXlEdDFtSwt5-w,11069
-letta_nightly-0.5.3.dev20241120104224.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-letta_nightly-0.5.3.dev20241120104224.dist-info/entry_points.txt,sha256=2zdiyGNEZGV5oYBuS-y2nAAgjDgcC9yM_mHJBFSRt5U,40
-letta_nightly-0.5.3.dev20241120104224.dist-info/RECORD,,
+letta_nightly-0.5.3.dev20241121022151.dist-info/LICENSE,sha256=mExtuZ_GYJgDEI38GWdiEYZizZS4KkVt2SF1g_GPNhI,10759
+letta_nightly-0.5.3.dev20241121022151.dist-info/METADATA,sha256=aZ1G_K6dHCtPcBzYou7tuKNzF8QEgntsnNvMYSHfbMQ,11070
+letta_nightly-0.5.3.dev20241121022151.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+letta_nightly-0.5.3.dev20241121022151.dist-info/entry_points.txt,sha256=2zdiyGNEZGV5oYBuS-y2nAAgjDgcC9yM_mHJBFSRt5U,40
+letta_nightly-0.5.3.dev20241121022151.dist-info/RECORD,,