phronomy 0.1.2 → 0.1.4

data/lib/phronomy/vector_store/in_memory.rb

@@ -14,13 +14,14 @@ module Phronomy
     class InMemory < Base
       def initialize
         @documents = {}
+        @mutex = Mutex.new
       end
 
       # @param id        [String]
       # @param embedding [Array<Float>]
       # @param metadata  [Hash]
       def add(id:, embedding:, metadata: {})
-        @documents[id] = {embedding: embedding, metadata: metadata}
+        @mutex.synchronize { @documents[id] = {embedding: embedding, metadata: metadata} }
         self
       end
 
@@ -28,7 +29,8 @@ module Phronomy
       # @param k               [Integer]
       # @return [Array<Hash>] sorted by descending score
       def search(query_embedding:, k: 5)
-        results = @documents.map do |id, doc|
+        snapshot = @mutex.synchronize { @documents.dup }
+        results = snapshot.map do |id, doc|
           score = cosine_similarity(query_embedding, doc[:embedding])
           {id: id, score: score, metadata: doc[:metadata]}
         end
@@ -36,18 +38,18 @@ module Phronomy
       end
 
       def remove(id:)
-        @documents.delete(id)
+        @mutex.synchronize { @documents.delete(id) }
         self
       end
 
       def clear
-        @documents.clear
+        @mutex.synchronize { @documents.clear }
         self
       end
 
       # @return [Integer] number of documents stored
       def size
-        @documents.size
+        @mutex.synchronize { @documents.size }
       end
 
       private

data/lib/phronomy/vector_store/redis_search.rb

@@ -38,6 +38,7 @@ module Phronomy
         @index_name = index_name
         @dimension = dimension
         @index_created = false
+        @mutex = Mutex.new
       end
 
       # @param id        [String]
@@ -79,37 +80,43 @@ module Phronomy
       end
 
       def clear
-        begin
-          @redis.call("FT.DROPINDEX", @index_name, "DD")
-        rescue => e
-          raise unless e.message.to_s.include?("Unknown Index name")
+        @mutex.synchronize do
+          begin
+            @redis.call("FT.DROPINDEX", @index_name, "DD")
+          rescue => e
+            raise unless e.message.to_s.include?("Unknown Index name")
+          end
+          @index_created = false
         end
-        @index_created = false
         self
       end
 
       private
 
       def ensure_index!(dim)
-        return if @index_created
-
-        @dimension ||= dim
-        begin
-          @redis.call(
-            "FT.CREATE", @index_name,
-            "ON", "HASH",
-            "PREFIX", 1, DOC_PREFIX,
-            "SCHEMA",
-            "embedding", "VECTOR", "FLAT", 6,
-            "TYPE", "FLOAT32",
-            "DIM", @dimension,
-            "DISTANCE_METRIC", "COSINE",
-            "metadata", "TEXT"
-          )
-        rescue => e
-          raise unless e.message.to_s.include?("Index already exists")
+        return if @index_created # fast path outside lock
+
+        @mutex.synchronize do
+          return if @index_created # re-check inside lock
+
+          @dimension ||= dim
+          begin
+            @redis.call(
+              "FT.CREATE", @index_name,
+              "ON", "HASH",
+              "PREFIX", 1, DOC_PREFIX,
+              "SCHEMA",
+              "embedding", "VECTOR", "FLAT", 6,
+              "TYPE", "FLOAT32",
+              "DIM", @dimension,
+              "DISTANCE_METRIC", "COSINE",
+              "metadata", "TEXT"
+            )
+          rescue => e
+            raise unless e.message.to_s.include?("Index already exists")
+          end
+          @index_created = true
         end
-        @index_created = true
       end
 
       # Pack a Float array as a FLOAT32 binary string for RediSearch.

data/lib/phronomy/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Phronomy
-  VERSION = "0.1.2"
+  VERSION = "0.1.4"
 end

data/lib/phronomy.rb

@@ -35,6 +35,45 @@ module Phronomy
     end
   end
 
+  # Namespace for graph-related classes (StateGraph, State, ParallelNode, …).
+  # Also serves as the registry for State classes that may be serialized to
+  # external stores (Redis, DB). Call +register_state_class+ at application
+  # startup so that only known classes can be deserialized.
+  module Graph
+    @state_class_registry = nil
+    @registry_mutex = Mutex.new
+
+    class << self
+      # Register one or more State classes that are allowed to be deserialized
+      # by StateStore backends. When at least one class is registered, only
+      # registered classes will be accepted by +StateStore::Base#safe_state_class+.
+      #
+      # Call this once at application startup (e.g. in a Rails initializer).
+      #
+      # @param classes [Array<Class>] classes including Phronomy::Graph::State
+      # @example
+      #   Phronomy::Graph.register_state_class(MyWorkflowState, OtherState)
+      def register_state_class(*classes)
+        @registry_mutex.synchronize do
+          @state_class_registry ||= {}
+          classes.each do |klass|
+            raise ArgumentError, "#{klass.inspect} is not a Class" unless klass.is_a?(Class)
+            @state_class_registry[klass.name] = klass
+          end
+        end
+      end
+
+      # Returns the current registry Hash, or nil when no class has been registered.
+      # @return [Hash{String => Class}, nil]
+      attr_reader :state_class_registry
+
+      # Clears the registry. Primarily used in tests.
+      def reset_state_class_registry!
+        @registry_mutex.synchronize { @state_class_registry = nil }
+      end
+    end
+  end
+
   class << self
     def configuration
       @configuration ||= Configuration.new

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: phronomy
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.4
 platform: ruby
 authors:
 - Raizo T.C.S
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2026-05-10 00:00:00.000000000 Z
+date: 2026-05-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby_llm