Skip to content

Commit 5950c73

Browse files
committed
Update steering.py
remove random steering test
1 parent 24fa443 commit 5950c73

File tree

1 file changed

+9
-20
lines changed

1 file changed

+9
-20
lines changed

optillm/autothink/steering.py

Lines changed: 9 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -666,29 +666,18 @@ def _try_token_match(self):
666666

667667
# Apply fuzzy match if found
668668
if best_match['vector'] is not None:
669-
self.match_found = True
670-
self.current_vector = best_match['vector']
671-
pattern = best_match['vector'].get("reasoning_pattern", "unknown")
672-
pivot_token = best_match['vector'].get("pivot_token", "")
673-
similarity = best_match.get('similarity', 0.0)
674-
675-
logger.info(f"STEERING: Found fuzzy match ({similarity:.2f} similarity) for {pattern} pattern")
676-
logger.info(f"STEERING: Pivot token: '{pivot_token}'")
677-
678-
return True
679-
680-
# TEMPORARY: Force a match for testing purposes sometimes
681-
if not self.match_found and len(self.manager.steering_vectors) > 0 and random.random() < 0.05:
682-
logger.info("STEERING: Forcing a random steering vector match for testing")
683-
# Pick a random vector
684-
random_vector = random.choice(self.manager.steering_vectors)
685669
self.match_found = True
686-
self.current_vector = random_vector
687-
pattern = random_vector.get("reasoning_pattern", "unknown")
688-
logger.info(f"STEERING: Forced '{pattern}' pattern for testing")
670+
self.current_vector = best_match['vector']
671+
pattern = best_match['vector'].get("reasoning_pattern", "unknown")
672+
pivot_token = best_match['vector'].get("pivot_token", "")
673+
similarity = best_match.get('similarity', 0.0)
674+
675+
logger.info(f"STEERING: Found fuzzy match ({similarity:.2f} similarity) for {pattern} pattern")
676+
logger.info(f"STEERING: Pivot token: '{pivot_token}'")
677+
689678
return True
690679

691-
return False
680+
return False
692681

693682
def _try_text_match(self):
694683
"""Try to match using text-based context (original approach)."""

0 commit comments

Comments
 (0)