Fixing imports

Signed-off-by: Dor Zohar <dor@emcie.co>

Author: mc-dorzo

src/parlant/core/engines/alpha/guideline_matching/generic/journey/journey_node_selection_batch.py

@@ -39,7 +39,6 @@
 from parlant.core.nlp.generation_info import GenerationInfo, UsageInfo
 from parlant.core.sessions import EventKind, ToolEventData
 from parlant.core.tools import ToolId
-from parlant.core.store_provider import ENGINE_CALL_SITE, StoreProvider
 from parlant.core.store_provider import StoreProvider, StoreProviderHints
 
 

src/parlant/sdk.py

@@ -4592,10 +4592,9 @@ async def _apply_single_evaluation(
         result: _CachedEvaluator.GuidelineEvaluation | _CachedEvaluator.JourneyEvaluation,
     ) -> None:
         if entity_type == "guideline":
-            if entity_type == "guideline":
-                guideline = await self._store_provider.get_store(
-                    GuidelineStore, StoreProviderHints(call_site="sdk")
-                ).read_guideline(guideline_id=cast(GuidelineId, entity_id))
+            guideline = await self._store_provider.get_store(
+                GuidelineStore, StoreProviderHints(call_site="sdk")
+            ).read_guideline(guideline_id=cast(GuidelineId, entity_id))
 
             properties = cast(_CachedEvaluator.GuidelineEvaluation, result).properties
 
@@ -4610,22 +4609,20 @@ async def _apply_single_evaluation(
                     value=value,
                 )
 
-            elif entity_type == "journey":
-                # Store evaluation results on Journey.node_properties (not
-                # individual nodes) to avoid cross-journey interference when
-                # the same node is referenced by multiple journeys. The
-                # projection injects this data into guidelines at read time.
-                journey_id = cast(JourneyId, entity_id)
-                eval_node_properties = cast(
-                    _CachedEvaluator.JourneyEvaluation, result
-                ).node_properties
+        elif entity_type == "journey":
+            # Store evaluation results on Journey.node_properties (not
+            # individual nodes) to avoid cross-journey interference when
+            # the same node is referenced by multiple journeys. The
+            # projection injects this data into guidelines at read time.
+            journey_id = cast(JourneyId, entity_id)
+            eval_node_properties = cast(_CachedEvaluator.JourneyEvaluation, result).node_properties
 
-                await self._store_provider.get_store(
-                    JourneyStore, StoreProviderHints(call_site="sdk")
-                ).set_node_properties(
-                    journey_id=journey_id,
-                    node_properties=cast(Mapping[str, JSONSerializable], eval_node_properties),
-                )
+            await self._store_provider.get_store(
+                JourneyStore, StoreProviderHints(call_site="sdk")
+            ).set_node_properties(
+                journey_id=journey_id,
+                node_properties=cast(Mapping[str, JSONSerializable], eval_node_properties),
+            )
 
     async def _setup_retrievers(self) -> None:
         async def setup_retriever(

tests/sdk/test_journeys.py

@@ -2184,7 +2184,6 @@ async def run(self, ctx: Context) -> None:
 
 
 class Test_that_tool_state_runs_again_after_missing_data(SDKTest):
-
     async def setup(self, server: p.Server) -> None:
         @tool
         def find_user_id_by_name(
@@ -2264,6 +2263,7 @@ async def run(self, ctx: Context) -> None:
 
         assert "john_smith_8831" in third_response.lower()
 
+
 class Test_that_same_sub_journey_can_be_linked_multiple_times_to_same_parent(SDKTest):
     async def setup(self, server: p.Server) -> None:
         self.agent = await server.create_agent(