enhancements

Author: andreibondarev

lib/langchainrb_overrides/vectorsearch/pgvector.rb

@@ -1,5 +1,8 @@
 # frozen_string_literal: true
 
+# Overriding Langchain.rb's Pgvector implementation to use ActiveRecord.
+# Original implementation: https://github.com/andreibondarev/langchainrb/blob/main/lib/langchain/vectorsearch/pgvector.rb
+
 module Langchain::Vectorsearch
   class Pgvector < Base
     #
@@ -20,7 +23,7 @@ class Pgvector < Base
     ]
     DEFAULT_OPERATOR = "cosine"
 
-    attr_reader :db, :operator, :llm
+    attr_reader :operator, :llm
     attr_accessor :model
 
     # @param url [String] The URL of the PostgreSQL database
@@ -119,7 +122,3 @@ def ask(question, k: 4, &block)
     end
   end
 end
-
-# Rails connection when configuring vectorsearch
-# Update READMEs
-# Rails migration to create a migration

spec/langchainrb_overrides/vectorsearch/pgvector_spec.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+RSpec.describe Langchain::Vectorsearch::Pgvector do
+  let(:llm) { Langchain::LLM::OpenAI.new(api_key: "123") }
+  subject { described_class.new(llm: llm) }
+
+  describe "#add_texts" do
+  end
+
+  describe "#update_text" do
+  end
+
+  describe "#create_default_schema" do
+  end
+
+  describe "#destroy_default_schema" do
+  end
+
+  describe "#similarity_search" do
+  end
+
+  describe "#similarity_search_by_vector" do
+  end
+
+  describe "#ask" do
+  end
+end

spec/langchainrb_rails/active_record/hooks_spec.rb

@@ -10,13 +10,10 @@ class Dummy
     expect(::Dummy.new).to respond_to(:as_vector)
   end
 
-  it "responds to class method: similarity_search" do
+  it "responds to class methods" do
     expect(::Dummy).to respond_to(:vectorsearch)
     expect(::Dummy).to respond_to(:similarity_search)
-  end
-
-  it "responds to class method: ask" do
-    expect(::Dummy).to respond_to(:vectorsearch)
     expect(::Dummy).to respond_to(:ask)
+    expect(::Dummy).to respond_to(:embed!)
   end
 end