A8065384/latticelm
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix context background and silent JWT

Browse Source
This commit is contained in:
Anibal Angulo
2026-03-05 06:55:44 +00:00
parent 214e63b0c5
commit ae2e1b7a80
11 changed files with 99 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
internal/server/health.go
View File

@@ -51,7 +51,7 @@ func (s *GatewayServer) handleReady(w http.ResponseWriter, r *http.Request) {
// Test conversation store by attempting a simple operation
testID := "health_check_test"
_, err := s.convs.Get(testID)
_, err := s.convs.Get(ctx, testID)
if err != nil {
checks["conversation_store"] = "unhealthy: " + err.Error()
allHealthy = false

8
internal/server/mocks_test.go
View File

@@ -156,7 +156,7 @@ func newMockConversationStore() *mockConversationStore {
}
}
func (m *mockConversationStore) Get(id string) (*conversation.Conversation, error) {
func (m *mockConversationStore) Get(ctx context.Context, id string) (*conversation.Conversation, error) {
m.mu.Lock()
defer m.mu.Unlock()
@@ -170,7 +170,7 @@ func (m *mockConversationStore) Get(id string) (*conversation.Conversation, erro
return conv, nil
}
func (m *mockConversationStore) Create(id string, model string, messages []api.Message) (*conversation.Conversation, error) {
func (m *mockConversationStore) Create(ctx context.Context, id string, model string, messages []api.Message) (*conversation.Conversation, error) {
m.mu.Lock()
defer m.mu.Unlock()
@@ -187,7 +187,7 @@ func (m *mockConversationStore) Create(id string, model string, messages []api.M
return conv, nil
}
func (m *mockConversationStore) Append(id string, messages ...api.Message) (*conversation.Conversation, error) {
func (m *mockConversationStore) Append(ctx context.Context, id string, messages ...api.Message) (*conversation.Conversation, error) {
m.mu.Lock()
defer m.mu.Unlock()
@@ -203,7 +203,7 @@ func (m *mockConversationStore) Append(id string, messages ...api.Message) (*con
return conv, nil
}
func (m *mockConversationStore) Delete(id string) error {
func (m *mockConversationStore) Delete(ctx context.Context, id string) error {
m.mu.Lock()
defer m.mu.Unlock()

6
internal/server/server.go
View File

@@ -107,7 +107,7 @@ func (s *GatewayServer) handleResponses(w http.ResponseWriter, r *http.Request)
// Build full message history from previous conversation
var historyMsgs []api.Message
if req.PreviousResponseID != nil && *req.PreviousResponseID != "" {
conv, err := s.convs.Get(*req.PreviousResponseID)
conv, err := s.convs.Get(r.Context(), *req.PreviousResponseID)
if err != nil {
s.logger.ErrorContext(r.Context(), "failed to retrieve conversation",
logger.LogAttrsWithTrace(r.Context(),
@@ -186,7 +186,7 @@ func (s *GatewayServer) handleSyncResponse(w http.ResponseWriter, r *http.Reques
ToolCalls: result.ToolCalls,
}
allMsgs := append(storeMsgs, assistantMsg)
if _, err := s.convs.Create(responseID, result.Model, allMsgs); err != nil {
if _, err := s.convs.Create(r.Context(), responseID, result.Model, allMsgs); err != nil {
s.logger.ErrorContext(r.Context(), "failed to store conversation",
logger.LogAttrsWithTrace(r.Context(),
slog.String("request_id", logger.FromContext(r.Context())),
@@ -543,7 +543,7 @@ loop:
ToolCalls: toolCalls,
}
allMsgs := append(storeMsgs, assistantMsg)
if _, err := s.convs.Create(responseID, model, allMsgs); err != nil {
if _, err := s.convs.Create(r.Context(), responseID, model, allMsgs); err != nil {
s.logger.ErrorContext(r.Context(), "failed to store conversation",
slog.String("request_id", logger.FromContext(r.Context())),
slog.String("response_id", responseID),