Merge pull request #272 from trungutt/get-sessions-by-agentName

Add endpoint to retrieve sessions by agent filename

Author: trungutt

pkg/server/server.go

@@ -101,6 +101,8 @@ func New(sessionStore session.Store, runConfig config.RuntimeConfig, teams map[s
 	group.DELETE("/agents", s.deleteAgent)
 	// List all sessions
 	group.GET("/sessions", s.getSessions)
+	// Get sessions by agent filename
+	group.GET("/sessions/agent/:id", s.getSessionsByAgent)
 	// Get a session by id
 	group.GET("/sessions/:id", s.getSession)
 	// Resume a session by id
@@ -769,6 +771,32 @@ func (s *Server) getSessions(c echo.Context) error {
 	return c.JSON(http.StatusOK, responses)
 }
 
+func (s *Server) getSessionsByAgent(c echo.Context) error {
+	agentFilename := c.Param("id")
+	if agentFilename == "" {
+		return c.JSON(http.StatusBadRequest, map[string]string{"error": "id parameter is required"})
+	}
+
+	sessions, err := s.sessionStore.GetSessionsByAgent(c.Request().Context(), agentFilename)
+	if err != nil {
+		return c.JSON(http.StatusInternalServerError, map[string]string{"error": "failed to get sessions for agent"})
+	}
+
+	responses := make([]api.SessionsResponse, len(sessions))
+	for i, sess := range sessions {
+		responses[i] = api.SessionsResponse{
+			ID:                         sess.ID,
+			Title:                      sess.Title,
+			CreatedAt:                  sess.CreatedAt.Format(time.RFC3339),
+			NumMessages:                len(sess.GetAllMessages()),
+			InputTokens:                sess.InputTokens,
+			OutputTokens:               sess.OutputTokens,
+			GetMostRecentAgentFilename: sess.GetMostRecentAgentFilename(),
+		}
+	}
+	return c.JSON(http.StatusOK, responses)
+}
+
 func (s *Server) createSession(c echo.Context) error {
 	var sessionTemplate session.Session
 	if err := c.Bind(&sessionTemplate); err != nil {

pkg/session/store.go

@@ -30,6 +30,7 @@ type Store interface {
 	AddSession(ctx context.Context, session *Session) error
 	GetSession(ctx context.Context, id string) (*Session, error)
 	GetSessions(ctx context.Context) ([]*Session, error)
+	GetSessionsByAgent(ctx context.Context, agentFilename string) ([]*Session, error)
 	DeleteSession(ctx context.Context, id string) error
 	UpdateSession(ctx context.Context, session *Session) error
 }
@@ -260,6 +261,32 @@ func (s *SQLiteSessionStore) GetSessions(ctx context.Context) ([]*Session, error
 	return sessions, nil
 }
 
+// GetSessionsByAgent retrieves all sessions for a specific agent
+func (s *SQLiteSessionStore) GetSessionsByAgent(ctx context.Context, agentFilename string) ([]*Session, error) {
+	allSessions, err := s.GetSessions(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	filteredSessions := make([]*Session, 0)
+	for _, session := range allSessions {
+		// Check if any message in this session belongs to the specified agent
+		hasAgentMessage := false
+		for _, item := range session.Messages {
+			if item.Message != nil && item.Message.AgentFilename == agentFilename {
+				hasAgentMessage = true
+				break
+			}
+		}
+
+		if hasAgentMessage {
+			filteredSessions = append(filteredSessions, session)
+		}
+	}
+
+	return filteredSessions, nil
+}
+
 // DeleteSession deletes a session by ID
 func (s *SQLiteSessionStore) DeleteSession(ctx context.Context, id string) error {
 	if id == "" {