fix: Colang 2.x doesn't support assistant messages

Author: christinaexyou

nemoguardrails/colang/v2_x/runtime/runtime.py

@@ -31,6 +31,7 @@
     ColangSyntaxError,
 )
 from nemoguardrails.colang.v2_x.runtime.flows import Event, FlowStatus
+from nemoguardrails.colang.v2_x.runtime.serialization import json_to_state
 from nemoguardrails.colang.v2_x.runtime.statemachine import (
     FlowConfig,
     InternalEvent,
@@ -394,10 +395,13 @@ async def process_events(
             state = State(flow_states={}, flow_configs=self.flow_configs, rails_config=self.config)
             initialize_state(state)
         elif isinstance(state, dict):
-            # TODO: Implement dict to State conversion
-            raise NotImplementedError()
-        #     if isinstance(state, dict):
-        #         state = State.from_dict(state)
+            # Convert dict to State object
+            if state.get("version") == "2.x" and "state" in state:
+                # Handle the serialized state format from API calls
+                state = json_to_state(state["state"])
+            else:
+                # TODO: Implement other dict to State conversion formats if needed
+                raise NotImplementedError("Unsupported state dict format")
 
         assert isinstance(state, State)
         assert state.main_flow_state is not None

tests/test_server_calls_with_state.py

@@ -44,6 +44,8 @@ def _test_call(config_id):
 
     # When making a second call with the returned state, the conversations should continue
     # and we should get the "Hello again!" message.
+    # For Colang 2.x, we only send the new user message, not the conversation history
+    # since the state maintains the conversation context.
     response = client.post(
         "/v1/chat/completions",
         json={
@@ -52,11 +54,7 @@ def _test_call(config_id):
                 {
                     "content": "hi",
                     "role": "user",
-                },
-                {
-                    "content": "hi",
-                    "role": "assistant",
-                },
+                }
             ],
             "state": res["state"],
         },