letta-nightly 0.8.17.dev20250723104501__py3-none-any.whl → 0.9.0.dev20250724104456__py3-none-any.whl

letta/server/rest_api/routers/v1/agents.py

@@ -293,7 +293,9 @@ async def attach_tool(
     Attach a tool to an agent.
     """
     actor = await server.user_manager.get_actor_or_default_async(actor_id=actor_id)
-    return await server.agent_manager.attach_tool_async(agent_id=agent_id, tool_id=tool_id, actor=actor)
+    await server.agent_manager.attach_tool_async(agent_id=agent_id, tool_id=tool_id, actor=actor)
+    # TODO: Unfortunately we need this to preserve our current API behavior
+    return await server.agent_manager.get_agent_by_id_async(agent_id=agent_id, actor=actor)
 
 
 @router.patch("/{agent_id}/tools/detach/{tool_id}", response_model=AgentState, operation_id="detach_tool")
@@ -307,7 +309,9 @@ async def detach_tool(
     Detach a tool from an agent.
     """
     actor = await server.user_manager.get_actor_or_default_async(actor_id=actor_id)
-    return await server.agent_manager.detach_tool_async(agent_id=agent_id, tool_id=tool_id, actor=actor)
+    await server.agent_manager.detach_tool_async(agent_id=agent_id, tool_id=tool_id, actor=actor)
+    # TODO: Unfortunately we need this to preserve our current API behavior
+    return await server.agent_manager.get_agent_by_id_async(agent_id=agent_id, actor=actor)
 
 
 @router.patch("/{agent_id}/sources/attach/{source_id}", response_model=AgentState, operation_id="attach_source_to_agent")
@@ -327,7 +331,8 @@ async def attach_source(
     agent_state = await server.agent_manager.attach_missing_files_tools_async(agent_state=agent_state, actor=actor)
 
     files = await server.file_manager.list_files(source_id, actor, include_content=True)
-    await server.insert_files_into_context_window(agent_state=agent_state, file_metadata_with_content=files, actor=actor)
+    if files:
+        await server.agent_manager.insert_files_into_context_window(agent_state=agent_state, file_metadata_with_content=files, actor=actor)
 
     if agent_state.enable_sleeptime:
         source = await server.source_manager.get_source_by_id(source_id=source_id)
@@ -338,6 +343,35 @@ async def attach_source(
     return agent_state
 
 
+@router.patch("/{agent_id}/folders/attach/{folder_id}", response_model=AgentState, operation_id="attach_folder_to_agent")
+async def attach_folder_to_agent(
+    agent_id: str,
+    folder_id: str,
+    server: "SyncServer" = Depends(get_letta_server),
+    actor_id: str | None = Header(None, alias="user_id"),
+):
+    """
+    Attach a folder to an agent.
+    """
+    actor = await server.user_manager.get_actor_or_default_async(actor_id=actor_id)
+    agent_state = await server.agent_manager.attach_source_async(agent_id=agent_id, source_id=folder_id, actor=actor)
+
+    # Check if the agent is missing any files tools
+    agent_state = await server.agent_manager.attach_missing_files_tools_async(agent_state=agent_state, actor=actor)
+
+    files = await server.file_manager.list_files(folder_id, actor, include_content=True)
+    if files:
+        await server.agent_manager.insert_files_into_context_window(agent_state=agent_state, file_metadata_with_content=files, actor=actor)
+
+    if agent_state.enable_sleeptime:
+        source = await server.source_manager.get_source_by_id(source_id=folder_id)
+        safe_create_task(
+            server.sleeptime_document_ingest_async(agent_state, source, actor), logger=logger, label="sleeptime_document_ingest_async"
+        )
+
+    return agent_state
+
+
 @router.patch("/{agent_id}/sources/detach/{source_id}", response_model=AgentState, operation_id="detach_source_from_agent")
 async def detach_source(
     agent_id: str,
@@ -368,6 +402,36 @@ async def detach_source(
     return agent_state
 
 
+@router.patch("/{agent_id}/folders/detach/{folder_id}", response_model=AgentState, operation_id="detach_folder_from_agent")
+async def detach_folder_from_agent(
+    agent_id: str,
+    folder_id: str,
+    server: "SyncServer" = Depends(get_letta_server),
+    actor_id: str | None = Header(None, alias="user_id"),
+):
+    """
+    Detach a folder from an agent.
+    """
+    actor = await server.user_manager.get_actor_or_default_async(actor_id=actor_id)
+    agent_state = await server.agent_manager.detach_source_async(agent_id=agent_id, source_id=folder_id, actor=actor)
+
+    if not agent_state.sources:
+        agent_state = await server.agent_manager.detach_all_files_tools_async(agent_state=agent_state, actor=actor)
+
+    files = await server.file_manager.list_files(folder_id, actor)
+    file_ids = [f.id for f in files]
+    await server.remove_files_from_context_window(agent_state=agent_state, file_ids=file_ids, actor=actor)
+
+    if agent_state.enable_sleeptime:
+        try:
+            source = await server.source_manager.get_source_by_id(source_id=folder_id)
+            block = await server.agent_manager.get_block_with_label_async(agent_id=agent_state.id, block_label=source.name, actor=actor)
+            await server.block_manager.delete_block_async(block.id, actor)
+        except:
+            pass
+    return agent_state
+
+
 @router.patch("/{agent_id}/files/close-all", response_model=List[str], operation_id="close_all_open_files")
 async def close_all_open_files(
     agent_id: str,
@@ -382,7 +446,78 @@ async def close_all_open_files(
     """
     actor = await server.user_manager.get_actor_or_default_async(actor_id=actor_id)
 
-    return server.file_agent_manager.close_all_other_files(agent_id=agent_id, keep_file_names=[], actor=actor)
+    return await server.file_agent_manager.close_all_other_files(agent_id=agent_id, keep_file_names=[], actor=actor)
+
+
+@router.patch("/{agent_id}/files/{file_id}/open", response_model=List[str], operation_id="open_file")
+async def open_file(
+    agent_id: str,
+    file_id: str,
+    server: "SyncServer" = Depends(get_letta_server),
+    actor_id: Optional[str] = Header(None, alias="user_id"),
+):
+    """
+    Opens a specific file for a given agent.
+
+    This endpoint marks a specific file as open in the agent's file state.
+    The file will be included in the agent's working memory view.
+    Returns a list of file names that were closed due to LRU eviction.
+    """
+    actor = await server.user_manager.get_actor_or_default_async(actor_id=actor_id)
+
+    # Get the agent to access files configuration
+    try:
+        per_file_view_window_char_limit, max_files_open = await server.agent_manager.get_agent_files_config_async(
+            agent_id=agent_id, actor=actor
+        )
+    except ValueError:
+        raise HTTPException(status_code=404, detail=f"Agent with id={agent_id} not found")
+
+    # Get file metadata
+    file_metadata = await server.file_manager.get_file_by_id(file_id=file_id, actor=actor, include_content=True)
+    if not file_metadata:
+        raise HTTPException(status_code=404, detail=f"File with id={file_id} not found")
+
+    # Use enforce_max_open_files_and_open for efficient LRU handling
+    closed_files, was_already_open = await server.file_agent_manager.enforce_max_open_files_and_open(
+        agent_id=agent_id,
+        file_id=file_id,
+        file_name=file_metadata.file_name,
+        source_id=file_metadata.source_id,
+        actor=actor,
+        visible_content=file_metadata.content[:per_file_view_window_char_limit] if file_metadata.content else "",
+        max_files_open=max_files_open,
+    )
+
+    return closed_files
+
+
+@router.patch("/{agent_id}/files/{file_id}/close", response_model=None, operation_id="close_file")
+async def close_file(
+    agent_id: str,
+    file_id: str,
+    server: "SyncServer" = Depends(get_letta_server),
+    actor_id: Optional[str] = Header(None, alias="user_id"),
+):
+    """
+    Closes a specific file for a given agent.
+
+    This endpoint marks a specific file as closed in the agent's file state.
+    The file will be removed from the agent's working memory view.
+    """
+    actor = await server.user_manager.get_actor_or_default_async(actor_id=actor_id)
+
+    # Use update_file_agent_by_id to close the file
+    try:
+        await server.file_agent_manager.update_file_agent_by_id(
+            agent_id=agent_id,
+            file_id=file_id,
+            actor=actor,
+            is_open=False,
+        )
+        return JSONResponse(status_code=status.HTTP_200_OK, content={"message": f"File id={file_id} successfully closed"})
+    except NoResultFound:
+        raise HTTPException(status_code=404, detail=f"File association for file_id={file_id} and agent_id={agent_id} not found")
 
 
 @router.get("/{agent_id}", response_model=AgentState, operation_id="retrieve_agent")
@@ -440,6 +575,19 @@ async def list_agent_sources(
     return await server.agent_manager.list_attached_sources_async(agent_id=agent_id, actor=actor)
 
 
+@router.get("/{agent_id}/folders", response_model=list[Source], operation_id="list_agent_folders")
+async def list_agent_folders(
+    agent_id: str,
+    server: "SyncServer" = Depends(get_letta_server),
+    actor_id: str | None = Header(None, alias="user_id"),  # Extract user_id from header, default to None if not present
+):
+    """
+    Get the folders associated with an agent.
+    """
+    actor = await server.user_manager.get_actor_or_default_async(actor_id=actor_id)
+    return await server.agent_manager.list_attached_sources_async(agent_id=agent_id, actor=actor)
+
+
 # TODO: remove? can also get with agent blocks
 @router.get("/{agent_id}/core-memory", response_model=Memory, operation_id="retrieve_agent_memory")
 async def retrieve_agent_memory(