langchainrb 0.6.10 → 0.6.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ba1fb0e3fbc05e4279fe3a698ad8fb1a25e02788991a8e6a7b27b411771096f3
-  data.tar.gz: 309cda1c8c7a4982b22c6ad2f82c20fb12ca3bdfdc3e8c0ebeaa9687a7f71ce0
+  metadata.gz: 78d1726db950b4234799cdf34185110411cbc666836225a63ed9d5be6b0a5575
+  data.tar.gz: 0ed5595b0aae9dcaa97ee5cbc653cf3c9cb21f6057f5439ec78711ecb1cf6b46
 SHA512:
-  metadata.gz: 3457cbad7efbc5504f4cb3b684e3837984be5d486c1ee21a718508d606dd63ccc2223dffc55f3cc9c52f5fd0a533b364407bbb4c3208515e2bab8ca2af9ea60a
-  data.tar.gz: 1ff3cded239c286ee87d7a2f0a1cfa45c734b185e35b6e6fbc63fd2951f7c7397e8562fa145a16f545ef07b2e983f40715010e9c96653b5f2bf128325a5a7577
+  metadata.gz: c0313618b12b10943e8825aaabd66306e21b0a24fefe9c4593cb4f28b1a03e099781944057ce9423c98851bad7dea07e2832c4ab37d416cf743b126a26a6a308
+  data.tar.gz: a2910e93f883bee84288e5bd22644f17b30957a6ae9a696b3adc01dc9b32e2f546b49d46c3587856a883cd4dc1f0a677a3970d357b5fe2dfd9216ce894d06120

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    langchainrb (0.6.10)
+    langchainrb (0.6.11)
       baran (~> 0.1.6)
       colorize (~> 0.8.1)
       json-schema (~> 4.0.0)
@@ -226,7 +226,7 @@ GEM
       zeitwerk (~> 2.5)
     rainbow (3.1.1)
     rake (13.0.6)
-    rb_sys (0.9.79)
+    rb_sys (0.9.81)
     rdiscount (2.2.7)
     regexp_parser (2.8.0)
     replicate-ruby (0.2.2)

data/README.md

@@ -209,6 +209,10 @@ Add `gem "anthropic", "~> 0.1.0"` to your Gemfile.
 anthropic = Langchain::LLM::Anthropic.new(api_key: ENV["ANTHROPIC_API_KEY"])
 ```
 
+```ruby
+anthropic.complete(prompt: "What is the meaning of life?")
+```
+
 ### Using Prompts 📋
 
 #### Prompt Templates

data/examples/open_ai_qdrant_function_calls.rb

@@ -11,10 +11,6 @@ functions = [
         controller_name: {
           type: :string,
           description: "the controller name, e.g. users_controller"
-        },
-        unit: {
-          type: "string",
-          enum: %w[celsius fahrenheit]
         }
       },
       required: ["controller_name"]

data/lib/langchain/llm/anthropic.rb

@@ -13,7 +13,8 @@ module Langchain::LLM
   class Anthropic < Base
     DEFAULTS = {
       temperature: 0.0,
-      completion_model_name: "claude-2"
+      completion_model_name: "claude-2",
+      max_tokens_to_sample: 256
     }.freeze
 
     # TODO: Implement token length validator for Anthropic
@@ -49,7 +50,7 @@ module Langchain::LLM
     private
 
     def compose_parameters(model, params)
-      default_params = {model: model, temperature: @defaults[:temperature]}
+      default_params = {model: model}.merge(@defaults.except(:completion_model_name))
 
       default_params.merge(params)
     end

data/lib/langchain/loader.rb

@@ -89,9 +89,9 @@ module Langchain
     end
 
     def load_from_path
-      raise FileNotFound unless File.exist?(@path)
+      return File.open(@path) if File.exist?(@path)
 
-      File.open(@path)
+      raise FileNotFound, "File #{@path} does not exist"
     end
 
     def load_from_directory(&block)

data/lib/langchain/vectorsearch/milvus.rb

@@ -32,7 +32,7 @@ module Langchain::Vectorsearch
             field: Array(texts)
           }, {
             field_name: "vectors",
-            type: ::Milvus::DATA_TYPES["binary_vector"],
+            type: ::Milvus::DATA_TYPES["float_vector"],
             field: Array(texts).map { |text| llm.embed(text: text) }
           }
         ]
@@ -47,7 +47,7 @@ module Langchain::Vectorsearch
       client.collections.create(
         auto_id: true,
         collection_name: index_name,
-        description: "Default schema created by Vectorsearch",
+        description: "Default schema created by langchain.rb",
         fields: [
           {
             name: "id",
@@ -66,7 +66,7 @@ module Langchain::Vectorsearch
             ]
           }, {
             name: "vectors",
-            data_type: ::Milvus::DATA_TYPES["binary_vector"],
+            data_type: ::Milvus::DATA_TYPES["float_vector"],
             is_primary_key: false,
             type_params: [
               {
@@ -79,6 +79,20 @@ module Langchain::Vectorsearch
       )
     end
 
+    # Create the default index
+    # @return [Boolean] The response from the server
+    def create_default_index
+      client.indices.create(
+        collection_name: "Documents",
+        field_name: "vectors",
+        extra_params: [
+          {key: "metric_type", value: "L2"},
+          {key: "index_type", value: "IVF_FLAT"},
+          {key: "params", value: "{\"nlist\":1024}"}
+        ]
+      )
+    end
+
     # Get the default schema
     # @return [Hash] The response from the server
     def get_default_schema
@@ -91,6 +105,12 @@ module Langchain::Vectorsearch
       client.collections.delete(collection_name: index_name)
     end
 
+    # Load default schema into memory
+    # @return [Boolean] The response from the server
+    def load_default_schema
+      client.collections.load(collection_name: index_name)
+    end
+
     def similarity_search(query:, k: 4)
       embedding = llm.embed(text: query)
 
@@ -101,15 +121,36 @@ module Langchain::Vectorsearch
     end
 
     def similarity_search_by_vector(embedding:, k: 4)
+      load_default_schema
+
       client.search(
         collection_name: index_name,
+        output_fields: ["id", "content", "vectors"],
         top_k: k.to_s,
         vectors: [embedding],
         dsl_type: 1,
         params: "{\"nprobe\": 10}",
-        anns_field: "content",
-        metric_type: "L2"
+        anns_field: "vectors",
+        metric_type: "L2",
+        vector_type: ::Milvus::DATA_TYPES["float_vector"]
       )
     end
+
+    # Ask a question and return the answer
+    # @param question [String] The question to ask
+    # @yield [String] Stream responses back one String at a time
+    # @return [String] The answer to the question
+    def ask(question:, &block)
+      search_results = similarity_search(query: question)
+
+      content_field = search_results.dig("results", "fields_data").select { |field| field.dig("field_name") == "content" }
+      content_data = content_field.first.dig("Field", "Scalars", "Data", "StringData", "data")
+
+      context = content_data.join("\n---\n")
+
+      prompt = generate_prompt(question: question, context: context)
+
+      llm.chat(prompt: prompt, &block)
+    end
   end
 end

data/lib/langchain/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Langchain
-  VERSION = "0.6.10"
+  VERSION = "0.6.11"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: langchainrb
 version: !ruby/object:Gem::Version
-  version: 0.6.10
+  version: 0.6.11
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-08-01 00:00:00.000000000 Z
+date: 2023-08-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: baran