2 files changed, 48 insertions, 2 deletions
diff --git a/state/memory_backend.go b/state/memory_backend.go
index 58e86e0..8e37a8d 100644
--- a/state/memory_backend.go
+++ b/state/memory_backend.go
@@ -36,6 +36,11 @@ func (ms *MemoryStateBackend) SetCurrentKey(key string) error {
 }
 
 func (ms *MemoryStateBackend) GetValueState(ctx context.Context, key string) (core.ValueState[any], error) {
-	//TODO implement me
-	panic("implement me")
+	// Assign if state not existed
+
+	if _, ok := ms.states[key]; !ok {
+		ms.states[key] = make(map[int]map[string]any)
+	}
+
+	return &memoryStateValue{backend: ms, stateName: key}, nil
 }
diff --git a/state/memory_state.go b/state/memory_state.go
new file mode 100644
index 0000000..b7d26e2
--- /dev/null
+++ b/state/memory_state.go
@@ -0,0 +1,41 @@
+package state
+
+import "context"
+
+type memoryStateValue struct {
+	backend   *MemoryStateBackend
+	stateName string
+}
+
+func (m *memoryStateValue) Value(ctx context.Context) (any, error) {
+	kg := m.backend.currentGroupKey
+	key := m.backend.currentKey
+
+	if kgMap, exist := m.backend.states[m.stateName][kg]; exist {
+		if v, hasKey := kgMap[key]; hasKey {
+			return v, nil
+		}
+	}
+
+	return nil, nil
+}
+
+func (m *memoryStateValue) Update(ctx context.Context, newVal any) (any, error) {
+	kg := m.backend.currentGroupKey
+	key := m.backend.currentKey
+	if _, exists := m.backend.states[m.stateName][kg]; !exists {
+		m.backend.states[m.stateName][kg] = make(map[string]any)
+	}
+
+	m.backend.states[m.stateName][kg][key] = newVal
+	return newVal, nil
+}
+
+func (m *memoryStateValue) Clear(ctx context.Context) error {
+	kg := m.backend.currentGroupKey
+	key := m.backend.currentKey
+	if kgMap, exists := m.backend.states[m.stateName][kg]; exists {
+		delete(kgMap, key)
+	}
+	return nil
+}