af981dc0b0
- Add new indirect dependencies including various Charmbracelet packages for improved UI handling. - Enhance logging in the Assistant module by adding tool completion and start logging for better traceability of tool calls. - Modify context handling in the RequestLogger to support a stack-based assistant ID management, improving the logging structure for agent requests. - Implement event service integration for better trace management and debugging capabilities.
64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
package trace
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/yaoapp/yao/trace/types"
|
|
)
|
|
|
|
// space implements the Space interface for custom space operations.
|
|
// Uses context.Background() for driver calls to decouple from caller context
|
|
// (fixes the context-fork bug where parent cancellation breaks child ops).
|
|
type space struct {
|
|
traceID string
|
|
data *types.TraceSpace
|
|
driver types.Driver
|
|
}
|
|
|
|
// NewSpace creates a new space instance
|
|
func NewSpace(traceID string, data *types.TraceSpace, driver types.Driver) types.Space {
|
|
return &space{
|
|
traceID: traceID,
|
|
data: data,
|
|
driver: driver,
|
|
}
|
|
}
|
|
|
|
// ID returns the space identifier
|
|
func (s *space) ID() string {
|
|
return s.data.ID
|
|
}
|
|
|
|
// Set stores a value by key
|
|
func (s *space) Set(key string, value any) error {
|
|
return s.driver.SetSpaceKey(context.Background(), s.traceID, s.data.ID, key, value)
|
|
}
|
|
|
|
// Get retrieves a value by key
|
|
func (s *space) Get(key string) (any, error) {
|
|
return s.driver.GetSpaceKey(context.Background(), s.traceID, s.data.ID, key)
|
|
}
|
|
|
|
// Has checks if a key exists
|
|
func (s *space) Has(key string) bool {
|
|
return s.driver.HasSpaceKey(context.Background(), s.traceID, s.data.ID, key)
|
|
}
|
|
|
|
// Delete removes a key-value pair
|
|
func (s *space) Delete(key string) error {
|
|
return s.driver.DeleteSpaceKey(context.Background(), s.traceID, s.data.ID, key)
|
|
}
|
|
|
|
// Clear removes all key-value pairs
|
|
func (s *space) Clear() error {
|
|
return s.driver.ClearSpaceKeys(context.Background(), s.traceID, s.data.ID)
|
|
}
|
|
|
|
// Keys returns all keys in the space
|
|
func (s *space) Keys() []string {
|
|
keys, err := s.driver.ListSpaceKeys(context.Background(), s.traceID, s.data.ID)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
return keys
|
|
}
|