Update state_orchestrator.py

state_orchestrator.py

@@ -375,46 +375,48 @@ class StateOrchestrator:
             data={"reason": "critical_error"}
         ))
         
-    async def transition_to(self, session_id: str, new_state: ConversationState):
-        """Transition to a new state"""
-        current_state = self.get_state(session_id)
-        
-        if current_state is None:
-            log_warning(f"⚠️ Session not found for transition", session_id=session_id)
-            return
+    async def transition_to(self, session_id: str, new_state: ConversationState) -> bool:
+        """
+        Transition to a new state with validation
+        """
+        try:
+            # Get session context
+            context = self._sessions.get(session_id)
+            if not context:
+                log(f"❌ Session not found for state transition | session_id={session_id}")
+                return False
             
-        # Check if transition is valid
-        if new_state not in self.VALID_TRANSITIONS.get(current_state, set()):
-            log_error(
-                f"❌ Invalid state transition",
-                session_id=session_id,
-                from_state=current_state.value,
-                to_state=new_state.value
-            )
+            # Get current state from context (DÜZELTME BURADA)
+            current_state = context.state  # context'i değil, context.state'i kullan
             
-            await publish_error(
-                session_id=session_id,
-                error_type="invalid_transition",
-                error_message=f"Cannot transition from {current_state.value} to {new_state.value}"
+            # Check if transition is valid
+            if new_state not in self.VALID_TRANSITIONS.get(current_state, set()):
+                log(f"❌ Invalid state transition | session_id={session_id}, current={current_state.value}, requested={new_state.value}")
+                return False
+            
+            # Update state
+            old_state = current_state
+            context.state = new_state
+            context.last_activity = datetime.utcnow()
+            
+            log(f"✅ State transition | session_id={session_id}, {old_state.value} → {new_state.value}")
+            
+            # Emit state transition event
+            await self._event_bus.publish(
+                EventType.STATE_TRANSITION,
+                {
+                    "session_id": session_id,
+                    "old_state": old_state.value,
+                    "new_state": new_state.value,
+                    "timestamp": datetime.utcnow().isoformat()
+                }
             )
-            return
             
-        # Update state
-        self.sessions[session_id] = new_state
-        
-        log_info(
-            f"🔄 State transition",
-            session_id=session_id,
-            from_state=current_state.value,
-            to_state=new_state.value
-        )
-        
-        # Publish state transition event
-        await publish_state_transition(
-            session_id=session_id,
-            from_state=current_state.value,
-            to_state=new_state.value
-        )
+            return True
+            
+        except Exception as e:
+            log_error(f"❌ State transition error | session_id={session_id}", e)
+            return False
         
     def get_state(self, session_id: str) -> Optional[ConversationState]:
         """Get current state for a session"""